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 categories 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.