Source code for the proposed src/models/cm3.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.
sports) underdatadir. - Enter
srcfolder and run with
python main.py -m CM3 -d sports
You may specify other parameters in CMD or config withconfigs/model/*.yamlandconfigs/dataset/*.yaml.