[发明专利]诗句生成模型的训练方法、自动写诗方法及装置

摘要

本申请提供了一种诗句生成模型的训练方法、自动写诗方法及装置。该方法在训练诗句生成模型时，诗句的每个节拍均作为主题词，并且对于每个主题词，以其所在诗句之前所有诗句的节拍序列作为输入序列，以主题词所在的诗句作为输出序列，从而使诗句生成模型学习的是根据主题词和主题词之前的诗句内容预测主题词对应诗句的能力，使诗句生成模型对诗句的主题相关性、诗句的语义通顺性、诗句之间的语义关联性等方面都得到的训练，从而，本申请提供的技术方案得到的诗句生成模型与现有技术的诗句生成模型相比，其自动创作的诗在诗句的主题相关性、诗句的语义通顺性、诗句之间的语义关联性等方面都会得到提升，创作诗质量更高。

主权项

1.一种诗句生成模型的训练方法，其特征在于，所述诗句生成模型包括编码层和解码层，所述编码层用于输入诗的主题词，所述解码层用于预测主题词对应的诗句，所述方法包括：对训练语料中的每一首诗的每句诗句进行节拍划分，得到每句诗句的节拍序列，其中，所述节拍序列由句首节拍和一个或一个以上的其余节拍组成，所述句首节拍对应诗句句首的预设数量的字符，所述其余节拍由诗句去除句首节拍后分词得到，每个其余节拍对应一个分词；对于每首诗，将第一句诗句对应的节拍序列的每个节拍逐一作为编码层的训练输入，每个节拍分别以所述第一句诗句作为解码层的训练输出；以及，对于除第一句诗句以外的其余每句诗句，分别将诗句中的每个节拍与当前诗句之前所有诗句的节拍序列拼接成输入序列，并将所述输入序列作为编码层的训练输入，每个输入序列分别以当前诗句作为解码层的训练输出，以训练诗句生成模型。