我正在使用 keras 和 WebSockets 实现一个简单的聊天机器人。我现在有了一个模型,可以对用户输入进行预测并发送相应的答案。
当我通过命令行执行此操作时,它工作正常,但是当我尝试通过 WebSocket 发送答案时,WebSocket 甚至不再启动。
这是我的工作 WebSocket 代码:
@sock.route('/api')
def echo(sock):
while True:
# get user input from browser
user_input = sock.receive()
# print user input on console
print(user_input)
# read answer from console
response = input()
# send response to browser
sock.send(response)
Run Code Online (Sandbox Code Playgroud)
这是我在命令行上与 keras 模型通信的代码:
while True:
question = input("")
ints = predict(question)
answer = response(ints, json_data)
print(answer)
Run Code Online (Sandbox Code Playgroud)
使用的方法有:
def predict(sentence):
bag_of_words = convert_sentence_in_bag_of_words(sentence)
# pass bag as list and get index 0
prediction = …Run Code Online (Sandbox Code Playgroud)