您正在使用IE低版浏览器,为了您的雷峰网账号安全和更好的产品体验,强烈建议使用更快更安全的浏览器
此为临时链接,仅用于文章预览,将在时失效
人工智能开发者 正文
发私信给AI研习社
发送

1

我用 tensorflow 实现的 “一个神经聊天模型”

本文作者:AI研习社 编辑:贾智龙 2017-09-07 10:26
导语:一个基于深度学习的聊天机器人。

雷锋网按:本文原作者小灰灰,原载于专栏。雷锋网已获得作者授权。

概述

这个工作尝试重现这个论文的结果 A Neural Conversational Model (aka the Google chatbot).
它使用了循环神经网络(seq2seq 模型)来进行句子预测。它是用 python 和 TensorFlow 开发。

程序的加载主体部分是参考 Torch 的  neuralconvo from macournoyer.

现在, DeepQA 支持一下对话语料:

To speedup the training, it's also possible to use pre-trained word embeddings (thanks to Eschnou). More info here.

安装

这个程序需要一下依赖 (easy to install using pip: pip3 install -r requirements.txt):

  • python 3.5

  • tensorflow (tested with v1.0)

  • numpy

  • CUDA (for using GPU)

  • nltk (natural language toolkit for tokenized the sentences)

  • tqdm (for the nice progression bars)

你可能需要下载附带的数据让 nltk 正常工作。

python3 -m nltk.downloader punkt

Cornell 数据集已经包括了。其他的数据集查看 readme 文件到他们所在的文件夹。  (在  data/).

网站接口需要一些附加的包:

  • django (tested with 1.10)

  • channels

  • Redis (see here)

  • asgi_redis (at least 1.0)

Docker 安装也是支持的,更多详细的教程参考  here.

运行

聊天机器人

训练这个模型,直接运行 main.py 。一旦训练完成,你可以测试结果用 main.py --test
(结果生成在  'save/model/samples_predictions.txt') 或者用 main.py --test interactive (更有趣).

Here are some flags which could be useful. For more help and options, use python main.py -h:

  • --modelTag: allow to give a name to the current model to differentiate between them when testing/training.

  • --keepAll: use this flag when training if when testing, you want to see the predictions at different steps (it can be interesting to see the program changes its name and age as the training progress). Warning: It can quickly take a lot of storage space if you don't increase the --saveEvery option.

  • --filterVocab 20 or --vocabularySize 30000: Limit the vocabulary size to and optimize the performances and memory usage. Replace the words used less than 20 times by thetoken and set a maximum vocabulary size.

  • --verbose: when testing, will print the sentences as they are computed.

  • --playDataset: show some dialogue samples from the dataset (can be use conjointly with --createDataset if this is the only action you want to perform).

To visualize the computational graph and the cost with TensorBoard, just run tensorboard --logdir save/.

默认的这个网络架构是一个标准的 encoder/decoder 有两个 LSTM layers (隐藏层大小 256) ,然后 vocabulary 的 embedding size 是  32. 这个网络用 ADAM 训练。最大的句子长度设置为 10 个单词,但是可以增加。

结果

当然,这个网络并不会很擅长聊天:

Q: Hi
A: Hi.

Q: What is your name ?
A: Laura.

Q: What does that mean ?
A: I dunno.

Q: How old are you ?
A: thirty-five.

Q: Will Google hire me ?
A: No.

Q: Tell me the alphabet
A: Fuck you.

Q: That's not nice
A: Yeah.

这里有一些情况它并不能正确回答:

Q: Who is Laura ?
A: My brother.

Q: Say 'goodbye'
A: Alright.

Q: What is cooking ?
A: A channel.

Q: Can you say no ?
A: No.

Q: Two plus two
A: Manny...

预训练模型

项目截图

我用 tensorflow 实现的 “一个神经聊天模型”

chatbot_miniature.png

实测截图

我用 tensorflow 实现的 “一个神经聊天模型”

Screenshot from 2017-09-05 14-47-52.png

一步一步教程

1. 下载这个项目:
https://github.com/Conchylicultor/DeepQA
2. 下载训练好的模型:
https://drive.google.com/file/d/0Bw-phsNSkq23OXRFTkNqN0JGUU0/view
(如果网址不能打开的话,今晚我会上传到百度网盘,分享到:http://www.tensorflownews.com/
3. 解压之后放在 项目 save 目录下,如图所示

我用 tensorflow 实现的 “一个神经聊天模型”

Screenshot from 2017-09-05 14-52-13.png

4. 复制 save/model-pretrainedv2/dataset-cornell-old-lenght10-filter0-vocabSize0.pkl 这个文件到  data/samples/,如图所示:

我用 tensorflow 实现的 “一个神经聊天模型”

Screenshot from 2017-09-05 14-55-00.png

5. 在项目目录执行一下命令:

python3 main.py --modelTag pretrainedv2 --test interactive

程序读取了预训练的模型之后,如图:

我用 tensorflow 实现的 “一个神经聊天模型”

Screenshot from 2017-09-05 14-57-14.png

聊天机器人资源合集

项目,语聊,论文,教程
https://github.com/fendouai/Awesome-Chatbot

DeepQA

https://github.com/Conchylicultor/DeepQA

备注:为了更加容易了解这个项目,说明部分翻译了项目的部分 readme ,主要是介绍使用预处理数据来运行这个项目。

雷峰网版权文章,未经授权禁止转载。详情见转载须知

我用 tensorflow 实现的 “一个神经聊天模型”

分享:
相关文章

编辑

聚焦数据科学,连接 AI 开发者。更多精彩内容,请访问:yanxishe.com
当月热门文章
最新文章
请填写申请人资料
姓名
电话
邮箱
微信号
作品链接
个人简介
为了您的账户安全,请验证邮箱
您的邮箱还未验证,完成可获20积分哟!
请验证您的邮箱
立即验证
完善账号信息
您的账号已经绑定,现在您可以设置密码以方便用邮箱登录
立即设置 以后再说