e839626c28d8dc6030ce71f3619e934b36fb6627,catalyst/metrics/functional.py,,process_multiclass_components,#,33

Before Change

            "expected 1D or 2D with size 1 in the second dim"
        )

    if targets.max() != len(torch.unique(targets)) - 1:
        if raise_class_labels_mismatch:
            raise Exception(
                "`targets` maximum does not represent number of classes"
            )
        // mapping classes
        outputs, targets = map_labels_to_classes(outputs, targets)
    return outputs, targets


def process_multilabel_components(

After Change

            outputs.shape[argmax_dim], int(targets.max().detach().item() + 1)
        )
        outputs = torch.argmax(outputs, dim=argmax_dim)
    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),
        )

    if outputs.dim() == 1:
        outputs = outputs.view(-1, 1)
    else:
        assert outputs.size(1) == 1 and outputs.dim() == 2, (

Instances