基于pytorch_rnn的古诗词生成

Overview

pytorch_peot_rnn

基于pytorch_rnn的古诗词生成

说明

config.py里面含有训练、测试、预测的参数,更改后运行:

python main.py

预测结果

if config.do_predict:
	result = trainer.generate('丽日照残春')
	print("".join(result))
	result = trainer.gen_acrostic('深度学习')
	print("".join(result))
	
丽日照残春,
风光摇落时。
不知花发意,
不得见春风。

深山高下有余灵,万里无人见钓矶。
度日茱萸人不得,一枝不得不相见。
学舞一枝花落叶,不知何处是君王。
习书不见金闺后,应是君王赐手间。

参考

https://github.com/chenyuntc/pytorch-book
其中第九章的古诗词生成,修改了以下地方:
1、重构了代码架构;
2、增加了数据集生成的过程;
3、RNN网络改为batch_first;
4、计算损失时不计算padding部分;

You might also like...
Owner
西西嘛呦
西西嘛呦