search

tensorflow / decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.
Apache License 2.0
663 stars 110 forks source link

predict_log_proba is missing on 0.1.7 #32

Closed Howard-ll closed 3 years ago

Howard-ll commented 3 years ago

I heard that predict_log_proba will be supported on 0.1.7 but it is still missing Please take a look at https://github.com/tensorflow/decision-forests/issues/26

achoum commented 3 years ago

The logits are available in v0.1.7, but the signature is different from sklearn:

# Trains a Gradient Boosted Trees that returns logits (assuming the dataset is a binary classification)
model = tfdf.keras.GradientBoostedTreesModel(apply_link_function=False)