/AncientChinesePoemRNN

RNN for Ancient Chinese Poems

Primary LanguagePython

AncientChinesePoemRNN

参照char-rnn-tensorflow，使用RNN的字符模型，学习并生成古诗。 数据来自于http://www16.zzu.edu.cn/qts/ ,总共4万多首唐诗。

准备环境

  • tensorflow 1.0
  • python2

训练

python train.py 在使用GPU的情况下，两个小时内即可完成训练

生成

  • python sample.py rnn神经网络会生成一首全新的古诗。例如： ”帝以诚求备，堪留百勇杯。教官日与失，共恨五毛宣。鸡唇春疏叶，空衣滴舞衣。丑夫归晚里，此地几何人。”
  • python sample.py --prime <这里输入指定汉字> rnn神经网络会利用输入的汉字生成一首藏头诗。例如： python sample.py --prime 如花似月 会得到 “如尔残回号，花枝误晚声。似君星度上，月满二秋寒。”