1. 一种迭代润色的诗歌自动生成方法,其特征在于:包括以下步骤:
S10,获取诗歌训练集;
S20,对诗歌训练集进行预处理,得到训练样本集;
S30,通过训练样本集对诗歌生成模型进行训练;
S40,将用户提供的表示用户意图的关键词作为诗歌生成模型的输入,训练好的诗歌生成模型输出生成的草诗;
S50,将草诗以及关键词作为诗歌生成模型的输入进行诗歌润色,并通过迭代,生成最终诗歌并输出。
2. 根据权利要求1所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述步骤S50,将草诗以及关键词作为诗歌生成模型的输入进行诗歌润色,并通过迭代,生成最终诗歌并输出,包括:
S501,初始化诗歌生成模型的参数;
S502,将草诗以及关键词作为诗歌生成模型的输入,诗歌生成模型输出润色后的诗歌;
S503,判断是否达到最大迭代次数,如是,则将润色后的诗歌作为最终结果输出,否则,执行步骤S504;
S504,判断是否达到寻优精度,如达到,则将润色后的诗歌作为最终结果输出,结束迭代;否则,当前迭代次数+1后,执行步骤S502。
3. 根据权利要求2所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述步骤S503中,判断是否达到寻优精度的过程为:
S5031,对诗歌生成模型两次输出的诗歌,进行诗歌向量的余弦相似度计算,得到两次诗歌之间的余弦相似度值;
S5032,判断余弦相似度值是否大于余弦相似度预设值,如是,则表示达到寻优精度,否则表示未达到寻优精度。
4. 根据权利要求2所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述步骤S502,将草诗以及关键词作为诗歌生成模型的输入,诗歌生成模型输出润色后的诗歌;具体包括:
S502-1,对部分关键词进行替换,得到替换后的关键词;
S502-2,将草诗以及替换后的关键词作为诗歌生成模型的输入,诗歌生成模型输出润色后的诗歌。
5. 根据权利要求1所述的一种迭代润色的诗歌自动生成方法,其特征在于:S20,对诗歌训练集进行预处理,得到训练样本集;具体包括:
S201,通过词嵌入将诗歌训练集中的每个原始诗歌映射到向量空间,得到每个原始诗歌对应的诗歌词向量;
S202,将诗歌训练集中的每个原始诗歌进行分词处理;分词后,原始诗歌中的每一句诗被分割为多个词语;
S203,通过TextRank公式计算每个词语的分值大小,提取得分最高的词语作为每句诗的关键词;
S204,每个原始诗歌对应的关键词和诗歌词向量构成预处理后的训练样本集。
6. 根据权利要求5所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述步骤S30,通过训练样本集对诗歌生成模型进行训练中,具体包括:
S301,所述的诗歌生成模型包括:双向GRU编码器和GRU解码器;
S302,将训练样本集中的关键词和诗歌词向量输入到双向GRU编码器编码后得到隐藏状态的向量表示;
S303,将隐藏状态的向量表示输入到GRU解码器中进行解码后,得到训练样本的输出诗歌。
7. 根据权利要求6所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述双向GRU编码器包括:采用正向学习的正向GRU-L及采用反向学习的反向GRU-R;
所述步骤S30,通过训练样本集对诗歌生成模型进行训练,还包括:
S304,采用Adam优化函数设置优化参数来训练反向GRU-R。
8. 根据权利要求6所述的一种迭代润色的诗歌自动生成方法,其特征在于:所述优化参数包括:学习率,所述学习率为0.02。
9. 一种计算机可读存储介质,其特征在于,当所述存储介质中的指令由电子设备的处理器执行时,使得电子设备能够执行如权利要求1至8所述的一种迭代润色的诗歌自动生成方法方法。
10. 终端,其特征在于:包括:
处理器,适于实现各指令;以及
存储介质,适于存储多条指令,所述指令适于由处理器加载并执行如权利要求1~8中任一所述的一种迭代润色的诗歌自动生成方法。