Downloads · 30 days
79
5% of all-time downloads
TF_Decision_Trees
keras-io/TF_Decision_Trees
TF_Decision_Trees is a tabular classification model from keras-io. Use it for the tabular classification task on the model card, and read the license before you ship it in a product. It is set up for tf-keras. The card lists the license as apache-2.0.
Use TF's Gradient Boosted Trees model in binary classification of structured data <br /
All-time downloads
1.7K
Public
Repo size
1.8 MB
Likes
7
Public
Repo makeup
Click a slice to open those files.
Other801 KB · 54%
At a glance
- Task
- Tabular Classification
- Library
- tf-keras
- License
- apache-2.0
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024
- SHA
- 2844fc78
Model card
From the Hugging Face model README
TensorFlow's Gradient Boosted Trees Model for structured data classification
Use TF's Gradient Boosted Trees model in binary classification of structured data <br />
- Build a decision forests model by specifying the input feature usage.
- Implement a custom Binary Target encoder as a Keras Preprocessing layer to encode the categorical features with respect to their target value co-occurrences, and then use the encoded features to build a decision forests model.<br />
The model is implemented using Tensorflow 7.0 or higher. The US Census Income Dataset containing approximately 300k instances with 41 numerical and categorical variables was used to train it. This is a binary classification problem to determine whether a person makes over 50k a year.<br />
Author: Khalid Salama
Adapted implementation: Tannia Dubon
Find the colab notebook at https://github.com/tdubon/TF-GB-Forest/blob/c0cf4c7e3e29d819b996cfe4eecc1f2728115e52/TFDecisionTrees_Final.ipynb
- Task
- Tabular Classification
- Library
- tf-keras
- License
- apache-2.0
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024