Downloads · 30 days
0
0% of all-time downloads
collaborative-filtering-movielens-copy
mindwrapped/collaborative-filtering-movielens-copy
collaborative-filtering-movielens-copy is a tabular classification model from mindwrapped. Use it for the tabular classification task on the model card, and read the license before you ship it in a product. It is set up for keras. The card lists the license as cc0-1.0.
This repo contains the model and the notebook on how to build and train a Keras model for Collaborative Filtering for Movie Recommendations.
All-time downloads
152
Public
Repo size
6.5 MB
Likes
12
Trending 2
Repo makeup
Click a slice to open those files.
.data-00000-of-000016.3 MB · 97%
At a glance
- Task
- Tabular Classification
- Library
- keras
- License
- cc0-1.0
- Access
- Public
- Created
- Jun 8, 2022
- Updated
- Aug 4, 2022
- SHA
- a5224e48
Model card
From the Hugging Face model README
Model description
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.
Intended uses & limitations
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.
Training and evaluation data
The dataset consists of user's ratings on specific movies. It also consists of the movie's specific genres.
Training procedure
The model was trained for 5 epochs with a batch size of 64.
Training hyperparameters
The following hyperparameters were used during training:
- optimizer: {'name': 'Adam', 'learning_rate': 0.001, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
- training_precision: float32
Training Metrics
| Epochs | Train Loss | Validation Loss |
|---|---|---|
| 1 | 0.637 | 0.619 |
| 2 | 0.614 | 0.616 |
| 3 | 0.609 | 0.611 |
| 4 | 0.608 | 0.61 |
| 5 | 0.608 | 0.609 |
Model Plot
<details> <summary>View Model Plot</summary>
- Task
- Tabular Classification
- Library
- keras
- License
- cc0-1.0
- Created
- Jun 8, 2022
- Updated
- Aug 4, 2022