tdubon commited on
Commit
4992c61
1 Parent(s): 989f366

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -38,8 +38,9 @@ pipeline_tag: "structured-data-classification"
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
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
+
43
+ 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 />
44
 
45
  Author: Khalid Salama
46
  Adapted implementation: Tannia Dubon