本项目是基于50w小黄鸡对话语料构建的Transformer生成式单轮对话模型。本项目受启发于另外一个使用seq2seq模型构建的单轮对话模型。
目前模型的效果相对一般(具体效果如下图所示),待进一步完善...
│ .gitignore
│ chat.py
│ config.py
│ data_processing.py
│ LICENSE
│ README.md
│ requirements.txt
│ train.py
│ train_helper.ipynb
│ transformer.py
│ utils.py
│
├─data
│ vocab.pkl
│ xhj_data.tsv
│
├─saved_models
环境配置只需要两个步骤:
- 下载项目文件
- 安装相关的第三方库
具体步骤如下:
git@github.com:Duguce/ChatBotX.git && cd ChatBotX
pip install requirements.txt
- 生成词表
python data_processing.py
- 模型训练
python train.py
在进行模型训练之前,需要调整一下config.py
中的参数设置。
- 模型预测
训练好的模型会保存在saved_models
目录下。
- 体验交互式聊天
python chat.py
本仓库下的作品若无特殊说明均采用 GNU General Public License v3.0 开源协议进行许可