在双句文本分类中,我们提供了多种模型进行选择。以下是模型的说明,您可以根据自己的具体场景,选择一个更适合的模型。
双句分类-高性能版:基于自研的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% |
文档内容是否对您有帮助?