SAVRN
Search Contact SAVRN

Open-weight model · Tabular classification

collaborative-filtering-movielens

by Keras keras-io/collaborative-filtering-movielens

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.

Parameters
Context
Weights126.4 KB
Licensecc0-1.0
AccessOpen weights
Monthly Downloads34

Model Card

By Keras, published under cc0-1.0, revision 4b1456f5b1fb.

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

Read Keras's full model card

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
View Model Plot ![Model Image](./model.png)

Identity and Version

Repository
keras-io/collaborative-filtering-movielens
Publisher
Keras
Task
Tabular classification
Modality
Tabular
Library
tf-keras
Parameters
Not stated by the source
Languages
tf-keras
Revision
4b1456f5b1fb82a652c3a2869735c35afdce7245
First published
2022-06-02
Last updated
2024-07-05

Files and Weights

10 files, 6.5 MB in total. The weights are 2 files totalling 126.4 KB in pb.

Weights2 files · 126.4 KB
Configuration1 file · 277 B
Documentation1 file · 1.5 KB
Other5 files · 6.4 MB
Repository1 file · 1.4 KB
Every file
FileTypeSizeSHA-256
keras_metadata.pbWeights4.9 KB 02876faecfbd
saved_model.pbWeights121.5 KB a900821259ed
history.jsonConfiguration277 B
README.mdDocumentation1.5 KB
logs/train/events.out.tfevents.1654143210.eea0d1945de5.451.0.v2Other86.3 KB 13999264415d
logs/validation/events.out.tfevents.1654143224.eea0d1945de5.451.1.v2Other818 B 1df9823a5ab6
model.pngOther1.7 KB 2a5c0aa3e028
variables/variables.data-00000-of-00001Other6.3 MB f6a93be4b60b
variables/variables.indexOther1.4 KB 94cbd7aba07d
.gitattributesRepository1.4 KB

License and Download

License
cc0-1.0
Access
Open weights, no gate
Download size
126.4 KB
Download from Keras

Released by Keras through its official repository on Hugging Face. Read the license.

Memory Requirements

PrecisionWeights in memory
As published126.4 KB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About collaborative-filtering-movielens

What license is collaborative-filtering-movielens released under?

cc0-1.0, as its publisher declares it. Read the license text before commercial use.

Similar Models

Model · Tabular classification

Nori-30M

Synthefy

Nori-30M is the ~29.2M-parameter variant of Nori, a tabular foundation model for regression via in-context learning (ICL). Given a few labeled rows as context, it predicts on new query rows in a single forward pass, with no task-specific training or fine-tuning. The model is trained entirely on synthetic data. Mean and median R² across 96 regression tasks from three public benchmark suites, on the same protocol as the base Nori: Stronger than the ~6M base on every suite. Evaluated with the bundled default inference config and the large-GPU protocol (up to 50k context rows per dataset). Paste this into Claude Code, Cursor, or any AI coding assistant and it will wire python from synthefynori…

Open weights apache-2.0 synthefy-nori

Model · Tabular classification

sap-rpt-1-oss

SAP

Go to SAP-RPT Playground ↗ Note: This model and repository were formerly known as ConTextTab. While the code and repository have now been updated in line with the new name sap-rpt-1-oss, the model checkpoint and functionality remain identical. Implementation of the deep learning model with the inference pipeline described in the paper "ConTextTab: A Semantics-Aware Tabular In-Context Learner". Tabular in-context learning (ICL) has recently achieved state-of-the-art (SOTA) performance on several tabular prediction tasks. Previously restricted to classification problems on small tables, recent advances such as TabPFN and TabICL have extended its use to larger datasets. While being…

Access requested at publisher apache-2.0 sap-rpt-1-oss

Model · Tabular classification

Nori

Synthefy

Nori is a tabular foundation model for regression via in-context learning (ICL). Given a few labeled rows as context, it predicts on new query rows in a single forward pass, with no task-specific training or fine-tuning. The model is trained entirely on synthetic data. Mean and median R² of the base model across 96 regression tasks from three public benchmark suites (single H200, up to 50K context rows per dataset): Large-N / long-context tables (common in TabArena) are the current focus of the large-table training stages. These numbers are reproducible end-to-end with one command — see Reproducing these numbers. Paste this into Claude Code, Cursor, or any AI coding assistant and it will…

Open weights apache-2.0 synthefy-nori

Model · Tabular classification

EXAONE-Tabular

LG AI Research

EXAONE Tabular is a transformer-based foundation model for tabular data that solves classification and regression through in-context learning: you pass the labeled rows to fit and the model predicts new rows in a single forward pass — no gradient updates and no per-dataset training. This repository is the exaonetabular inference runtime — a self-contained package that loads a released checkpoint and serves predictions through a small, scikit-learn-style API. The code here is permissively licensed; the released weights are non-commercial — see Both checkpoints are released: EXAONETabularClassifier and EXAONETabularRegressor each fetch their own weights with a single frompretrained() call.…

Open weights other

Model · Tabular classification

TabPFN-v2-clf

Prior Labs

TabPFN is a transformer-based foundation model for tabular data that leverages prior-data based learning to achieve strong performance on small tabular datasets without requiring task-specific training. For detailed usage examples and best practices, check out: - Python ≥ 3.9 - PyTorch ≥ 2.1 - scikit-learn ≥ 1.0 This repository hosts the production TabPFN-v2 base checkpoints. Files matching the pattern tabpfn-v2-classifier-finetuned-.ckpt are content-identical aliases of the corresponding base checkpoints (e.g. tabpfn-v2-classifier-finetuned-gn2p4bpt-xp6f0iqb.ckpt is identical to tabpfn-v2-classifier-gn2p4bpt.ckpt; tabpfn-v2-classifier-finetuned-zk73skhh.ckpt is identical to…

Open weights other tabpfn

Model · Tabular classification

tabfm-1.0.0-pytorch

Google

TabFM is a zero-shot tabular foundation model from Google Research. It supports classification and regression on structured/tabular data with mixed numerical and categorical columns, requiring no fine-tuning or hyperparameter search - training examples are passed as context and predictions are made in a single forward pass. This repository contains the PyTorch weights. For the JAX/Flax weights see You can also load directly using the HuggingFace Hub API: Developed by the Google Research team. - Tabular data with numerical and/or categorical columns - Binary and multiclass classification (up to 10 classes) TabFM uses alternating row and column attention to capture both feature interactions…

Open weights other tabfm