LSTM_model.py:LSTM网络模型,提供了end_points接口,被其他部分调用
poetry_porcess.py:数据读取、预处理部分,会返回打包好的batch,被main调用
gen_poetry.py:古诗生成程序,拥有可选的风格参数,被main调用
main.py:主函数,既可以调用前两个程序获取预处理数据并使用LSTM网络进行训练,也可以调用gen_poetry.py生成古诗
head:床前明月光 + style:黄沙百战金甲:
床前明月光辉,魏武征夫血絮红。
数步崩云复遗主,缟衣东,帝京举,玉轮还满出书初。
秋秋惨惨垂杨柳,梦断黄莺欲断肠。
花凋柳映阮家几,屋前病,歇马空留门。
当年皆月林,独往深山有素。
head:少小离家老大回 + style:山雨欲来风满楼:
少小离家老大回,四壁百月弄鸦飞。
扫香花间春风地,隔天倾似烂桃香。
近来谁伴清明日,两株愁味在罗帏。
仍通西疾空何处,轧轧凉吹日方明。
head:少小离家老大回 + style:铁马冰河入梦来:
少小离家老大回,化空千里便成丝。
官抛十里同牛颔,莫碍风光雪片云。
饮水远涛飞汉地,云连城户翠微低。
一树铁门万象耸,白云三尺各关高。
同言东甸西游子,谁道承阳要旧忧。
少小离家老大回,含颦玉烛拂楼台。
初齐去府芙蓉死,细缓行云向国天。