基于BERT的自适应文本分类方法及装置

    公开(公告)号:CN113220892A

    公开(公告)日:2021-08-06

    申请号:CN202110658382.7

    申请日:2021-06-15

    申请人: 苏州大学

    IPC分类号: G06F16/35 G06N3/08

    摘要: 本申请涉及一种基于BERT的自适应文本分类方法及装置,属于中文信息处理技术领域,包括:获取语料样本数据并对语料样本数据进行预处理;构建预设网络模型;将所述预处理后的样本数据输入预设的网络模型,并使用预设的损失函数进行监督训练,得到分类模型;设置所述分类模型的输出阈值,得到设置后的分类模型,所述输出阈值控制分类结果的提前输出,所述设置后的分类模型用于对输入的文本进行分类。相较于传统的BERT模型,可在不损失精度的情况下,缩短模型推理时间。