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