Bootstrap

NLP实践项目五:基于GRU的古诗生成(pytorch实现)

直接看代码: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] [<sta

;