search

google / yggdrasil-decision-forests

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.
https://ydf.readthedocs.io/
Apache License 2.0
499 stars 54 forks source link

Is the model prediction probability calibrated? #15

Closed Vedant-R closed 2 years ago

Vedant-R commented 2 years ago

I understand that the sklearn random forest models prediction probabilities are not calibrated and we need to add steps in between to calibrate it.

Just wanted to understand if the prediction probabilities of yggdrasil-decision-forests are calibrated?

If not, how can be calibrate it?

achoum commented 2 years ago

See the answer to this question on TF-DF: https://github.com/tensorflow/decision-forests/issues/71