TensorFlow:基于RNN生成古诗词

时间:2017-07-26  发布人:SHX  浏览次数:2778  评论:0

实例来源于网络,但是以前的代码都是基于TensorFlow1.0以前的版本写的,实际运行时会报错,对于小白来说是一头雾水。这里的代码经过改进能在

python 3.5 + ubuntu 16.04 + TensorFlow 1.0以上 正常运行

使用的数据集:全唐诗(43030首):https://pan.baidu.com/s/1o7QlUhO

源码下载:http://pan.baidu.com/s/1jI5ONcy  密码:ka19  文件:RNN.zip 

train.py  ---  训练文件 可训练出模型保存在 save 目录中 (我的主机配置比较低当时跑了一个星期)

poetry.txt  ---- 训练用的唐诗

test.txt --- 测试文件 需要有训练出的模型才能运行出结果

去掉其中注释就可以运行(不要两项一起运行,会计算比较慢不会出结果)


网上其他版本报错原因(我出现的错误):

1.0修改了很多地方,错误可取官网搜索。

tf.nn.seq2seq.sequence_loss_by_example 改成 tf.contrib.legacy_seq2seq.sequence_loss_by_example

tf.nn.rnn_cell. 改成 tf.contrib.rnn.

initialize_all_variables 改成 global_variables_initializer

all_variables 改成 global_variables


返回

上一篇:人工智能之谷歌开源框架TensorFlow下一篇:人工智能ColorHandPose3D手指识别技术

我要评论

    loading
 

关闭

loading

回顶部