Downloads · 30 days
0
mf-movielens20m-baseline
AnjaliBankapur/mf-movielens20m-baseline
mf-movielens20m-baseline is a machine learning model from AnjaliBankapur. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for custom.
This repository contains the trained Matrix Factorization baseline model used in a recommender systems case study on the MovieLens 20M dataset.
Access
Public
Updated Mar 11, 2026
Repo size
40.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt40.7 MB · 100%
At a glance
- Library
- custom
- Access
- Public
- Created
- Mar 11, 2026
- Updated
- Mar 11, 2026
- SHA
- e9052cac
Model card
From the Hugging Face model README
Matrix Factorization (BPR) - MovieLens 20M
This repository contains the trained Matrix Factorization baseline model used in a recommender systems case study on the MovieLens 20M dataset.
Model
- Model type: Matrix Factorization with BPR objective
- Final saved model:
mf_bpr_model.pt
Dataset
- MovieLens 20M
Configuration
- Number of users: 138238
- Number of items: 20639
- Embedding dimension: 64
- Learning rate: 0.001
- Weight decay: 0.0
- Seed: 42
- Device: CPU
Training Summary
Training loss across 5 epochs:
- Epoch 1: 0.2113
- Epoch 2: 0.0996
- Epoch 3: 0.0849
- Epoch 4: 0.0754
- Epoch 5: 0.0689
Files
mf_bpr_model.pt- final trained modelconfig_mf_data.json- configuration/data settingsmf_val_tuning_results.csv- validation tuning results
- Library
- custom
- Created
- Mar 11, 2026
- Updated
- Mar 11, 2026