Skip to content

Latest commit

 

History

History
31 lines (25 loc) · 1.2 KB

README.md

File metadata and controls

31 lines (25 loc) · 1.2 KB

bd-chat-2020

2020语言与智能技术竞赛:面向推荐的对话任务
第二名 强行跳大 团队

介绍

  • 最终只使用了集成Goal预测、文本回复的单向注意力Bert模型。此项目还附带了一些抛弃的尝试方案,包含了基础的阅读理解模型(model/model_rc.py)、分类模型(model/model_context.py,model/model_goal.py)、QA召回(检索)模型(model/model_recall.py)的实现。
  • 训练的数据存放位置和输出位置可以参考 cfg.py。训练的顺序是先运行 data_deal/pre_trans.py,再运行train/train_bert_lm.py。
  • 预测test集的代码是code/predict/predict_lm.py,人工评估阶段使用的是code/predict/predict_final.py里的预测

数据

https://aistudio.baidu.com/aistudio/competition/detail/48 里面的推荐对话任务数据即是,格式一致

依赖

keras==2.3.1
tensorflow-gpu==1.14.0
easydict
nltk==3.5

文件介绍:

  • conversation_client.py: 类似官方的那种client
  • test_client.py: 交互式的client
  • conversation_server.py: 类似官方的服务端。

参考

注意

  • 不要覆盖 data/roberta 里的 vocab.txt 和 config