davechallis / rust-xgboost

Rust bindings for XGBoost.
MIT License
102 stars 37 forks source link

Expose raw probabilities from MultiSoftmax learning objective. #18

Closed montanalow closed 1 year ago

montanalow commented 1 year ago

It would be nice to have predict_proba functionality similar to that provided by scikit, to see the probability of each class in a multiclass classification model.

montanalow commented 1 year ago

I see MultiSoftprob now.