扩展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