公开(公告)号：CN114048755A

公开(公告)日：2022-02-15

申请号：CN202111330086.0

申请日：2021-11-10

Applicant: 北京邮电大学

Inventor： 周锋 ,  江宇峰 ,  李小勇 ,  王一童 ,  乔雷

IPC: G06F40/35 ,  G06F40/289 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明提供了一种基于生成对抗网络生成古诗的方法，包括如下步骤：步骤S101，设计生成对抗网络的生成器模型和判别器模型；步骤S102，对基于Transformer的生成器进行预训练；步骤S103，对基于卷积神经网络的判别器进行预训练；步骤S104，进行对抗训练，轮流训练生成器和判别器，直至模型收敛；步骤S105，对输入的关键词进行拓展；步骤S106，对拓展后的关键词进行词嵌入，得到对应的词向量；步骤S107，将得到的词向量输入到基于Transformer的生成器中，得到输出的下一个字，将已生成的序列当作输入，再次输入到生成器中，不断重复步骤S107，直至整首古诗生成完毕。采用本发明的方法，实现基于对抗生成网络的古诗生成模型，提高了生成古诗的质量。