Movie recommender system using Graph Neural Network.
Libraries used for GNN:
- PyTorch
- torch_geometric
Generate intermediate dataset
python ./src/dataset.py
Training the model
python ./src/train.py
Trained models are located in models
Evaluating the model
python ./benchmark/evaluation.py