Skip to main content

如何定义摘要评估器

一些指标只能在整个实验层面定义,而不能针对实验的单个运行。 例如,您可能希望计算评估目标在整个数据集中所有示例上的整体通过率或 f1 分数。 这些被称为summary_evaluators

基础示例

在这里,我们将计算 F1 分数,它是精确率和召回率的综合指标。

此类指标仅能基于实验中所有示例进行计算,因此我们的评估器接收一个输出列表和一个参考输出列表。

def f1_score_summary_evaluator(outputs: list[dict], reference_outputs: list[dict]) -> dict:
true_positives = 0
false_positives = 0
false_negatives = 0
for output_dict, reference_output_dict in zip(outputs, reference_outputs):
output = output_dict["class"]
reference_output = reference_output_dict["class"]
if output == "Toxic" and reference_output == "Toxic":
true_positives += 1
elif output == "Toxic" and reference_output == "Not toxic":
false_positives += 1
elif output == "Not toxic" and reference_output == "Toxic":
false_negatives += 1

if true_positives == 0:
return {"key": "f1_score", "score": 0.0}

precision = true_positives / (true_positives + false_positives)
recall = true_positives / (true_positives + false_negatives)
f1_score = 2 * (precision * recall) / (precision + recall)
return {"key": "f1_score", "score": f1_score}

然后您可以将此评估器传递给evaluate方法,如下所示:

from langsmith import Client

ls_client = Client()
dataset = ls_client.clone_public_dataset(
"https://smith.langchain.com/public/3d6831e6-1680-4c88-94df-618c8e01fc55/d"
)

def bad_classifier(inputs: dict) -> dict:
return {"class": "Not toxic"}

def correct(outputs: dict, reference_outputs: dict) -> bool:
"""Row-level correctness evaluator."""
return outputs["class"] == reference_outputs["label"]

results = ls_client.evaluate(
bad_classified,
data=dataset,
evaluators=[correct],
summary_evaluators=[pass_50],
)

在 LangSmith UI 中,您将看到评估器的分数及其对应的键。

摘要评估器参数

摘要评估函数必须具有特定的参数名称。它们可以接受以下参数的任意子集:

  • inputs: list[dict]: 数据集中单个示例对应的输入列表。
  • outputs: list[dict]: 每个实验在给定输入上产生的字典输出列表。
  • reference_outputs/referenceOutputs: list[dict]: 与示例关联的参考输出列表(如果可用)。
  • runs: list[Run]: 由两个实验在给定示例上生成的完整 Run 对象列表。如果您需要访问每个运行的中间步骤或元数据,请使用此参数。
  • examples: list[Example]: 数据集的所有 示例 对象,包括示例输入、输出(如果可用)和元数据(如果可用)。

摘要评估器输出

摘要评估器应返回以下类型之一:

Python 和 JS/TS

  • dict: dicts of the form {"score": ..., "name": ...} allow you to pass a numeric or boolean score and metric name.

目前仅支持 Python

  • int | float | bool: 这被解释为一个可以平均、排序等的连续指标。函数名称用作该指标的名称。

此页面有帮助吗?


您可以留下详细的反馈 在 GitHub 上