基于RNN自动生成古诗

專 欄

❈

作者：yonggege，Python中文社区专栏作者

GitHub地址：https://github.com/wzyonggege

❈

0. char-rnn

关于RNN，LSTM，GRU的介绍已经有很多。而char-rnn是一个多层的RNN网络，本次基于经过Tensorflow改写的sherjilozair/char-rnn-tensorflow，输入大量的古诗，让机器学会自己创作。

1. 使用

环境

Python 3.6
Tensorflow 1.2.0

使用

帮助

命令行显示：

训练样本数据

生成古诗

上面的训练可能会花点时间，当然你也可以减少数据量去训练（GPU可以无视）。

训练完成之后：

即可生成古诗：

（可以选择选取多少个高频的汉字，若模型生成的不在选取的字典中，用‘*’代替）

南晓弦门络丹墀，
晚来兰槛酒盘弯。
故人无岁江水长，
两泪任身泪满缨。

（先不管内容了～～）

生成藏头诗

明年襟宠任，
月出画床帘。
别有平州伯性悔，
枝边折得李桑迷。
惊腰每异年三杰，
鹊出交钟玉笛频。

再来一首

明排东西落，
月浣绮罗纷。
别月鲜方淡，
枝枝胜鸟争。
惊传元羽节，
鹊堞吹桑衫。

2. 最后

代码代码：github.com/wzyonggege/RNN_poetry_generator （欢迎Star）