movielens

Movielens 1-M dataset.

Movielens 1-M dataset contains 1 million ratings from 6000 users on 4000 movies, which was collected by GroupLens Research. This module will download Movielens 1-M dataset from http://files.grouplens.org/datasets/movielens/ml-1m.zip and parse training set and test set into paddle reader creators.

paddle.dataset.movielens.get_movie_title_dict()[source]

Get movie title dictionary.

paddle.dataset.movielens.max_movie_id()[source]

Get the maximum value of movie id.

paddle.dataset.movielens.max_user_id()[source]

Get the maximum value of user id.

paddle.dataset.movielens.max_job_id()[source]

Get the maximum value of job id.

paddle.dataset.movielens.movie_categories()[source]

Get movie categoriges dictionary.

paddle.dataset.movielens.user_info()[source]

Get user info dictionary.

paddle.dataset.movielens.movie_info()[source]

Get movie info dictionary.

class paddle.dataset.movielens.MovieInfo(index, categories, title)[source]

Movie id, title and categories information are stored in MovieInfo.

class paddle.dataset.movielens.UserInfo(index, gender, age, job_id)[source]

User id, gender, age, and job information are stored in UserInfo.