基于深度学习的诗歌生成方法、装置、设备及存储介质

摘要

本发明实施例提供一种基于深度学习的诗歌生成方法、装置、设备及存储介质，本发明实施例的方法，预先根据各诗歌类型对应的训练语料，对基于深度学习的transformer模型训练，得到各诗歌类型对应的诗歌生成模型；在使用时，获取待生成诗歌的主题信息和诗歌类型；将主题信息输入诗歌类型对应的诗歌生成模型中，诗歌生成模型用于根据主题信息直接生成完整的诗歌，得到包括多首诗歌的候选集，能够从整体上把握所生成诗歌的主题内容，提高生成诗歌的质量；通过计算候选集中每首诗歌的质量系数；针对不同诗歌类型设定不同的筛选策略，根据质量系数从候选集中筛选出质量最高且与主题信息关联最强的至少一首诗歌，进一步提高了生成诗歌的质量。