python小黄鸡nlp聊天机器人

babyai997

已于 2022-04-07 18:47:27 修改

阅读量2.3k

点赞数 1

分类专栏： nlp 文章标签： python

于 2022-04-07 18:45:26 首次发布

本文链接：https://blog.csdn.net/babyai996/article/details/124023582

版权

nlp 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该博客介绍了一个使用Python实现的基于Seq2Seq模型的聊天机器人，利用TensorFlow进行训练和预测。机器人通过DataProcessing模块处理输入，允许GPU显存增长，并通过Tornado Web服务器提供API接口。提供了下载链接，适用于进行自然语言处理的对话交互。

摘要由CSDN通过智能技术生成

python小黄鸡nlp聊天机器人

# encoding: utf-8

import os

import tensorflow as tf
import numpy as np

from SequenceToSequence import Seq2Seq
import DataProcessing
from CONFIG import BASE_MODEL_DIR, MODEL_NAME, data_config, model_config

import tornado.web
import tornado.ioloop
from tornado.web import RequestHandler


def chatbot_api(infos):
    du = DataProcessing.DataUnit(**data_config)
    save_path = os.path.join(BASE_MODEL_DIR, MODEL_NAME)
    batch_size = 1
    tf.reset_default_graph()
    model = Seq2Seq(batch_size=batch_size,
                    encoder_vocab_size=du.vocab_size,
                    decoder_vocab_size=du.vocab_size,
                    mode='decode',
                    **model_config)
    # 创建session的时候允许显存增长
    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True

    with tf.Session(config=config) as sess:
        init = tf.global_variables_initializer()
        sess.run(init)
        model.load(sess, save_path)
        while True:
            q = infos
            if q is None or q.strip() == '':
                return "请输入聊天信息"
                continue
            q = q.strip()
            indexs = du.transform_sentence(q) # 把句子转换为索引
            x = np.asarray(indexs).reshape((1, -1))  # 转为1行
            xl = np.asarray(len(indexs)).reshape((1,))
            pred = model.predict(
                sess, np.array(x),
                np.array(xl)
            )
            result = du.transform_indexs(pred[0])  # 将索引转为句子
            return result


class BaseHandler(RequestHandler):
    """解决JS跨域请求问题"""

    def set_default_headers(self):
        self.set_header('Access-Control-Allow-Origin', '*')
        self.set_header('Access-Control-Allow-Methods', 'POST, GET')
        self.set_header('Access-Control-Max-Age', 1000)
        self.set_header('Access-Control-Allow-Headers', '*')
        # self.set_header('Content-type', 'application/json')


class IndexHandler(BaseHandler):
    # 添加一个处理get请求方式的方法
    def get(self):
        # 向响应中，添加数据
        infos = self.get_query_argument("infos")
        print("Q:", infos)
        # 捕捉服务器异常信息
        #try:
        result = chatbot_api(infos=infos)
        #except:
        #    result = "服务器内部错误"
        print("A:", "".join(result))
        self.write("".join(result))


if __name__ == '__main__':
    # 创建一个应用对象
    app = tornado.web.Application([(r'/api/chatbot', IndexHandler)])
    # 绑定一个监听端口
    app.listen(8000)
    # 启动web程序，开始监听端口的连接
    tornado.ioloop.IOLoop.current().start()

下载链接：

https://download.csdn.net/download/babyai996/85090271