首页 > 解决方案 > Pytorch 点燃指标 - 传递对象

问题描述

我一直在使用 pytorch ignite 中的指标,并且遇到了将 ignite 对象传递给其他 ignite 对象的问题,例如这段代码

cm = metrics.ConfusionMatrix(10)
cm.reset()
cm.update((y_pred, y_true_sparse))
print(cm.compute())
dice = metrics.DiceCoefficient(cm)
print(dice.compute())

ignite.exceptions.NotComputableError: Confusion matrix must have at least one example before it can be computed.即使混淆矩阵没有失败,也会抛出。当我尝试计算 F beta 分数(通过精度和召回对象)时,也发生了同样的问题。我尝试了一些实验,例如在传递之前不计算混淆矩阵或再次更新数据但没有成功。

我究竟做错了什么?这种行为对我来说似乎很奇怪,即使深入研究我似乎无法弄清楚的文档,也没有出现相关的谷歌搜索结果。

标签: pythonpytorchpytorch-ignite

解决方案


推荐阅读