Source code for LIRDRec is located at src/models/lirdrec.py.
The script has been tested running under Python 3.7.11, with the following packages installed (along with their dependencies):
torch: 1.11.0+cu113pandas: 1.3.5pyyaml: 6.0numpy: 1.21.5scipy: 1.7.3sentence-transformers: 2.2.0
- The model is based on MMRec. Please refer to the datasets used in the MMRec framework on GitHub.
- Put your downloaded data (e.g.
baby) underdatadir. - Enter
srcfolder and run with:
python main.py -m lirdrec -d baby
You may specify other parameters in CMD or config withconfigs/model/LIRDRec.yamlandconfigs/dataset/*.yaml. For reproducibility, model settings are specified inconfigs/model/LIRDRec.yaml.