mixmo.learners.model_wrapper.get_predictions

mixmo.learners.model_wrapper.get_predictions(logits)[source]

Convert logits into softmax predictions