Hi, I made the ROC curve on the validation_set of my dataset using a pre-trained model (“fpr, tpr, thresholds = roc_curve (…)”). Now I wanted to test this model on new images to be able to see the results of my work for this binary classification. At the moment I just used a “torch.max (output, 1)” to determine which class the input image belongs to. Is it possible to do this classification by modifying the choice according to the type of sensitivity I want?