Python聊天机器人实现代码【NLTK】

ezpod 2019-11-15

python 函数 Image 聊天机器人 机器人 chatbot nltk

聊天机器人(Chatbot)是一种人工智能软件,利用它你可以通过网站、手机App或电话等途径和用户进行自然语言对话。聊天机器人可以在不同的行业中应用于不同的场景。NLTK是进行自然语言处理(NLP)的领先的Python开发包 — 另一个常用的NLP开发包是Spacy — 在这个教程中,我们将使用NLTK开发库创建一个简单的聊天机器人。

1、导入开发包及数据

首先导入必要的开发包:

在这里插入图片描述

然后将数据集导入Pandas数据帧:

在这里插入图片描述

上面的数据包含1592个样本,每个样本为两个字段,分别表示查询及响应文本。可以看到数据集里有空值,是因为样本数据是分组的,每组的不同查询文本都对应相同的相应文本。我们可以使用ffill()进行处理:

在这里插入图片描述

2、处理步骤

在这里插入图片描述
首先执行第一步,文本规范化处理。我们将所有的数据转化为小写,删除特殊符号,提取词干。这部分代码我们使用函数step1进行封装:

在这里插入图片描述
可以看到文本已经干净多了。分词指的是将文本字符串切分为词条:

在这里插入图片描述

pos_tag函数返回每个词条在文本中的作用:

在这里插入图片描述

我们现在将创建一个函数来整合上面这些环节的代码:

在这里插入图片描述

让我们用这个函数处理数据集:

在这里插入图片描述

下一步是单词嵌入表示,两次词义相近的词,其嵌入表示的距离也相近。有两种模型可以用于这个处理环节:词袋模型和tf-idf模型。

3、词袋模型

词袋模型是描述文本中出现的单词的一种表示方法。例如,假设我们的词典中包含单词{Playing, is, love},我们希望矢量化文本“Playing football is love”,那么得到的矢量就是: (1, 0, 1, 1)。

在这里插入图片描述

上图表格中的第一行,显示了数据集的第一个样本的BOW模型,只有1和0。

停止词指的是那些经常出现的单词,因此这些单词对于特定的文本来说意义就不大,我们可以把这些单词从词典中排除出去。下面是预定义的停止词:

在这里插入图片描述
考虑下面的示例,我们尝试获取查询对应的响应:

在这里插入图片描述

上面的代码中,我们可以看到对于查询‘Will you help me and tell me about yourself more’ ,我们进行文本规范化处理然后转化为词袋表示。下面我们将使用余弦相似算法来找出相关的相应文本。

4、余弦相似性

余弦相似性是衡量两个矢量相似性的一种指标。其计算方法是用两个矢量的点积除以两个矢量的模的乘积:

Cosine Similarity (a, b) = Dot product(a, b) / ||a|| * ||b||

计算代码如下:

在这里插入图片描述
可以看到194#样本的相似性最高,让我们输出其查询文本看看是否相关:

在这里插入图片描述
的确是相关的!

5、TF-IDF模型

tf指的是词频,表示单词在当前文档中出现的频率,idf指的是逆文档频率,表示单词在文档集中出现的频率倒数。这里我们说的文档表示一个样本,文档集表示所有的样本。

在这里插入图片描述

上面是使用tf-idf处理得到的值。现在使用余弦相似算法来找出相关的响应。

在这里插入图片描述

4#样本的相似度最高,让我们显示出来看一下:

在这里插入图片描述

使用tf-idf我们得到一个不同的响应,不过看起来也很好!

现在让我们组织一下代码:

在这里插入图片描述

看看其他响应:

在这里插入图片描述

Coooooooooooool!

6、总结

我们也可以实现词袋版本的对话。我们创建的模型没有使用任何人工智能,但是效果还是不错的。完整的代码可以在这里获取。


原文链接:NLTK聊天机器人实现原理及代码下载 - 汇智网 ,转载请标明出处

登录 后评论
下一篇
云栖号资讯小编
1195人浏览
2020-06-01
相关推荐
GitHub值得关注记录
1035人浏览
2017-12-25 22:20:00
0
0
0
1097