Evaluator

用于模型评估的评测器。

class mindnlp.engine.evaluator.Evaluator(network, eval_dataset=None, metrics=None, callbacks=None, jit=False)[源代码]

基类:object

用于测试模型的评测器。

参数
  • network (Cell) – 用于评估的模型网络。

  • eval_dataset (Dataset) – 测试数据集迭代器。

  • batc_size (int) – 每个batch的样本数量。

  • metrcis (Optional[list[Metric], Metric]) – 模型评估时应使用的metric对象列表。 默认值:无。

  • callbacks (Optional[list[Callback], Callback]) – 在模型训练过程中应当执行的回调函数方法列表。默认值:无。

  • jit (bool) – 是否使用Just-In-Time编译模式。

clear_metrics()[源代码]

清空评估指标值。

run(tgt_columns=None)[源代码]

评测过程入口。

参数

tgt_columns (Optional[list[str], str]) – 用于计算损失函数的目标标签列名称。