工程中引入torchnet.meter,记录下使用方式。
安装:pip install torchnet
1、AverageValueMeter
函数作用:添加单值数据,进行取平均及方差计算。
from torchnet import meter
loss_meter = meter.AverageValueMeter()
loss_meter.reset()
for i in range(10):
loss_meter.add(i)
print(loss_meter.value())
=>
(4.5, 3.0276503540974917) # (mean, std)
使用重置(清空序列):
loss_meter.reset()
2、ConfusionMeter
混淆矩阵。