PyTorch-Ignite PyTorch-Ignite

指标

Ignite 提供了各种机器学习任务的现成指标列表。支持两种计算指标的方式:

  1. 在线
  2. 存储整个输出历史

指标可以附加到 Engine

from ignite.metrics import Accuracy

accuracy = Accuracy()

accuracy.attach(evaluator, "accuracy")

state = evaluator.run(validation_data)

print("Result:", state.metrics)
# > {"accuracy": 0.12345}

或者可以作为独立对象使用:

from ignite.metrics import Accuracy

accuracy = Accuracy()

accuracy.reset()

for y_pred, y in get_prediction_target():
    accuracy.update((y_pred, y))

print("Result:", accuracy.compute())

完整的指标列表和API可以在 ignite.metrics 模块中找到。