分类精度计算细节

最新推荐文章于 2024-10-04 05:34:26 发布

chengamoo

最新推荐文章于 2024-10-04 05:34:26 发布

阅读量84

点赞数 2

文章标签：分类数据挖掘人工智能

本文链接：https://blog.csdn.net/m0_55764111/article/details/141020114

版权

    preds = torch.tensor(preds).to('cuda') if not isinstance(preds,torch.Tensor) else preds
    targets = torch.tensor(targets).to('cuda') if not isinstance(targets, torch.Tensor) else targets
    probs = torch.tensor(probs).to('cuda') if not isinstance(probs, torch.Tensor) else probs

    accuracy = torchmetrics.Accuracy(num_classes=2, average='micro').to('cuda')
    lesion_acc = accuracy(preds, targets)
        
    cm = confusion_matrix(targets.squeeze().cpu().numpy(), preds.squeeze().cpu().numpy())

在计算acc的时候，pred和target需要是torch.Tensor类型，且在cuda上，在计算混淆矩阵的时候，则需要转成numpy，且在cpu上