本项目是基于 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