amandarp / wideanddeep

Example application of Google's Wide & Deep model implemented in R Keras

Wide & Deep Learning

Application of Google’s Wide & Deep network using R Keras. The model is described in the following paper:

Heng-Tze Cheng, Levent Koc, Jeremiah Harmsen, Tal Shaked, Tushar Chandra, Hrishi Aradhye, Glen Anderson, Greg Corrado, Wei Chai, Mustafa Ispir, Rohan Anil, Zakaria Haque, Lichan Hong, Vihan Jain, Xiaobing Liu, and Hemal Shah. 2016. Wide & Deep Learning for Recommender Systems. In Proceedings of the 1st Workshop on Deep Learning for Recommender Systems (DLRS 2016). Association for Computing Machinery, New York, NY, USA, 7–10. DOI:https://doi.org/10.1145/2988450.2988454

Application

Wine price prediction given rating review text and wine variety. R Keras model in wine-model.R. Mostly follows layout described in this TensorFlow blog post.