MegaRec

1. Requirements

Here are our main environment dependencies for running the repository：

• NVIDIA-SMI 535.183.01
• cuda 12.2
• python 3.11.5
• pytorch 2.4.0
• recbole 1.2.0
• casual-conv1d 1.4.0
• timm 1.0.11

3. Datasets

This repository contains the ML-1M dataset. If you want to train our model on other datasets, they can also be found at RecDatasets.

4. Run

To reproduce the results reported in our paper, just run it:

python run_MegaRec.py

5. Model testing

If you want to directly check and reproduce our results, you can perform the test:

python test.py