SAVRN
Search Contact SAVRN

Organization

Keras

keras-io

Reproducible Open-Source Machine Learning

Models in Library1
Datasets in Library0
Models on Hugging Face104
Followers329

Models

Model · Tabular classification

collaborative-filtering-movielens

Keras

This repo contains the model and the notebook on how to build and train a Keras model for Collaborative Filtering for Movie Recommendations. Full credits to Siddhartha Banerjee. Based on a user and movies they have rated highly in the past, this model outputs the predicted rating a user would give to a movie they haven't seen yet (between 0-1). This information can be used to find out the top recommended movies for this user. The dataset consists of user's ratings on specific movies. It also consists of the movie's specific genres. The model was trained for 5 epochs with a batch size of 64. The following hyperparameters were used during training: - trainingprecision: float32

Open weights cc0-1.0 tf-keras