DeepなFactorization Machinesの実装たち

[AkihikoWatanabe]

https://github.com/rixwew/pytorch-fm

[AkihikoWatanabe]

下記モデルが実装されているすごいリポジトリ。論文もリンクも記載されており、Factorization Machinesを勉強する際に非常に参考になると思う。MITライセンス。各手法はCriteoのCTRPredictionにおいて、AUC0.8くらい出ているらしい。

• Logistic Regression
• Factorization Machine
• Field-aware Factorization Machine
• Higher-Order Factorization Machines
• Factorization-Supported Neural Network
• Wide&Deep
• Attentional Factorization Machine
• Neural Factorization Machine
• Neural Collaborative Filtering
• Field-aware Neural Factorization Machine
• Product Neural Network
• Deep Cross Network
• DeepFM
• xDeepFM
• AutoInt (Automatic Feature Interaction Model)
• AFN(AdaptiveFactorizationNetwork Model)