基于pytorch的聊天机器人项目

写一个闲聊式的聊天机器人，类似微软小冰

代码见 https://github.com/skyerhxx/Chatbot

 

环境

• pytorch1.3 
• python3.7
• cuda10.1

注意pytorch必须是1.3 ,  1.4的话会报错

 

主体模型是Seq2Seq+Attention

 

青云数据集

一个比较有意思的数据集

 

项目流程

 

项目架构

 

 

中文公开聊天语料库

https://blog.csdn.net/qq_28168421/article/details/88414914

 

 

数据预处理

pre_process.py

 

模型搭建

核心库

torch.nn

nltk

 

 

算法详解

 

模型训练

 

模型测试

 

 

 

 

demo

 

代码部署