直接看代码:github传送门 数据集:古诗数据集 1. 数据预处理: 将古诗存为上图的格式,每行一首诗,便于使用torchtext读取。 2. 模型定义 模型就是一个简单的GRU模型, 3. 模型训练和测试 训练过程中如同翻译模型中的teaching forcing,输入是 [ < s t a r t > , x 1 , x 2 , . . . x n ] [<start>,x_1, x_2, ... x_n] [<star