扩展JointBERT支持中文训练, 提供从数据合成到意图和槽位联合训练, 测试完整流程.

这里提供训练好的模型52AI/generalQA_intent_slotFilling 供测试. 国内下载容易中断,多运行两次.

$ python3 predict.py --task generalQA \
                     --input_file data/testcase/generalQAtest.txt \
                     --output_file local/generalQAtest_predict.txt \
                     --model_dir out/generalQA
-> 请 问 [你:B-TransEnZhSentence] [几:I-TransEnZhSentence] [岁:I-TransEnZhSentence] [了:I-TransEnZhSentence] 用 英 语 怎 么 说 ?

-> 翻 译 : [i:B-TransEnZhSentence] [love:I-TransEnZhSentence] [you:I-TransEnZhSentence]

-> 用 [美:B-CreateSenEntity] [好:I-CreateSenEntity] 写 一 个 句 子

-> [明:B-AntonymEntity] [天:I-AntonymEntity] 的 反 义 词

-> [后:B-SynonymEntity] [天:I-SynonymEntity] 的 同 义 词

测试结果: local/generalQAtest_predict.txt

项目地址: JointBERT-zh

Downloads last month
6
Inference API
Unable to determine this model’s pipeline type. Check the docs .