IMF-Pytorch Requirments torch==1.6.0 numpy==1.21.4 pandas==1.2.4 Usage Model Training python main.py --dataset DB15K --model IMF --cuda 0