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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support