seq2seq_chatbot
seq2seq_chatbot copied to clipboard
使用tensorflow构建的seq2seq+attention+beansearch 的聊天机器人模型
seq2seq_chatbot
环境依赖
| 程序 | 版本 |
|---|---|
| python | 3.52 |
| Tensorflow | 1.10.0 |
| CUDA | 9.0.103 |
| cuDNN | 7.0 |
Run
-
预料获取
-
wget https://lvzhe.oss-cn-beijing.aliyuncs.com/dgk_shooter_min.conv.zip输出 :dgk_shooter_min.conv.zip
-
unzip dgk_shooter_min.conv.zip输出 : dgk_shooter_min.conv
-
-
文本处理
-
python data_process.py输出: chatbot.pkl
-
训练数据
-
python train_anti.py在./model 得到训练好的模型
-
-
chatbot测试
python test_anti.py
-

有问题欢迎在Issues留言 ,如果觉得不错请给个star !