movielens

Movielens 1-M数据集。

Movielens 1-M数据集是由GroupLens Research采集的6000个用户对4000个电影的的100万个评级。 该模块将从 http://files.grouplens.org/datasets/movielens/ml-1m.zip 下载Movielens 1-M数据集,并将训练集和测试集解析为paddle reader creator。

  • paddle.dataset.movielens.get_movie_title_dict()[源代码]

获取电影标题词典。

获取电影ID的最大值。

获取用户ID的最大值。

获取职业ID的最大值。

  • paddle.dataset.movielens.movie_categories()[源代码]

获取电影类别词典。

获取用户信息词典。

获取电影信息词典。

  • paddle.dataset.movielens.convert(path)

将数据集转换为recordio格式。

  • class paddle.dataset.movielens.MovieInfo(index, categories, title)[源代码]

电影ID,标题和类别信息存储在MovieInfo中。

  • class paddle.dataset.movielens.UserInfo(index, gender, age, job_id)[源代码]

用户ID,性别,年龄和工作信息存储在UserInfo中。