GraphSAGE Recommender - MovieLens 20M
This repository contains the trained GraphSAGE model used in a graph-based recommender system study.
Model
GraphSAGE Graph Neural Network for recommendation.
Dataset
MovieLens 20M
Files
- model_final.pt : trained GraphSAGE model
- run_config.json : training configuration
- train_log.csv : training metrics
- demo_user_recommendations.csv : example recommendations for users
Purpose
This model is part of a comparative study including:
- ItemKNN
- Matrix Factorization
- LightGCN
- GraphSAGE
