File size: 1,354 Bytes
aaaa4f9 b9d9695 ad71777 b9d9695 aaaa4f9 b9d9695 |
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 47 48 49 |
---
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"
---
# Classification with TensorFlow Decision Forests
#### Using TensorFlow Decision Forests for structured data classification
<br />
##### This example uses Gradient Boosted Trees model in binary classification of structured data, and covers the following scenarios:
1. Build a decision forests model by specifying the input feature usage.
2. 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.
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.
Author: Khalid Salama <br />
Adapted implementation: Tannia Dubon
|