chatbot startkit

TensorFlow 1.4 中的聊天机器人 Startkit

有关如何使用tensorflow 1.4版本在python中使用seq2seq模型实现简单聊天机器人的示例。这个聊天机器人示例还展示了注意力机制和分桶。

数据集

我在这个例子中使用了康奈尔电影对话语料库。您可以下载它：这里

安装

支持的Python版本

- 本项目使用的Python版本：3.5+

使用的库

• 熊猫0.18.0
• numpy 1.10.4
• TensorFlow 1.4.0

代码

核心 seq2seq 模型函数都在model_utils.py中。

数据预处理和 NLP 函数位于cornell_data_utils.py内部。

如果您想使用模型超参数，请使用config.py 。

跑步

要运行此项目，您将需要一些软件，例如 Anaconda，它提供对运行 .ipynb 文件（Jupyter Notebook）的支持。

确保您拥有之后，您可以从终端或 cmd 运行下一行：

ipython notebook chatbot.ipynb

或者

jupyter notebook chatbot.ipynb

执照

信息技术许可证

版权所有 (c) 2017 卢卡·阿尼辛

特此免费授予获得本软件和相关文档文件（“软件”）副本的任何人不受限制地使用本软件，包括但不限于使用、复制、修改、合并的权利、发布、分发、再许可和/或销售软件的副本，并允许向其提供软件的人员这样做，但须满足以下条件：

上述版权声明和本许可声明应包含在本软件的所有副本或主要部分中。

本软件按“原样”提供，不提供任何明示或暗示的保证，包括但不限于适销性、特定用途的适用性和不侵权的保证。在任何情况下，作者或版权持有者均不对因本软件或本软件中的使用或其他交易而产生或与之相关的任何索赔、损害或其他责任负责，无论是合同、侵权行为还是其他行为。软件。