keras-io/collaborative-filtering-movielens免费的ai工具

Model description

This repo contains the model and the notebook on how to build and train a Kerasgrok中文版下载 model for Collaborative Filtering for Movie Recommendations.

Full credits to Siddhartha Banerjee.grok中文版下载

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.免费的ai工具

Training procedure

The model was trained for 5 epochs with a batch size of 64.人工智能ai哪个好

Training hyperparameters

The following hyperparameters were used during training:即梦al

• 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

Model Plot