Ahh ok. Thanks. When I want to calculate the metrics on the predictions in most cases a simple preds = pred.predictions.argmax(-1) is enough. There is no need to apply softmax or sigmoid infront of it. Only in the following cases I have to apply an “activation function” with numpy:

when I have multi label lassification I can not do argmax

when I want to compute metrics like ROC-AUC where I need “probability estimates”

@sgugger @valhalla is that right?