[发明专利]一种诗歌生成方法、系统及可读存储介质

说明书

本发明公开了一种诗歌生成方法、系统及可读存储介质，涉及AI自然语言处理领域。本发明提供了一种诗歌生成方法，所述方法包括：获取待生成的目标诗歌的主题信息；获取待生成的目标诗歌的类型信息，所述类型信息包括诗歌的格律或词牌信息；将所述主题信息以及所述类型信息输入预训练AI写诗模型，生成目标诗歌。用户输入用于生成诗歌的主题信息以及诗歌的类型信息，本方案能够根据输入的信息，自动生成符合要求的诗歌；并且，诗歌生成过程中，采用全局最优算法来对解码结果进行优化，防止出现局部最优解，从而实现改善诗歌质量的作用。

技术领域

本发明涉及AI自然语言处理领域，特别涉及一种诗歌生成方法、系统及可读存储介质。

背景技术

AI(人工智能)作为一门新兴科学技术，可以利用深度学习网络，通过海量的数据进行训练以达到自动学习的目的。当前，人工智能写诗便是在教育领域用于辅助教学的一种实际应用，AI+学科融合，既可让学生在实际案例中学习到人工智能的知识，同时可以学习到作诗的乐趣，提高文学水平，这种寓教于乐的方式必定会大受欢迎。因此，如何根据用户提供的诗歌主题等信息，自动生成高质量的诗歌，是本领域技术人员关注的焦点。

发明内容

本发明提供一种根据诗歌主题、类型，自动生成高质量的诗歌的方法、系统及可读存储介质，本发明的方案如下：

第一方面，本发明提供了一种诗歌生成方法，所述方法包括：

获取待生成的目标诗歌的主题信息；

获取待生成的目标诗歌的类型信息，所述类型信息包括诗歌的格律或词牌信息；

将所述主题信息以及所述类型信息输入预训练AI写诗模型，生成目标诗歌。

其中，所述将所述主题信息以及所述类型信息输入AI写诗模型，生成目标诗歌，包括：

将所述主题信息和所述类型信息按预定的规则预处理成输入序列，而后输入AI写诗模型进行预测，生成候选诗歌；

计算每个候选诗歌的输出概率，选取最大输出概率的候选诗歌作为目标诗歌。

其中，所述将输入序列输入到AI写诗模型进行预测，生成候选诗歌的具体方法包括：

将所述输入序列按字进行切分，构建出多个不同长度的子输入序列；

分别将每个子输入序列依次输入编码网络以及解码网络进行预测下一个输出的字，将所述输出的字加入到对应的子输入序列中作为新的输入预测下一个输出的字，直至预测出目标诗歌最后一个字，生成候选诗歌。

其中，将每个子输入序列依次输入编码网络以及解码网络进行预测下一个输出的字，具体包括：

将子输入序列输入编码网络以及解码网络，输出当前时刻输出概率topN的字构成概率候选集，而后，将当前时刻的概率候选集作为输入，输出下一时刻输出概率topN的字构成概率候选集，重复以上步骤，直至输出候选诗歌最后一个字对应的概率候选集。

其中，所述计算每个候选诗歌的输出概率，选取最大输出概率的候选诗歌作为目标诗歌，包括：

将候选诗歌中预测每个字的输出概率进行累加，累加结果作为该候选诗歌的输出概率。

其中，所述AI写诗模型包括：

词嵌入网络，用于将输入序列处理成词向量；

编码网络，用于将所述词向量处理成表征向量；

解码网络，用于将所述表征向量进行解码，所述解码过程采用全局最优算法寻找出最优解，作为输出的目标诗歌。

第二方面，本发明提供一种诗歌生成系统，包括：

第一输入模块，用于获取待生成的目标诗歌的主题信息；