Automatic Movie Recommendation System

carla 871e41fbad Add DAIT artifact to setup development environment. 4 years ago
dair-artifacts 871e41fbad Add DAIT artifact to setup development environment. 4 years ago
modelstrt b068085c47 Add movierec model zip 4 years ago
movierec 654ba84d75 Fix message in trt_client and set defeault db name in argparser. 4 years ago
test 79de7c4e50 Extract download logic to movielens_utils and download by default on 4 years ago
.gitignore 6d97522b25 Add cloning instructions to readme and CUDA to gitignore. 4 years ago
LICENSE b951fc0d9d Initial commit 4 years ago 5efbcc5cce Fix spacing in readme that was preventing proper formating of section 4 years ago


Automatic Movie Recommendation System using Tensorflow and TensorRT.

This project implements a Movie Recommendation System using MovieLens data.

The model and training and evaluation methods are based on Neural Collaborative Filtering (He et al. 2017).

The data processing, model and training modules leverage Tensorflow and Keras. There are scripts to export a trained model into a NVIDIA TensorRT compatible model, and to run a sample command-line client to request inferences to a model deployed and running in a NVIDIA TensorRT Inference Server.

How to Clone to a Local Repository

Make sure you have git installed on your local machine, you have an ssh key for and you have properly setup your ~/.ssh/config. (See support section below if you need help with that.)

Once you are ready to clone the repository, run the following command on your local machine:

git clone        

How to Get Support

Send email to