This repository contains the code and dasaset (MovieLens-100k) to perform a graph convolutional matrix completion in order to predict movie ratings. The method is based based on Van der Berg et al. work, we use their source code which can be found here: Github repository.
First you will need to install the external libraries listed below. To obtain extended features by scrapping you can run the get_imdb_id.ipynb notebook. Then to load the data, visualize them, build features matrix and train the model you can run the ntds_project.ipynb notebook.
Some files were too big to be upload on github you can download them from this link : https://drive.google.com/drive/folders/1UCQ5SwlZunKpeFSAj8qcZ7t6myTOsYs1?usp=sharing
- Nessreddine Loudiy @Loudiy
- Stanislas Ducotterd @StanislasDucotterd
- Lamyae Omari Alaoui @lomarial
- Arthur Babey @arthurbabey
This project has been developped using Python 3.6.9
.
We use the following libraries with the mentionned version.
Tensorflow==1.8.0,
Scipy==1.2.1,
Numpy==1.16.2,
Pandas==0.25.2,
Networkx==2.2,
Pgeocode==0.2.0
. ├── README.md │ ├── movielens-100k data set │ ├── get_imdb_id.ipynb scrapping notebook │ ├── ntds_project.ipynb main notebook │ ├── model.py model implementation ├── data_utils.py util function ├── layers.py layers for model ├── metrics.py metrics function ├── run_gcmc.py training ├── preprocessing.py preprocessing function ├── utils.py util function ├── train_mini_batch.py mini_batch training ├── initialization.py variable initialization ├── __intit__.py