昇腾社区首页
中文
注册

evaluate

功能描述

评估大语言模型抽取的三元组关系。

函数原型

def evaluate(relations)

输入参数说明

参数名

数据类型

是否必选

说明

relations

list[dict]

知识图谱关系列表,每个元素为一个字典,必须包括如下键:raw_text, file_id, entity_relations, event_entity_relations, event_relations, output_stage_one, output_stage_two, output_stage_three。

返回值说明

无。函数会打印输出三组精度、召回以及F1得分,分别对应实体,事件中实体以及事件的抽取情况。输出的得分越高,抽取的质量越好。