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.