pytorch实现聊天机器人(ChatBot)。

Song • 1590 次浏览 • 1 个回复 • 2018年03月05日

PyTorch不可思议的ChatBot

旨在建立一个奇妙的ChatBot

概要

这是ChatBot聊天机器人的第一个也是唯一的开源项目,我将继续亲自更新这个项目,旨在建立一个智能的ChatBot,作为Jarvis的下一个版本。

这个项目将保持基于PyTorch建立一个奇妙的ChatBot,欢迎给星并提交PR。 PyTorch不可思议的ChatBot

已经完成了

目前这个项目做了这些工作:

  • 基于官方教程,这个项目将继续开发一个seq2seq聊天机器人,QA系统;
  • 重新构建整个项目,分开的餐厅代码插入datamodeltrain logic;
  • 模型可以保存到本地,并从以前保存的目录中重新加载,这在官方教程中是缺少的;
  • 只需更换数据集即可训练您自己的数据! 最后但并非最不重要的一点,这个项目将在未来维持或转向其他项目协议,但我们将继续实施一个实用的基于seq2seq的项目,以构建您想要的任何内容:翻译机, 连天机器人,QA系统 ......任何你想要的。

环境需求

PyTorch
python3
Ubuntu任何版本
CPU和GPU均可

用法

在研究这个项目之前,您要浏览整个结构,我们有这些设置:

  • config:包含这个项目中的全局配置参数,你可以在这里改变一个全局参数;
  • datasets:包含数据和data_loader,使用你自己的数据集,你应该实现你自己的data_loader,但是这个只是一个简单的改变;
  • models:包含seq2seq模型定义;
  • utils:这个文件夹是非常有帮助的,它包含了一些可能帮助你摆脱烦人的事情的代码,比如保存模型,或者捕获KeyboardInterrupt异常或者加载之前的模型,都可以在这里完成。

训练模型也很简单,只需键入:

python3 train.py

项目地址:jinfagang/pytorch_chatbot


原创文章,转载请注明 :pytorch实现聊天机器人(ChatBot)。 - pytorch中文网
原文出处: https://ptorch.com/news/135.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • Song Song 2017-07-16 12:00:58 回复

    注意:
    在我测试的时候我发现这个项目好像不能正常运行和测试,我会持续关注,请大家等我的留言可以了再尝试!

Pytorch是什么?关于Pytorch! [ 源码讲解 ] pytorch通过Seq2Seq开发聊天机器人