Hey community,
I am trying to create a ROC (weighted and per class) in my multiclass segmentation model. But, in the code example of sklearn they binarize the y (before even train test split) to do this multiclass task. So my question is: how to do this binarize my model, which I have the label (y_true, a tensor Bx1xHxW) and the model prediction (y_pred, a tensor BxCxHxW).
Best,
Giulia