Install | Getting Started | Documentation |
TorchCTR
TorchCTR is a scalable and easy-to-use ML package for CTR (Click Through Rate) prediction and ranking in recommender system with Facebook PyTorch.
Classic Models
- Logistic Regression
- Factorization Machine by Osaka Univ.
- Field-aware Factorization Machine by Criteo, CMU & NTU
Deep Learning Models
- Wide and Deep by Google
- DeepFM by Huawei & HIT
- Neural Factorization Machine by NUS
- Field-aware Neural Factorization Machine
- Factorization-supported Neural Network
- Product-based Neural Network
- Attentional Factorization Machine
- Deep and Cross Network
- Deep Interest Network
- eXtreme Deep Factorization Machine
Install
pip install torchctr --user
Getting Started
please check exmaples/