I have a multi-class problem, the classes are all encoded 0-72.
I have an preds tensor of [256, 72].
Passing it through probs = torch.nn.functional(input, dim = 1) results in a tensor
with the same dimensionality.
Where probs[0] is a list of probabilities of each class being the correct prediction.
I would like to analyse the predictions my model is making, how can I link the probabilities to specific classes?