TensorFlow：基于RNN生成古诗词

时间：2017-07-26 发布人：SHX 浏览次数：6449 评论：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

发表评论

技术宅