🌲 Try YDF, the successor of TensorFlow Decision Forests using the same algorithms but with more features and faster training!
Old code
import tensorflow_decision_forests as tfdf tf_ds = tfdf.keras.pd_dataframe_to_tf_dataset(ds, label="l") model = tfdf.keras.RandomForestModel(label="l") model.fit(tf_ds)
New code
import ydf model = ydf.RandomForestLearner(label="l").train(ds)
(Learn more in the migration guide)