Many thanks for implementing SOLO! I have a question about the predict() module.
(i) Are the prediction scores returned with predict(soft=True) the probabilities of the logit plus calibration term?
(ii) When returning the labels with predict(soft=False) the label with the higher probability is returned by preds_df = preds_df.idxmax(axis=1). My apologies if I get confused here but how can I threshold the class prediction as described in the original publication (default 0.5)?
Many thanks for your help!