Chatbot

聊天机器人PyTorch实现

数据集

小黄鸡语料

运行步骤

训练模型，模型将保存在save文件夹下
```
python main.py --mode train
```
聊天
```
python main.py --mode chat
```