Fix probability calculation for uBoostClassifier

arogozhnikov / hep_ml

Machine Learning for High Energy Physics.

https://arogozhnikov.github.io/hep_ml/

Other

179 stars 64 forks source link

Closed ehhov closed 1 month ago

ehhov commented 1 month ago

Change method names uBoostBDT._uboost_predict_score and _uboost_staged_predict_score to *_predict_proba to avoid confusion.
In uBoostClassifier.predict_proba and staged_predict_proba, average the results of individual uBoostBDTs instead of applying a sigmoid.

This pull request is following the recently opened issue https://github.com/arogozhnikov/hep_ml/issues/87.

arogozhnikov commented 1 month ago

yup, thanks for renaming too!