import xgboost as xgb
train = xgb.DMatrix(X_train, y_train)
test = xgb.DMatrix(X_test, y_test)
bst = xgb.train(params, train, max_rounds, evals=[(test, 'validation_0')], verbose_eval=True)
I am using xgboost for two-class (0, 1) classification.
I can get predicted labels by bst.predict(), where bst is the name of my model.
However, I need probability of each class, which is usually given by predict_proba() for other classification algorithms.
But I cannot find bst.predict_proba(), as shown in the scheenshot below.
Could you please advise me?
Thank you so much…