Update README.md
Browse files
README.md
CHANGED
|
@@ -34,13 +34,12 @@ pipeline_tag: "structured-data-classification"
|
|
| 34 |
---
|
| 35 |
|
| 36 |
# TensorFlow Decision Forests for structured data classification
|
| 37 |
-
|
| 38 |
-
Use TF's Gradient Boosted Trees model in binary classification of structured data
|
| 39 |
-
|
| 40 |
* Build a decision forests model by specifying the input feature usage.
|
| 41 |
-
* 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
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
<br />
|
| 45 |
Author: Khalid Salama
|
| 46 |
Adapted implementation: Tannia Dubon
|
|
|
|
| 34 |
---
|
| 35 |
|
| 36 |
# TensorFlow Decision Forests for structured data classification
|
| 37 |
+
|
| 38 |
+
Use TF's Gradient Boosted Trees model in binary classification of structured data <br />
|
| 39 |
+
|
| 40 |
* Build a decision forests model by specifying the input feature usage.
|
| 41 |
+
* 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 />
|
| 42 |
+
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 />
|
| 43 |
+
|
|
|
|
| 44 |
Author: Khalid Salama
|
| 45 |
Adapted implementation: Tannia Dubon
|