一种文本分类方法和装置与流程

1.本发明涉及计算机技术领域，尤其涉及一种文本分类方法和装置。


背景技术：

2.在实际应用场景中，常常需要对用户反馈的文本进行分类，以确定用户反馈的问题所属的类别。目前，一般通过用户反馈的文本及其类别标签训练分类模型，利用训练后的分类模型预测文本所属的类别。
3.在系统升级的过程中，数据的风格会产生变化。对于同一类别的问题，在不同系统中可能存在不同的表述，例如，文本的长短和文本的用词发生变化。以第一系统升级至第二系统为例，为了适应系统的变化，现有技术对从第二系统中获取的文本进行标注，重新训练分类模型。
4.但是，由于需要标注大量的文本，现有技术对文本进行分类的成本较高，而且，需要花费较长的时间。


技术实现要素：

5.有鉴于此，本发明实施例提供一种文本分类方法和装置，能够降低文本分类的成本，缩短文本分类的时间。
6.第一方面，本发明实施例提供了一种文本分类方法，包括：
7.根据第一系统的第一训练文本及所述第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型；
8.根据所述第一训练文本及其类别标签、第二系统的第二训练文本训练所述文本分类模型，得到预测模型；
9.将所述第二系统的预测样本输入所述预测模型，得到所述预测样本的类别标签。
10.可选地，
11.所述根据所述第一训练文本及其类别标签、第二系统的第二训练文本训练所述文本分类模型，得到预测模型，包括：
12.根据经过k-1次训练的文本分类模型，确定所述第二训练文本第k 次分类的类别标签；
13.根据所述第一训练文本及其类别标签、所述第二训练文本及第二训练文本第k次分类的类别标签训练经过k-1次训练的文本分类模型；
14.根据经过k次训练的文本分类模型，确定所述第二训练文本第k 1 次分类的类别标签；
15.确定第二训练文本第k次和第k 1次分类的类别标签是否满足预设的相似条件，如果是，确定经过k次训练的文本分类模型为所述预测模型，否则，将k更新为k 1后执行所述根据经过k-1次训练的文本分类模型，确定所述第二训练文本第k次分类的类别标签；
16.其中，k为大于0的整数。
17.可选地，
18.所述相似条件，包括：第二训练文本第k次和第k 1次分类的类别标签的重合率大于相似阈值。
19.可选地，
20.所述初始分类模型，包括：bert(bidirectional encoderrepresentations from transformers，基于transformer的双向编码器表征)模型和全连接层。
21.可选地，
22.进一步包括：
23.确定超参数；其中，所述超参数用于控制所述第一训练文本和所述第二训练文本对损失值的影响程度；
24.根据所述超参数，确定所述损失函数；
25.所述根据所述第一训练文本及其类别标签、所述第二训练文本及第二训练文本第k次分类的类别标签训练经过k-1次训练的文本分类模型，包括：
26.根据所述损失函数、所述第一训练文本及其类别标签、所述第二训练文本及其第k次分类的类别标签训练经过k-1次训练的文本分类模型。
27.可选地，
28.进一步包括：
29.确定验证文本及其类别标签；
30.根据所述验证文本及其类别标签，确定所述超参数对应的分类准确率；
31.根据所述分类准确率调整所述超参数。
32.第二方面，本发明实施例提供了一种文本分类装置，包括：
33.第一训练模块，配置为根据第一系统的第一训练文本及所述第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型；
34.第二训练模块，配置为根据所述第一训练文本及其类别标签、第二系统的第二训练文本训练所述文本分类模型，得到预测模型；
35.预测模块，配置为将所述第二系统的预测样本输入所述预测模型，得到所述预测样本的类别标签。
36.可选地，
37.所述第二训练模块，配置为根据经过k-1次训练的文本分类模型，确定所述第二训练文本第k次分类的类别标签；根据所述第一训练文本及其类别标签、所述第二训练文本及第二训练文本第k次分类的类别标签训练经过k-1次训练的文本分类模型；根据经过k次训练的文本分类模型，确定所述第二训练文本第k 1次分类的类别标签；确定第二训练文本第k次和第k 1次分类的类别标签是否满足预设的相似条件，如果是，确定经过k次训练的文本分类模型为所述预测模型，否则，将k更新为k 1后执行所述根据经过k-1次训练的文本分类模型，确定所述第二训练文本第k次分类的类别标签；
38.其中，k为大于0的整数。
39.第三方面，本发明实施例提供了一种电子设备，包括：
40.一个或多个处理器；
41.存储装置，用于存储一个或多个程序，
42.当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现上述任一实施例所述的方法。
43.第四方面，本发明实施例提供了一种计算机可读介质，其上存储有计算机程序，所述程序被处理器执行时实现上述任一实施例所述的方法。
44.上述发明中的一个实施例具有如下优点或有益效果：考虑到系统变化前后，文本的类别标签是一致的，因此，本发明实施例基于第一系统的第一训练文本和第二系统的第二训练文本对文本分类模型进行训练，使得到的预测模型能够对第二系统产生的预测样本进行分类。本发明实施例不需要对第二训练文本进行标注，能够节省标注的成本，提高文本分类的效率。
45.上述的非惯用的可选方式所具有的进一步效果将在下文中结合具体实施方式加以说明。
附图说明
46.附图用于更好地理解本发明，不构成对本发明的不当限定。其中：
47.图1是本发明的一个实施例提供的一种文本分类方法的流程图；
48.图2是本发明的另一个实施例提供的一种文本分类方法的流程图；
49.图3是本发明的一个实施例提供的一种文本分类装置的示意图；
50.图4是本发明实施例可以应用于其中的示例性系统架构图；
51.图5是适于用来实现本发明实施例的终端设备或服务器的计算机系统的结构示意图。
具体实施方式
52.以下结合附图对本发明的示范性实施例做出说明，其中包括本发明实施例的各种细节以助于理解，应当将它们认为仅仅是示范性的。因此，本领域普通技术人员应当认识到，可以对这里描述的实施例做出各种改变和修改，而不会背离本发明的范围和精神。同样，为了清楚和简明，以下的描述中省略了对公知功能和结构的描述。
53.当第一系统转变为第二系统时，现有技术对从第二系统中获取的文本进行标注，以确定文本的类别标签，并根据该文本及其标签训练分类模型。
54.但是，由于需要标注大量的文本，现有技术对文本进行分类的成本较高。另外，第二系统产生训练需要的文本需要一定的时间，只有当文本积累到一定数量后，才能够训练分类模型，因此，现有技术需要花费较长的时间。
55.鉴于此，如图1所示，本发明实施例提供了一种文本分类方法，包括：
56.步骤101：根据第一系统的第一训练文本及第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型。
57.第一系统为系统升级前的旧系统，第二系统为系统升级后的新系统。第一训练文本为旧系统产生的用户反馈文本。
58.虽然第一系统和第二系统产生的用户反馈文本的风格发生变化，但是，两者的用户反馈文本所涉及的类别是相同的，例如，第一系统涉及产品质量问题、售后服务问题、发货问题，则第二系统产生的用户反馈文本的类别属于产品质量问题、售后服务问题和发货
问题中一种。需要说明的是，本发明实施例提到的第一训练文本的类别标签为第一训练文本的实际类别标签。
59.因此，本发明实施例基于第一训练文本训练得到文本分类模型，以便于确定第二训练文本的标签类别，避免对第二训练文本进行标注，能够节省标注时间和成本。
60.步骤102：根据第一训练文本及其类别标签、第二系统的第二训练文本训练文本分类模型，得到预测模型。
61.第二训练文本为新系统产生的用户反馈文本。
62.步骤103：将第二系统的预测样本输入预测模型，得到预测样本的类别标签。
63.第二训练文本可以是1-6月份的用户反馈文本，而预测样本可以是 7月份的用户反馈文本。
64.考虑到系统变化前后，文本的类别标签是一致的，因此，本发明实施例基于第一系统的第一训练文本和第二系统的第二训练文本对文本分类模型进行训练，使得到的预测模型能够对第二系统产生的预测样本进行分类。本发明实施例不需要对第二训练文本进行标注，能够节省标注的成本，提高文本分类的效率。
65.在本发明的一个实施例中，根据第一训练文本及其类别标签、第二系统的第二训练文本训练文本分类模型，得到预测模型，包括：
66.根据经过k-1次训练的文本分类模型，确定第二训练文本第k次分类的类别标签；
67.根据第一训练文本及其类别标签、第二训练文本及第二训练文本第k次分类的类别标签训练经过k-1次训练的文本分类模型；
68.根据经过k次训练的文本分类模型，确定第二训练文本第k 1次分类的类别标签；
69.确定第二训练文本第k次和第k 1次分类的类别标签是否满足预设的相似条件，如果是，确定经过k次训练的文本分类模型为预测模型，否则，将k更新为k 1后执行根据经过k-1次训练的文本分类模型，确定第二训练文本第k次分类的类别标签；
70.其中，k为大于0的整数。
71.在本发明实施例中，如果k＝1，则经过0次训练的文本分类模型指的是由初始分类模型训练得到的文本分类模型。如果第二训练文本第k 次和第k 1次分类的类别标签满足预设的相似条件，则说明第二训练文本前后两次的分类结果相似程度大，可以终止训练过程。
72.在本发明的一个实施例中，相似条件，包括：第二训练文本第k 次和第k 1次分类的类别标签的重合率大于相似阈值。
73.例如，第二训练文本第k次和第k 1次分类的类别标签的重合率＝第二训练文本第k次和第k 1次分类的类别标签相同的第二训练文本的数量/第二训练文本的总数量。例如，第二训练文本包括a、b、c、 d，第1次分类的类别标签分别为a1、b1、c1、d1，第2次分类的类别标签分别为a1、b1、c2、d2，则第二训练文本第1次和第2次分类的类别标签的重合率＝2/4。
74.其中，相似阈值可以根据实际业务需求进行调整，例如，在一个对分类结果要求较高的场景下，相似阈值为99％，在另一个对分类结果要求较低的场景下，相似阈值为90％。
75.在本发明的一个实施例中，初始分类模型，包括：bert模型和全连接层。
76.在本发明实施例中，对第一训练文本进行预处理，对经过预处理的第一训练文本进行分词，得到若干字符，对各个字符进行编码，将编码结果输入bert模型，将bert模型的
输出结果输入全连接层。
77.预处理用于去除不规范字符，其中，不规范字符可以根据实际应用场景的需求确定，例如，在某一场景下，不规范字符为“～”。
78.与之类似，在根据第二训练文本训练经过k-1次训练的文本分类模型时，对第二训练文本也需要进行预处理、分词、编码等。
79.本发明实施例利用bert模型在自然语言处理中的优势，提高对文本分类的准确性。在实际应用场景中，初始分类模型还可以为rnn (recurrent neural network，循环网络模型)、textcnn(textconvolutional neural networks，文本卷积神经网络)等。
80.在本发明的一个实施例中，该方法还包括：
81.确定超参数；其中，超参数用于控制第一训练文本和第二训练文本对损失值的影响程度；
82.根据超参数，确定损失函数；
83.根据第一训练文本及其类别标签、第二训练文本及第二训练文本第k次分类的类别标签训练经过k-1次训练的文本分类模型，包括：
84.根据损失函数、第一训练文本及其类别标签、第二训练文本及其第k次分类的类别标签训练经过k-1次训练的文本分类模型。
85.在本发明实施例中，损失函数包括：
[0086][0087]
其中，loss用于表征第k次训练的损失值，batch
l
用于表征第一训练文本的数量，batch
u
用于表征第二训练文本的数量，λ用于表征超参数，ce(
·
)为交叉熵损失函数，为第i个第一训练文本，为第i个第一训练文本的实际类别标签，为第i个第一训练文本第k次分类的预测类别标签，为第i个第二训练文本，为第i个第二训练文本第k次分类的预测类别标签，为第i个第二训练文本第k 1次分类的预测类别标签,。
[0088]
以第一次训练文本分类模型为例，为第i个第一训练文本第1 次分类的预测类别标签，为第i个第二训练文本第2次分类的预测类别标签。
[0089]
本发明实施例通过λ调整第一训练文本和第二训练文本对loss的影响程度。
[0090]
在本发明的一个实施例中，该方法还包括：
[0091]
确定验证文本及其类别标签；
[0092]
根据验证文本及其类别标签，确定超参数对应的分类准确率；
[0093]
根据分类准确率调整超参数。
[0094]
该验证文本为第二系统产生的用户反馈文本，通过分类准确率调整超参数的值，使得最终的预测结果更加准确。
[0095]
如图2所示，本发明实施例提供了一种文本分类方法，包括：
[0096]
步骤201：根据第一系统的第一训练文本及第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型。
[0097]
初始分类模型包括：bert模型和全连接层。
[0098]
利用大量第一训练样本训练初始分类模型，其中，对每个第一训练样本的处理过程包括：对第一训练文本及其类别标签进行预处理，对预处理结果进行编码，将编码结果输入bert模型，将bert模型的输出作为全连接层的输入，根据全连接层的输出训练初始分类模型。
[0099]
在该步骤中，可以采用现有的交叉熵损失函数训练初始分类模型。
[0100]
步骤202：根据经过k-1次训练的文本分类模型，确定第二训练文本第k次分类的类别标签。
[0101]
对第二训练文本进行预处理，对预处理结果进行编码，将编码结果输入文本分类模型，得到第二训练文本的分类标签。其中，k为大于 0的整数。
[0102]
步骤203：确定超参数；其中，超参数用于控制第一训练文本和第二训练文本对损失值的影响程度。
[0103]
在首次训练的时候，该超参数等于预设值，在后续训练的过程中，通过验证文本对超参数进行调整。
[0104]
步骤204：根据超参数，确定损失函数。
[0105]
根据式(1)确定损失函数。
[0106]
步骤205：根据损失函数、第一训练文本及其类别标签、第二训练文本及其第k次分类的类别标签训练经过k-1次训练的文本分类模型。
[0107]
步骤206：根据经过k次训练的文本分类模型，确定第二训练文本第k 1次分类的类别标签。
[0108]
步骤207：确定第二训练文本第k次和第k 1次分类的类别标签的重合率是否大于相似阈值，如果是，执行步骤208，否则，将k更新为 k 1后执行步骤202。
[0109]
步骤208：确定经过k次训练的文本分类模型为预测模型。
[0110]
步骤209：将第二系统的预测样本输入预测模型，得到预测样本的类别标签。
[0111]
本发明实施例基于第一训练文本及其类别标签，以及第二训练文本及分类得到的类别标签对文本分类模型进行训练，得到预测模型。该方法由第一训练文本训练得到的初始分类模型确定第二训练文本第 1次分类的类别标签，无需人工标注，能够节省标注的成本和时间，提高文本分类的效率。
[0112]
如图3所示，本发明实施例提供了一种文本分类装置，包括：
[0113]
第一训练模块301，配置为根据第一系统的第一训练文本及第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型；
[0114]
第二训练模块302，配置为根据第一训练文本及其类别标签、第二系统的第二训练文本训练文本分类模型，得到预测模型；
[0115]
预测模块303，配置为将第二系统的预测样本输入预测模型，得到预测样本的类别标签。
[0116]
在本发明的一个实施例中，第二训练模块302，配置为根据经过 k-1次训练的文本分类模型，确定第二训练文本第k次分类的类别标签；根据第一训练文本及其类别标签、第二训练文本及第二训练文本第k 次分类的类别标签训练经过k-1次训练的文本分类模型；根据经过k 次训练的文本分类模型，确定第二训练文本第k 1次分类的类别标签；确定第二训练文本第k次和第k 1次分类的类别标签是否满足预设的相似条件，如果是，确定经过k次
训练的文本分类模型为预测模型，否则，将k更新为k 1后执行根据经过k-1次训练的文本分类模型，确定第二训练文本第k次分类的类别标签；
[0117]
其中，k为大于0的整数。
[0118]
在本发明的一个实施例中，相似条件，包括：第二训练文本第k 次和第k 1次分类的类别标签的重合率大于相似阈值。
[0119]
在本发明的一个实施例中，初始分类模型，包括：基于transformer 的双向编码器表征bert模型和全连接层。
[0120]
在本发明的一个实施例中，第二训练模块302，配置为确定超参数；其中，超参数用于控制第一训练文本和第二训练文本对损失值的影响程度；根据超参数，确定损失函数；根据损失函数、第一训练文本及其类别标签、第二训练文本及其第k次分类的类别标签训练经过k-1 次训练的文本分类模型。
[0121]
在本发明的一个实施例中，第二训练模块302，配置为确定验证文本；根据验证文本确定超参数对应的分类准确率；根据分类准确率调整超参数。
[0122]
本发明实施例提供了一种电子设备，包括：
[0123]
一个或多个处理器；
[0124]
存储装置，用于存储一个或多个程序，
[0125]
当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现上述任一实施例的方法。
[0126]
图4示出了可以应用本发明实施例的文本分类方法或文本分类装置的示例性系统架构400。
[0127]
如图4所示，系统架构400可以包括终端设备401、402、403，网络404和服务器405。网络404用以在终端设备401、402、403和服务器405之间提供通信链路的介质。网络404可以包括各种连接类型，例如有线、无线通信链路或者光纤电缆等等。
[0128]
用户可以使用终端设备401、402、403通过网络404与服务器405 交互，以接收或发送消息等。终端设备401、402、403上可以安装有各种通讯客户端应用，例如购物类应用、网页浏览器应用、搜索类应用、即时通信工具、邮箱客户端、社交平台软件等(仅为示例)。
[0129]
终端设备401、402、403可以是具有显示屏并且支持网页浏览的各种电子设备，包括但不限于智能手机、平板电脑、膝上型便携计算机和台式计算机等等。
[0130]
服务器405可以是提供各种服务的服务器，例如对用户利用终端设备401、402、403所浏览的购物类网站提供支持的后台管理服务器 (仅为示例)。后台管理服务器可以对接收到的产品信息查询请求等数据进行分析等处理，并将处理结果(例如目标推送信息、产品信息
--ꢀ
仅为示例)反馈给终端设备。
[0131]
需要说明的是，本发明实施例所提供的文本分类方法一般由服务器405执行，相应地，文本分类装置一般设置于服务器405中。
[0132]
应该理解，图4中的终端设备、网络和服务器的数目仅仅是示意性的。根据实现需要，可以具有任意数目的终端设备、网络和服务器。
[0133]
下面参考图5，其示出了适于用来实现本发明实施例的终端设备的计算机系统500的结构示意图。图5示出的终端设备仅仅是一个示例，不应对本发明实施例的功能和使用范围带来任何限制。
[0134]
如图5所示，计算机系统500包括中央处理单元(cpu)501，其可以根据存储在只读存储器(rom)502中的程序或者从存储部分508 加载到随机访问存储器(ram)503中的程序而执行各种适当的动作和处理。在ram 503中，还存储有系统500操作所需的各种程序和数据。cpu 501、rom 502以及ram 503通过总线504彼此相连。输入/ 输出(i/o)接口505也连接至总线504。
[0135]
以下部件连接至i/o接口505：包括键盘、鼠标等的输入部分506；包括诸如阴极射线管(crt)、液晶显示器(lcd)等以及扬声器等的输出部分507；包括硬盘等的存储部分508；以及包括诸如lan卡、调制解调器等的网络接口卡的通信部分509。通信部分509经由诸如因特网的网络执行通信处理。驱动器510也根据需要连接至i/o接口505。可拆卸介质511，诸如磁盘、光盘、磁光盘、半导体存储器等等，根据需要安装在驱动器510上，以便于从其上读出的计算机程序根据需要被安装入存储部分508。
[0136]
特别地，根据本发明公开的实施例，上文参考流程图描述的过程可以被实现为计算机软件程序。例如，本发明公开的实施例包括一种计算机程序产品，其包括承载在计算机可读介质上的计算机程序，该计算机程序包含用于执行流程图所示的方法的程序代码。在这样的实施例中，该计算机程序可以通过通信部分509从网络上被下载和安装，和/或从可拆卸介质511被安装。在该计算机程序被中央处理单元(cpu) 501执行时，执行本发明的系统中限定的上述功能。
[0137]
需要说明的是，本发明所示的计算机可读介质可以是计算机可读信号介质或者计算机可读存储介质或者是上述两者的任意组合。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、装置或器件，或者任意以上的组合。计算机可读存储介质的更具体的例子可以包括但不限于：具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机访问存储器(ram)、只读存储器(rom)、可擦式可编程只读存储器(eprom或闪存)、光纤、便携式紧凑磁盘只读存储器(cd-rom)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本发明中，计算机可读存储介质可以是任何包含或存储程序的有形介质，该程序可以被指令执行系统、装置或者器件使用或者与其结合使用。而在本发明中，计算机可读的信号介质可以包括在基带中或者作为载波一部分传播的数据信号，其中承载了计算机可读的程序代码。这种传播的数据信号可以采用多种形式，包括但不限于电磁信号、光信号或上述的任意合适的组合。计算机可读的信号介质还可以是计算机可读存储介质以外的任何计算机可读介质，该计算机可读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。计算机可读介质上包含的程序代码可以用任何适当的介质传输，包括但不限于：无线、电线、光缆、rf等等，或者上述的任意合适的组合。
[0138]
附图中的流程图和框图，图示了按照本发明各种实施例的系统、方法和计算机程序产品的可能实现的体系架构、功能和操作。在这点上，流程图或框图中的每个方框可以代表一个模块、程序段、或代码的一部分，上述模块、程序段、或代码的一部分包含一个或多个用于实现规定的逻辑功能的可执行指令。也应当注意，在有些作为替换的实现中，方框中所标注的功能也可以以不同于附图中所标注的顺序发生。例如，两个接连地表示的方框实际上可以基本并行地执行，它们有时也可以按相反的顺序执行，这依所涉及的功能而定。也要注意的是，框图或流程图中的每个方框、以及框图或流程图中的方框的组合，可以用执行规
定的功能或操作的专用的基于硬件的系统来实现，或者可以用专用硬件与计算机指令的组合来实现。
[0139]
描述于本发明实施例中所涉及到的模块可以通过软件的方式实现，也可以通过硬件的方式来实现。所描述的模块也可以设置在处理器中，例如，可以描述为：一种处理器包括发送模块、获取模块、确定模块和第一处理模块。其中，这些模块的名称在某种情况下并不构成对该模块本身的限定，例如，发送模块还可以被描述为“向所连接的服务端发送图片获取请求的模块”。
[0140]
作为另一方面，本发明还提供了一种计算机可读介质，该计算机可读介质可以是上述实施例中描述的设备中所包含的；也可以是单独存在，而未装配入该设备中。
[0141]
上述计算机可读介质承载有一个或者多个程序，当上述一个或者多个程序被一个该设备执行时，使得该设备包括：
[0142]
根据第一系统的第一训练文本及所述第一训练文本的类别标签训练预设的初始分类模型，得到文本分类模型；
[0143]
根据所述第一训练文本及其类别标签、第二系统的第二训练文本训练所述文本分类模型，得到预测模型；
[0144]
将所述第二系统的预测样本输入所述预测模型，得到所述预测样本的类别标签。
[0145]
根据本发明实施例的技术方案，基于第一系统的第一训练文本和第二系统的第二训练文本对文本分类模型进行训练，使得到的预测模型能够对第二系统产生的预测样本进行分类。本发明实施例不需要对第二训练文本进行标注，能够节省标注的成本，提高文本分类的效率。
[0146]
上述具体实施方式，并不构成对本发明保护范围的限制。本领域技术人员应该明白的是，取决于设计要求和其他因素，可以发生各种各样的修改、组合、子组合和替代。任何在本发明的精神和原则之内所作的修改、等同替换和改进等，均应包含在本发明保护范围之内。