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

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.
Frequently asked
- What is tensorflow/decision-forests?
- A TensorFlow library for training, serving, and interpreting decision forest models including Random Forests and Gradient Boosted Trees in Keras.
- Is decision-forests open source?
- Yes — tensorflow/decision-forests is open source, released under the Apache-2.0 license.
- What language is decision-forests written in?
- tensorflow/decision-forests is primarily written in Python.
- How popular is decision-forests?
- tensorflow/decision-forests has 694 stars on GitHub.
- Where can I find decision-forests?
- tensorflow/decision-forests is on GitHub at https://github.com/tensorflow/decision-forests.