[发明专利]一种可控情感与风格的古诗生成方法

权利要求书

1.一种可控情感与风格的古诗生成方法，包括如下模块，其特征在于：

1)古诗分类：对收集的海量古诗依据风格标签、情感标签、主关键词标签进行分类，根据预先人工精确分类的有标签古诗，采用诗词向量的余弦距离判断相似度的方法对无标签古诗进行风格、情感分类，采用检索同义词的方法对无标签古诗进行主关键词分类，将每首语料库古诗预处理成统一格式，输入解码器模型进行预训练；

2)古诗生成：采用带掩膜的自注意力机制屏蔽到后续信息，增强对标签的关注并实现古诗生成，首先对输入的文本进行关键词筛选、关键词校正和关键词分配，将文本分词后筛选出概括性关键词，并根据古诗语料进行校正，分配出主关键词标签与副关键词标签，之后，将用户给定的格式标签、风格标签、情感标签和主关键词标签包装成标签集，解码器预训练模型再根据模块1)分类的训练语料库、标签集与副关键词标签自动生成关联诗句，且进行格式和韵律的检查；

3)古诗评分和交互：本方法对传统文本生成的评价指标进行微调，令其适用于古诗生成评价，并将训练语料中的名诗名篇标记为优秀集，对模块2)生成的古诗质量进行自动评分，同时也支持人工对其情感标签、风格标签的契合度进行评分；用户还可以对生成诗歌不满意的位置进行修改与润色，并且本方法会对指定修改位置提供替换词，以达到人机交互、共同润色的作用。

2.根据权利要求1所述的一种可控情感与风格的古诗生成方法，其特征在于，步骤1)中对收集的海量古诗依据风格标签、情感标签、主关键词标签进行分类，采用人工检索与机器自动计算相似度相结合的方法，构建按情感标签、风格标签、主关键词标签细粒度分类的训练古诗语料库和由名诗名篇组成的优秀集。

3.根据权利要求1所述的一种可控情感与风格的古诗生成方法，其特征在于，步骤2)中对输入预处理增设关键词分配环节，根据筛选、校正后的概括性关键词和预先设置的标准主关键词来确定唯一的主关键词标签，其余均作为副关键词标签。

4.根据权利要求1所述的一种可控情感与风格的古诗生成方法，其特征在于，步骤2)中依据用户给定的格式标签、情感标签、风格标签和主关键词标签包装成标签集，自动生成古诗，并在生成诗句时自动关联副关键词标签，以此达到情感、风格、多关键词可控的效果，最后本方法还对古诗进行筛选，剔除不满足格式、押韵需求的古诗。