多阶段训练文本标题生成模型的方法及装置
摘要:
本公开涉及机器学习技术领域,提供了一种多阶段训练文本标题生成模型的方法及装置。该方法包括:获取第一训练数据集,利用第一训练数据集对文本标题生成模型进行第一阶段训练;获取第一文本数据集,利用第一阶段训练后的文本标题生成模型为第一文本数据集中的文本生成标题,根据第一文本数据集中的文本以及该文本对应的标题生成第二训练数据集;利用第二训练数据集对标题打分模型进行训练;获取第二文本数据集,利用第二文本数据集和训练后的标题打分模型对第一阶段训练后的文本标题生成模型进行第二阶段训练。采用上述技术手段,解决现有技术中,生成的文本标题不通顺以及语义不连贯的问题。
公开/授权文献
0/0