Automatic Movie Recommendation System

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