[发明专利]内容分类方法、装置、计算机设备和存储介质

说明书

本申请涉及一种内容分类方法、装置、计算机设备和存储介质，方法包括：获取待分类的目标内容对应的目标特征向量；获取已训练得到的目标分类模型，目标分类模型包括第一分类模型以及第二分类模型；将目标特征向量输入到第一分类模型中，得到目标内容对应的第一内容类别，第一内容类别为第一分类层级对应的内容类别；获取第一分类层级对应的第一类别特征信息；将第一类别特征信息以及目标特征向量输入到第二分类模型中，得到目标内容对应的第二内容类别，第二内容类别为第二分类层级对应的内容类别，第二分类层级的级别低于第一分类层级的级别；将第一内容类别以及第二内容类别作为目标内容对应的分类结果。上述方法可以提高内容分类准确度。

技术领域

本申请涉及信息处理领域，特别是涉及内容分类方法、装置、计算机设备和存储介质。

背景技术

随着互联网技术的发展，互联网上的内容越来越多，在许多场景下都会存在对内容分类的需求。例如需要根据内容的类型对内容进行管理，根据用户选择的内容类别推送对应类别的内容。如对视频进行分类，以得到视频的类别，视频的类别例如可以是喜剧类或者剧情类。可以通过人工智能模型对内容进行分类，但目前分类得到的内容类别精细度以及准确度差。

发明内容

基于此，有必要针对上述分类得到的内容类别精细度以及准确度差的问题，提供一种内容分类方法、装置、计算机设备和存储介质。

一种内容分类方法，所述方法包括：获取待分类的目标内容对应的目标特征向量；获取已训练得到的目标分类模型，所述目标分类模型包括第一分类模型以及第二分类模型；将所述目标特征向量输入到所述第一分类模型中，得到所述目标内容对应的第一内容类别，所述第一内容类别为第一分类层级对应的内容类别；获取所述第一分类层级对应的第一类别特征信息；将所述第一类别特征信息以及所述目标特征向量输入到所述第二分类模型中，得到所述目标内容对应的第二内容类别，所述第二内容类别为第二分类层级对应的内容类别，所述第二分类层级的级别低于所述第一分类层级的级别；将所述第一内容类别以及所述第二内容类别作为所述目标内容对应的分类结果。

一种内容分类装置，所述装置包括：目标特征向量获取模块，用于获取待分类的目标内容对应的目标特征向量；分类模型获取模块，用于获取已训练得到的目标分类模型，所述目标分类模型包括第一分类模型以及第二分类模型；第一内容类别得到模块，用于将所述目标特征向量输入到所述第一分类模型中，得到所述目标内容对应的第一内容类别，所述第一内容类别为第一分类层级对应的内容类别；第一类别特征信息获取模块，用于获取所述第一分类层级对应的第一类别特征信息；第二内容类别得到模块，用于将所述第一类别特征信息以及所述目标特征向量输入到所述第二分类模型中，得到所述目标内容对应的第二内容类别，所述第二内容类别为第二分类层级对应的内容类别，所述第二分类层级的级别低于所述第一分类层级的级别；分类结果得到模块，用于将所述第一内容类别以及所述第二内容类别作为所述目标内容对应的分类结果。

在一些实施例中，所述第一类别特征信息获取模块包括：类别向量得到单元，用于获取所述第一分类层级中各个候选内容类别对应的类别向量；类别权重系数获取单元，用于获取各个所述候选内容类别对应的类别权重系数；第一类别特征信息得到单元，用于根据所述候选内容类别对应的类别向量以及对应的类别权重系数计算得到所述第一分类层级对应的第一类别特征信息。

在一些实施例中，所述第一内容类别得到模块用于：将所述目标特征向量输入到所述第一分类模型中，得到各个所述候选内容类别对应的被选中率；根据各个所述候选内容类别对应的被选中率确定所述目标内容对应的第一内容类别；所述类别权重系数获取单元用于：根据所述第一分类模型输出的所述候选内容类别对应的被选中率，得到所述候选内容类别对应的类别权重系数。

在一些实施例中，所述类别权重系数获取单元用于：计算所述候选内容类别对应的类别向量与所述目标内容的第一特征向量的相关度；根据所述相关度得到所述候选内容类别对应的类别权重系数。