tensorflow-DeepFM

This project includes a Tensorflow implementation of DeepFM [1].

For usage, please see folder example.

Reference

[1] DeepFM: A Factorization-Machine based Neural Network for CTR Prediction, Huifeng Guo, Ruiming Tang, Yunming Yey, Zhenguo Li, Xiuqiang He.

Acknowledgments

This project gets inspirations from the following projects:

• He Xiangnan's neural_factorization_machine
• Jian Zhang's YellowFin (yellowfin optimizer is taken from here)

License

MIT