0

I am training an xgboost model for binary classification using objective as 'binary:logistic'. The model should predict probability but it is outputting either 0 or 1. I want the model to output the probability. These are the parameters I am using enter image description here Thanks a lot in advance!

shivani
  • 150
  • 10

1 Answers1

1

How are you getting the predictions from your model? It is likely that the default predict method automatically converts the probability to a 0/1 prediction, and that there's a separate method to get the underlying probability (e.g. see the predict and predict_proba methods for the LogisticRegression in scikit-learn).

Oxbowerce
  • 8,522
  • 2
  • 10
  • 26