pytorch聊天机器人
使用PyTorch的Seq2Seq聊天机器人实现功能:Seq2Seq +光束搜索+ antiLM
要求
Python3
火炬0.3
语料库
用法
训练
python train . py
测试
python console python console . py . / ckpt model
光束搜索示例:
me: hi .
Bot: how can i help you ? score:-0.66
Bot: where are you going to go ? score:-0.66
Bot: i am sorry to hear that . what can i do for you ? score:-0.67
Bot: where are you going ? score:-0.68
Bot: how are you goi
1