- NGCF-PT_ejlee(Neural Graph Collaborative Filtering Model with PyTorch)
- Reference: https://github.com/metahexane/ngcf_pytorch_g61
- My Repository URL: https://github.com/witheunjin/NGCF-PT_ejlee
|__ path: ~/NGCF-PT_ejlee/data
|__ /100K/ratings.csv: 100K 개의 Ratings data
|__ /1M/ratings.csv: 1M 개의 Ratings data
|__ /20M/ratings.csv: 20M 개의 Ratings data
|__ /25M/ratings.csv: 25M 개의 Ratings data
|__ /27M/ratings.csv: 27M 개의 Ratings data
*20M이상의 Dataset을 Training하려고 하면 'out of memory' 문제발생
~/NGCF-PT_ejlee에서 다음과 같은 명령어를 사용하여 실행
$ python run_ngcf.py
(1M의 Dataset에 대한 Training 결과(400개의 Epochs) 출력)$ python run_ngcf.py --data_size 100K
(100K dataset에 대한 Training 결과(400개의 Epochs) 출력)$ python run_ngcf.py --data_size 1M --n_epochs 100
(1M Dataset에 대한 Training 결과(100개의 Epochs) 출력)
- NGCF-PT_ejlee_100K_result: 100K Dataset에 대한 Training 결과(Epoch 100)
- NGCF-PT_ejlee_1M_result: 1M Dataset에 대한 Training 결과(Epoch 400)