代码语言：python

所属分类：人工智能

代码描述：python基于nlpia及keras通过LSTM实现智能机器人对话代码，通过对电影中的对话的数据的训练，实现了基于LSTM模型的智能机器人对话代码。

代码标签： python nlpia keras LSTM 智能 机器人 对话

下面为部分代码预览，完整代码请点击下载或在bfwstudio webide中打开

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import tensorflow as tf
from  nlpia.loaders import  get_data
from  tensorflow.python.keras.models import  Model,load_model
from  tensorflow.python.keras.layers import  Input,LSTM,Dense
import os

import numpy as np
#为训练准备语料库
df=get_data('moviedialog')
input_texts,target_texts=[],[]#数组保存从语料文库中读取输入文本和目标文本
input_vocabulary=set()#这个集合保存输入文本和目标文本中出现的字符
output_vocabulary=set()
start_token='\t'  #目标序列用startstop词条进行注释
stop_token='\n'
max_training_samples=min(25000,len(df)-1) #定义了训练使用的行数
for input_text,target_text in zip(df.statement,df.reply):
    target_text=start_token+target_text\
            +stop_token
    input_texts.append(input_text)
    target_texts.append(target_text)
    for char in input_text:
        if char not in input_vocabulary:
            input_vocabulary.add(char)
    for char in target_text:
        if char not in output_vocabulary:
            output_vocabul.........完整代码请登录后点击上方下载按钮下载查看

网友评论0

取消

评论提交中...

提交评论