/chatbot

a solution for chatbot

Primary LanguageJupyter NotebookMIT LicenseMIT

chatbot

a solution for chatbot

dataset

  1. 数据来自武侠小说中的对话,目前只是抽取了其中带有问答的上下句。对话数量为774142,其中上句的最大长度118, 下句的最大长度98
  2. 可以通过***test_data.ipynb***查看

model

  1. 采用***bert encoder + seq2seq decoder***
  2. 可以通过***test_model.ipynb***查看

parameters

  1. 通过***params.ini***设置

pretrained bert model

  1. 下载bert官方的中文模型,解压,并在***params.ini***中配置其路径

训练效果

如图是运行了不到一个epoch的效果,持续更新中。。。