e839626c28d8dc6030ce71f3619e934b36fb6627,catalyst/metrics/functional.py,,get_multiclass_statistics,#,167

Before Change

    if num_classes is None:
        // as far as we expect the outputs/targets tensors to be int64
        // we could find number of classes as max available number
        num_classes = max(
            int(outputs.max().detach().item() + 1),
            int(targets.max().detach().item() + 1),
        )

    tn = torch.zeros((num_classes,), device=outputs.device)
    fp = torch.zeros((num_classes,), device=outputs.device)
    fn = torch.zeros((num_classes,), device=outputs.device)

After Change

        tensor([0., 0., 0., 1., 1.]), tensor([1., 1., 0., 0., 0.]),
        tensor([1., 1., 0., 1., 1.])
    
    outputs, targets, num_classes = process_multiclass_components(
        outputs=outputs,
        targets=targets,
        argmax_dim=argmax_dim,

Instances