A TensorFlow library for training, serving, and interpreting decision forest models like Random Forests and Gradient Boosted Trees.
TensorFlow Decision Forests is a library for training, serving, and interpreting decision forest models within TensorFlow. It provides state-of-the-art algorithms like Random Forests and Gradient Boosted Trees for classification, regression, and ranking tasks, seamlessly integrating with the TensorFlow and Keras ecosystems.
Machine learning engineers and data scientists who use TensorFlow and want to incorporate powerful, interpretable tree-based models into their workflows without leaving the TensorFlow environment.
Developers choose TF-DF because it offers the performance and interpretability of decision forests with the convenience of TensorFlow's API, enabling unified model development and deployment alongside neural networks.
A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Allows training decision forests using the familiar Keras API, making it easy for TensorFlow users to adopt without learning a new interface, as shown in the usage example with tfdf.keras methods.
Models are compatible with Yggdrasil Decision Forests (YDF), enabling export and use across multiple platforms like C++, JavaScript, and Go, which is highlighted in the README's model compatibility note.
Includes tools for interpreting and explaining model predictions, enhancing transparency without external libraries, as mentioned in the key features.
Supports data loading, training, evaluation, and model export within TensorFlow, providing a cohesive pipeline demonstrated in the minimal example from the README.
The README explicitly recommends migrating to YDF, which is faster and has more functionality, suggesting TF-DF may not be the primary focus for future development and could become deprecated.
Only available on Linux and Mac; Windows users must rely on WSL, adding complexity and excluding native Windows deployments, as stated in the installation notes.
Compared to YDF, TF-DF is slower, and for high-performance needs, users might prefer optimized standalone libraries like XGBoost or LightGBM, despite the TensorFlow integration benefits.