[发明专利]基于深度学习的诗歌生成方法、装置、设备及存储介质

说明书

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

技术领域

本发明实施例涉及计算机技术领域，尤其涉及一种基于深度学习的诗歌生成方法、装置、设备及存储介质。

背景技术

诗歌作为中华文化精髓的一部分，经过长时间的积累，产生了很多质量较高的古诗和现代诗等诗歌语料。随着深度学习技术的迅猛发展，基于深度学习的模型也被应用在诗歌创作上。深度学习模型通过学习现有的古诗或现代诗语料，学习它们的写作特征和方式，然后进行模仿、创作、再生成，从而实现自动生成诗歌的功能。

目前，基于深度学习的诗歌生成模型，采用逐句循环生成的方式生成诗歌，具体的，通过输入诗歌的主题信息或者诗歌的起始诗句，生成诗歌的下一句诗句，然后将生成的诗句作为下一次的输入，通过模型再次生成下一句，以此循环，直至得到完整的诗歌。通过这种逐句循环生成的方式生成的诗歌，诗歌中后一句诗句的产生仅仅依赖于前一句诗句的信息，一旦诗歌的句子数量较多时，后续生成的诗句会严重偏离主题信息，无法从整体上把握所生成诗歌的主题内容，生成诗歌的质量差。

发明内容

本发明实施例提供一种基于深度学习的诗歌生成方法、装置、设备及存储介质，用以解决现有技术中通过这种逐句循环生成的方式生成的诗歌，诗歌中后一句诗句的产生仅仅依赖于前一句诗句的信息，一旦诗歌的句子数量较多时，后续生成的诗句会严重偏离主题信息，无法从整体上把握所生成诗歌的主题内容，生成诗歌的质量差的问题。

本发明实施例的一个方面是提供一种基于深度学习的诗歌生成方法，包括：

获取待生成诗歌的主题信息和诗歌类型；

将所述主题信息输入所述诗歌类型对应的诗歌生成模型中，生成包括多首诗歌的候选集，所述诗歌类型对应的诗歌生成模型是采用所述诗歌类型对应的训练语料对基于深度学习的transformer模型训练得到；

计算所述候选集中每首诗歌的质量系数；

根据所述质量系数，确定所述候选集中的至少一首诗歌。

本发明实施例的另一个方面是提供一种基于深度学习的诗歌生成装置，包括：

数据获取模块，用于获取待生成诗歌的主题信息和诗歌类型；

诗歌生成模块，用于将所述主题信息输入所述诗歌类型对应的诗歌生成模型中，生成包括多首诗歌的候选集，所述诗歌类型对应的诗歌生成模型是采用所述诗歌类型对应的训练语料对基于深度学习的transformer模型训练得到；

质量筛选模块，用于：计算所述候选集中每首诗歌的质量系数；根据所述质量系数，确定所述候选集中的至少一首诗歌。

本发明实施例的另一个方面是提供一种基于深度学习的诗歌生成设备，包括：

存储器，处理器，以及存储在所述存储器上并可在所述处理器上运行的计算机程序，

所述处理器运行所述计算机程序时实现上述所述的基于深度学习的诗歌生成方法。

本发明实施例的另一个方面是提供一种计算机可读存储介质，存储有计算机程序，

所述计算机程序被处理器执行时实现上述所述的基于深度学习的诗歌生成方法。