AI通过数据构建一个独有对话机器人,尝试构建快速构建专有知识的机器人。
前端使用tinker实现一个简单的对话窗口,
后端使用自己的数据进行不断的训练,有需要的可以依据自己的实际情况进行修改,和优化
import tkinter as tk
from tkinter import scrolledtext
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity
import jieba
import string, psycopg2
# 连接PostgreSQL数据库
conn = psycopg2.connect(database="postgres", user="postgres", password="postgres", host="localhost", port="5433")
cur = conn.cursor()
cur.execute("SELECT questions, answers FROM public.stock_chatbot")
data_db = cur.fetchall()
conn.close()
# 准备数据
questions = [q for q, _ in data_db]
answers = [a for _, a in data_db]