Skip to content

arthurbabey/DeepMovieRecommender

Repository files navigation

DeepMovieRecommender

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.

How to run the code

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

Contributors

Python and external libraries

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

Repo structure

.
├── 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       

   

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published