故事生成模型训练方法、装置、设备及存储介质

摘要

本申请实施例公开了一种故事生成模型训练方法、装置、设备及存储介质，属于自然语言处理领域。方法包括：获取第一样本提示文本和第一样本提示文本对应的第一故事文本；基于故事生成模型，根据第一样本提示文本、已经生成的至少一个词语及至少一个词语中的每个词语的第一注意力权重，继续生成至少一个词语，得到包含多个词语的第二故事文本，第一注意力权重由故事生成模型确定；若第二故事文本中包括代词，则获取第二故事文本中位于代词之前的每个词语的第一注意力权重和预设的第二注意力权重；根据第一故事文本、第二故事文本、每个词语的第一注意力权重和每个词语预设的第二注意力权重，训练故事生成模型，提高了模型的准确度。