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

权利要求书

1.一种诗歌生成方法，其特征在于：所述方法包括：

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

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

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

其中，

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

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

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

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

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

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

2.如权利要求1所述的一种诗歌生成方法，其特征在于：将每个子输入序列依次输入编码网络以及解码网络进行预测下一个输出的字，具体包括：

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

3.如权利要求1所述的一种诗歌生成方法，其特征在于：所述计算每个候选诗歌的输出概率，选取最大输出概率的候选诗歌作为目标诗歌，包括：

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

4.如权利要求1所述的一种诗歌生成方法，其特征在于：所述AI写诗模型包括：

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

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

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

5.一种诗歌生成系统，其特征在于，包括：

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

第二输入模块，用于获取待生成的目标诗歌的类型信息，所述类型信息包括诗歌的格律或词牌信息；

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

所述诗歌生成模块包括：

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

第二子模块，用于计算每个候选诗歌的输出概率，选取最大输出概率的候选诗歌作为目标诗歌；

所述第一子模块包括：

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

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

6.一种可读存储介质，其特征在于，包括处理器，所述处理器用于加载并执行存储介质中的指令以实现权利要求1至4中任一权利要求所述的诗歌生成方法。