File size: 1,245 Bytes
aaaa4f9 b9d9695 ad71777 b9d9695 aaaa4f9 b9d9695 d2e7b42 989f366 7492a93 989f366 eb2b838 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
---
language:
- en
thumbnail:
tags:
- classification
- gradient boosted trees
- keras
- TensorFlow
libraries: TensorBoard
license: apache-2.0
metrics:
- accuracy
model-index:
- name: TF_Decision_Trees
results:
- task:
type: structured-data-classification
dataset:
type: census
name: Census-Income Data Set
metrics:
- type: accuracy
value: 96.57
pipeline_tag: "structured-data-classification"
---
# TensorFlow Decision Forests 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 example uses Tensorflow 7.0 or higher. It uses the US Census Income Dataset containing approximately 300k instances with 41 numerical and categorical variables. This is a binary classification problem to determine whether a person makes over 50k a year.<br />
Author: Khalid Salama
Adapted implementation: Tannia Dubon
|