当前位置: 首页 > 工具软件 > Torchnet > 使用案例 >

torchnet.meter使用记录

张英范
2023-12-01

工程中引入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

混淆矩阵。

 类似资料: