文本摘要生成方法、装置、电子设备及存储介质
摘要:
本发明涉及人工智能技术领域,提供一种文本摘要生成方法、装置、电子设备及存储介质,提取语料文本中的第一命名实体,从而基于语料文本及第一命名实体生成训练文本并基于多个训练文本训练BERT模型,通过训练得到的BERT模型输出预测文本摘要,在获取预测文本摘要中的第二命名实体之后,根据第一命名实体、预测文本摘要及第二命名实体生成整体风险损失值,从而基于整体风险损失值优化所述BERT模型,得到文本摘要生成模型,最后使用文本摘要生成模型生成目标文本的文本摘要。本发明通过引入命名实体训练BERT模型,再通过BERT模型预测文本摘要,提高了文本摘要中命名实体的准确度,从而减少了文本摘要中命名实体错误的情况,提高了生成文本摘要的准确度。
0/0