jaspajjr / wide-and-deep-learning-keras

Wide and Deep Learning implemented with Keras

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Wide and Deep Learning implemented with Keras

            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
                    Version 2, December 2004

 Copyright (C) 2004 Sam Hocevar 

 Everyone is permitted to copy and distribute verbatim or modified
 copies of this license document, and changing it is allowed as long
 as the name is changed.

            DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE
   TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION

  0. You just DO WHAT THE FUCK YOU WANT TO.

Requirements

  • Python >= 3.6
  • TensorFlow >= 1.6
  • Keras >= 2.0.0

Model Plot

Model Plot The model is based on Heng-Tze Cheng, et al. Wide & Deep Learning for Recommender Systems (2016)
I used UCI Machine Learning Repository: Adult Data Set as example data.
There are 8 categorical features so I put every one of them into a embedding layer.
And I just put the rest 5 continuous feature into a dense layer and concatenate it with all embedding layers.
Then add some dense layer and before sigmoid layer I concatente the dense layer output with logistic part input.

Best Test Set Accuracy

85.8%

About

Wide and Deep Learning implemented with Keras


Languages

Language:Python 100.0%