A pytorch implementation for paper "Criterion-based Heterogeneous Collaborative Filtering for Multi-behavior Implicit Recommendation"
- pytorch 1.4
- loguru
- scipy
- sklearn
[BeiBei][https://github.com/chenchongthu/EHCF] [Taobao][https://github.com/chenchongthu/EHCF]
Here we use the GMF as the CF basic model.
How to train Beibei : python search.py 0 [GPU_ID]
How to train Taobao : python search.py 1 [GPU_ID]
Our model will be restored in the checkpoints.