指标
Ignite 提供了各种机器学习任务的现成指标列表。支持两种计算指标的方式:
- 在线
- 存储整个输出历史
指标可以附加到
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 模块中找到。