← all repositories

tensorflow/decision-forests

A TensorFlow library for training, serving, and interpreting decision forest models including Random Forests and Gradient Boosted Trees in Keras.

694 stars Python ML Frameworks
decision-forests
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

TensorFlow Decision Forests (TF-DF) is a library for training and serving decision forest models such as Random Forests and Gradient Boosted Trees within the TensorFlow ecosystem. It supports classification, regression, and ranking tasks through a Keras-integrated API. The library accepts Pandas dataframes converted to TensorFlow datasets, exports models as TensorFlow SavedModels, and is powered by the underlying Yggdrasil Decision Forests C++ implementation.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.