Install | Getting Started | Documentation |
TorchCTR is a scalable and easy-to-use ML package for CTR (Click Through Rate) prediction and ranking in recommender system with Facebook PyTorch.
- Logistic Regression
- Factorization Machine by Osaka Univ.
- Field-aware Factorization Machine by Criteo, CMU & NTU
- 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
pip install torchctr --user
please check exmaples/