补全TN计算的代码如下:

TN = torch.zeros(14).cuda()

for ak in range(14):
    if ak == 0:
        continue
    TN[ak] = slicetotal - TP[ak] - FP[ak] - FN[ak]

将以上代码添加到原有代码中的合适位置即可。


原文地址: http://www.cveoy.top/t/topic/iTW0 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录