用chatterbot从零开始搭建一个聊天机器人（二）

wenknow
wenknow
2020-09-03 03:34:19 字数 2929 阅读 585

一、获取语料

  • 中文聊天语料这一块，我选择了网上大神整理的资料，选取其中一部分，小黄鸡语料
    公开中文语料包：https://github.com/codemayq/chinese_chatbot_corpus

二、训练

1、上传语料包

  • 上传处理好的小黄鸡语料到服务器上，有条件的可以使用 Google colab进行训练

2、在服务器上安装 ChatterBot

pip3 install chatterbot
pip3 install chatterbot-corpus

3、试运行chatterbot

  • 运行以下代码将会自动安装nltk_data，此过程可能会很久
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer

chatbot = ChatBot("bot")
trainer = ListTrainer(chatbot)

4、修改chatterbot中nltk_data的目录

  • 找到python的第三方包site-packages的路径下的chatterbot
cd /usr/local/lib/python3.6/site-packages/chatterbot/
vim utils.py
  • 修改内容如下
def download_nltk_stopwords():

"""

Download required NLTK stopwords corpus if it has not already been downloaded.

"""

nltk_download_corpus('corpora/stopwords')

def download_nltk_wordnet():

"""

Download required NLTK corpora if they have not already been downloaded.

"""

nltk_download_corpus('corpora/wordnet')

def download_nltk_averaged_perceptron_tagger():

"""

Download the NLTK averaged perceptron tagger that is required for this algorithm

to run only if the corpora has not already been downloaded.

"""

nltk_download_corpus('taggers/averaged_perceptron_tagger')

def download_nltk_vader_lexicon():

"""

Download the NLTK vader lexicon for sentiment analysis

that is required for this algorithm to run.

"""
nltk_download_corpus('sentiment/vader_lexicon')

5、创建一个 Chat Bot并进行训练

from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer

chatbot = ChatBot("小明")
trainer = ListTrainer(chatbot)
with open('part.data', encoding='utf-8') as f:
data = f.read().replace('\t', '\n')
data = data.split("\n")

trainer.train(data)

6、本地测试

from chatterbot import ChatBot
import sys

bot = ChatBot(
    '小明',
    database_uri='sqlite:///db.sqlite3'
 )
 
print('Type something to begin...')
 
while True:
    try:
        user_input = input()

        bot_response = bot.get_response(user_input)

        print(bot_response)

    # Press ctrl-c or ctrl-d on the keyboard to exit
    except (KeyboardInterrupt, EOFError, SystemExit):
        break

三、部署成服务

  • 安装flask
pip3 install flask
  • 安装uwsgi
yum install -y pcre pcre-devel pcre-static
yum install -y python3-devel
pip3 install uwsgi --no-cache-dir
  • 新建api.py文件
vim api.py
  • 添加如下内容
from flask import Flask, render_template, request, jsonify
from chatterbot import ChatBot
 
app = Flask(__name__)
 
bot = ChatBot(
    'С˼',
    database_uri='sqlite:///db.sqlite3'
)

@app.route("/")
def home():
	return render_template("index.html")

@app.route("/api/<text>")
def get_bot_api(text):
    res = str(bot.get_response(text))
    return jsonify(res), 200


if __name__ == "__main__":
	app.run(host='0.0.0.0')
  • 新建在项目目录下，添加uwsgi配置
vim uwsgi.ini
  • 添加如下内容
[uwsgi]

http = 0.0.0.0:5000
chdir = /usr/share/nginx/html/chatbot/chatterbot
wsgi-file = api.py
callable = app
processes = 4
threads = 2
master = true
vacuum = true
  • 运行uwsgi
uwsgi uwsgi.ini
  • 调用
$url = "http://127.0.0.1:5000/api/{$word}";
$reply = $this->getData($url);

附：源码

https://github.com/Wc241/chatterbot

评论