Hey, I am making a multi-class classifier with 4 classes. Now I have printed Sensitivity and Specificity along with a confusion matrix. Now I want to print the ROC plot of 4 class in the curve. As ROC is binary metric, so it is ‘given class vs rest’, but I want to add all 4 classes in the same plot.
With this code, I have got my probability -
output = model.forward(images) p = torch.nn.functional.softmax(output, dim=1)
Can you guys please help me to get this Multi class ROC graph? Thanks.