模型说明

在双句文本分类中,我们提供了多种模型进行选择。以下是模型的说明,您可以根据自己的具体场景,选择一个更适合的模型。

  • 双句分类-高性能版:基于自研的bilstm+cross attention架构。速度快,计算资源要求低,适合对性能要求高的场景。

  • 双句分类-高精度版:基于StructBERT双句分类架构。精度较高,适合对精度要求高,性能要求低的场景。

  • 性能对比(仅供参考,不同数据集可能差异较大)

模型

数据集

数据量

训练参数(默认参数)

训练耗时(gpu)

预测耗时(cpu)

准确率

双句分类-高性能版

CLUE-tnews

5.3w

epoch=20

16min

150ms

63.67%

双句分类-高精度版

CLUE-tnews

5.3w

epoch=4

46min

500ms

66.65%