IMF-Pytorch

Requirments

torch==1.6.0 numpy==1.21.4 pandas==1.2.4

Model Training

python main.py --dataset DB15K --model IMF --cuda 0

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
dataset		dataset
layers		layers
models		models
utils		utils
.DS_Store		.DS_Store
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
main.py		main.py