This project is the implementation for 'Heterogeneous Graph-based Framework with Disentangled Representations Learning for Multi-target Cross Domain Recommendation'
Link to our paper: https://arxiv.org/abs/2407.00909
-
Make sure that your virtual env has the following packages:
- PyTorch (GPU Ver recommended)
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
- Numpy
- Pandas
- DGL (0.7.2 ver)
pip install dgl-cu113==0.7.2 -f https://data.dgl.ai/wheels/repo.html
- Tensorboard
- Scikit-Learn
- PyTorch (GPU Ver recommended)
-
download the project from the repository
-
download the Douban data from the url: https://github.com/fengzhu1/GA-DTCDR/tree/main/Data, and move the dataset to the dir: ./data/doudan
-
run the command in the virtual env: if you want to run the Douban data
python src/model_douban.py
-
the whole project support NVIDIA GPU ACCELERATION.