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