Home

Awesome

MultiCBR

Pytorch implementation for "MultiCBR: Multi-view Contrastive Learning for Bundle Recommendation"

Environment

Run the code

To train MultiCBR on dataset NetEase with GPU 0, simply run:

python train.py -g 0 -m MultiCBR -d NetEase

You can indicate GPU id or dataset with cmd line arguments, and the hyper-parameters are recorded in config.yaml.