iiihunter / writepoemswithlstm

using lstm to write poems automatically

WritePoems

若使用data目录数据，则使用.txt文件里的３w多唐诗训练lstm,全部诗词不区分类别；
若使用database目录里面的数据，则包含30W+首诗，以及2w+首宋词以及诗经等，数据同时标注有平仄、作者、类别等标签；
两种数据均可用来训练模型，数据读取函数分别在文件prepare_data与database/readdb中，详细可参照代码。

使用方法:

1. 预先安装python3.5 和　tensorflow
2. 运行python train.py，训练模型（若跳过此步骤直接使用训练好的文件可能会有不兼容问题，详细可自行尝试）
3. 运行write_poem函数每次运行自动生成一首诗（可能会有无法生成的情况产生，可多次运行）