chatbot_by_similarity

根据文本相似度实现问答的聊天机器人（弱智版）

项目介绍

这是根据工作需求写的一个简易版本的聊天机器人，主要目的是根据问题从知识库中匹配相应的答案，从而帮助使用者去更方便的查询到一些知识性内容。

模块简介

用法比较简单，给文本列表，经过训练后去匹配问题返回相似的答案。

结构及用法

模块由五个部分构成：测试数据、预处理分词、计算词向量、计算相似度和聊天机器人

文本的预处理(cut_text.py)：
提供知识库的word文档数据和小黄鸡聊天记录

from chatbot import load_data

# word数据
texts = load_data(type='knowledge')

# 小黄鸡数据
texts = load_data(type='chat')

文本的预处理(cut_text.py)：
用于分词、剔除停用词(这里偷懒直接把长度为1的剔除)

texts = ['我爱北京***', '我爱北京长城']
results = cut_texts(texts=texts, need_cut=True, word_len=2)

results: [['北京', '***'], ['北京', '长城']]

文本转向量(text2vec.py)：
通过word2vec计算词向量，然后求均值转文本向量，空闲的话我会考虑结合tf-idf计算权重优化向量加权方法

texts_cut = [['北京', '***'], ['北京', '长城']]
model_word2vec = creat_dict(texts_cut=texts_cut,
                            sg=1,
                            size=4,
                            window=5,
                            min_count=1)
texts_vec = text2vec(texts_cut=texts_cut,
                     model_word2vec=model_word2vec,
                     merge=True)

texts_vec:
 [[ 0.05730793  0.01469728  0.03473849  0.04489793]
 [-0.01690295  0.12111638 -0.01626978  0.02280862]]

计算相似度(cal_similarity.py)：
目前只写了余弦值，余弦修正、和欧氏距离，有机会的话会考虑通过监督学习计算相似度

x = np.array([1, 1])
y = np.array([1, 2])
cos = cal_similarity(x, y, mode='cos')
Euclidean = cal_similarity(x, y, mode='Euclidean')

cos: 0.9486832980505138
Euclidean: 1.0

聊天机器人的训练与使用(chatbot.py)：
整合前三个步骤，计算问题和知识库每一条知识的相似度，返回排名靠前的知识。实际使用中只需要调用chatbot.py的功能即可，具体参考demo_train.py、demo_ask&answer.py，可以调整前三个步骤的算法实现优化。我的训练语料是一些word文档，model_word_document.pkl是训练好的一个简单模型。
train：训练语料库，输入 texts=文本列表、mode=模式
get_answer：获取答案，参数 ask=问题、sample=抽样匹配数、mode=相似度计算方法、modify=是否进行余弦修正、threshold=相似度阈值、topn=返回知识数量、process_num=进程数
数据量大循环计算相似度会很慢，写了多进程，不知道为什么用queue存储会堵塞，就改成每个进程保存计算结果
数据量不大的时候IO更耗时，建议process_num=1就好
数据量大的时候计算更耗时，建议根据cpu数量开多进程会快很多

from chatbot import chatbot

texts = ['我爱北京***', '我爱北京长城']
chatbot_try = chatbot()
chatbot_try.train(texts=texts)
answer=chatbot_try.get_answer(ask=ask, sample=50000, mode='cos', modify=False, threshold=0, topn=5, process_num=10)

知识库的问答

闲聊

lxz1992 / chatbot_by_similarity Goto Github PK