tree: 52ef046396d8ffd8b7d4c861f4508d734279abce [path history] [tgz]
  1. config.py
  2. data.py
  3. inference.py
  4. model.py
  5. README.md
  6. train.py
example/sparse/wide_deep/README.md

Wide and Deep Learning

The example demonstrates how to train wide and deep model. The Census Income Data Set that this example uses for training is hosted by the UC Irvine Machine Learning Repository. Tricks of feature engineering are adapted from tensorflow's wide and deep tutorial.

The final accuracy should be around 85%. For training:

  • python train.py

For inference:

  • python inference.py