The application provides a text categorization method and device. The method includes: extracting the keyword in the target text; based on the extracted keyword, using the pretrained word vector model, the word vector features of the target text are obtained, and the word vector features of the target text are calculated, and the similarity values between the words Vector Features of each class category of the pre set class N class tree are respectively calculated. N is a positive integer; the category corresponding to the maximum similarity value is determined as the category of the target text. This application achieves the automatic classification of news. Compared with the manual classification method in the existing technology, this application greatly improves the efficiency of news classification.
【技术实现步骤摘要】
一种文本分类方法和装置
本申请涉及文本处理
,尤其涉及一种文本分类方法和装置。
技术介绍
面对海量的新闻信息,目前主要是通过用户输入关键词搜索所关注的新闻,或通过页面推荐由用户选择所关注的新闻。为了提高搜索或者推荐用户所关注的新闻的效率和准确率,目前采用预先对新闻进行分类,以实现用户在其所关注的类目领域下进行搜索或在用户所关注的类目领域下进行推荐。然而,目前的新闻分类方法往往依赖于人工操作,费时费力,效率很低。
技术实现思路
有鉴于此,本申请提供一种文本分类方法和装置,以提高新闻分类的效率。技术方案如下:本申请的一方面提供一种文本分类方法,所述方法包括:提取目标文本中的关键词;基于提取的关键词,利用预先训练好的词向量模型,得到所述目标文本的词向量特征;计算所述目标文本的词向量特征,分别与预先设置的N级类目树中各级类目的词向量特征间的相似度值,N为正整数;将最大相似度值对应的类目确定为所述目标文本所属的类目。可选地,设置N级类目树包括:根据新闻内容的特点,先制定一级类目,再在所述一级类目下制定二级类目;再在所述二级类目下制定三级类目,以此类推,直至完成在N-1级类目下制定N级类目;其中,同一级类目中的各个类目所代表的领域不同。可选地,所述方法还包括:对所述N级类目树中的各级类目分别进行关键词标记;利用预先训练好的词向量模型,分别获取所述各级类目中各关键词的词向量特征;基于获取的所述各级类目中各关键词的词向量特征,得到所述各级类目的词向量特征。可选地,所述提取目标文本中的关键词之前,所述方法还包括:预处理所述目标文本;其中所述预处理所述目标文本至少包括:去除 ...
【技术保护点】
一种文本分类方法,其特征在于,所述方法包括:提取目标文本中的关键词;基于提取的关键词,利用预先训练好的词向量模型,得到所述目标文本的词向量特征;计算所述目标文本的词向量特征,分别与预先设置的N级类目树中各级类目的词向量特征间的相似度值,N为正整数;将最大相似度值对应的类目确定为所述目标文本所属的类目。
【技术特征摘要】
1.一种文本分类方法,其特征在于,所述方法包括:提取目标文本中的关键词;基于提取的关键词,利用预先训练好的词向量模型,得到所述目标文本的词向量特征;计算所述目标文本的词向量特征,分别与预先设置的N级类目树中各级类目的词向量特征间的相似度值,N为正整数;将最大相似度值对应的类目确定为所述目标文本所属的类目。2.根据权利要求1所述的方法,其特征在于,设置N级类目树包括:根据新闻内容的特点,先制定一级类目,再在所述一级类目下制定二级类目;再在所述二级类目下制定三级类目,以此类推,直至完成在N-1级类目下制定N级类目;其中,同一级类目中的各个类目所代表的领域不同。3.根据权利要求2所述的方法,其特征在于,所述方法还包括:对所述N级类目树中的各级类目分别进行关键词标记;利用预先训练好的词向量模型,分别获取所述各级类目中各关键词的词向量特征;基于获取的所述各级类目中各关键词的词向量特征,得到所述各级类目的词向量特征。4.根据权利要求1所述的方法,其特征在于,所述提取目标文本中的关键词之前,所述方法还包括:预处理所述目标文本;其中所述预处理所述目标文本至少包括:去除所述目标文本中对所述目标文本分类有影响的词语、去除所述目标文本中的url链接。5.根据权利要求1-4任一项所述的方法,其特征在于,所述计算所述目标文本的词向量特征,分别与预先设置的N级类目树中各级类目的词向量特征间的相似度值包括:利用余弦相似度算法,计算所述目标文本的词向量特征,分别与预先设置的N级类目树中各级类目的词向量特征间的相似度值。6.一种文本分类装置,其特征在于,所述装置包括:关键词提取单元,用于提...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。