影视作品集合

电影数据集

The MovieLens Dataset

存在多种数据集可供推荐研究使用。其中, MovieLens 数据集可能是最广为人知的数据集之一。它是一个基于网络平台的非营利性电影推荐系统,创建于1997年,由明尼苏达大学的研究实验室 GroupLens 管理,其目标是为了收集与电影评级相关的数据,从而对该系统的数据对包括个性化推荐与社会心理学在内的多个领域都具有重要意义

1. Getting the Data

该平台的数据集合由GroupLens研究团队维护管理。该平台提供多个版本供研究人员选择使用。为了方便研究工作我们决定采用MovieLens 100K版本的数据集合作为实验基础这一版本的数据集合涵盖了来自不同背景的943位用户的评分记录包含了从五星满分到一星评分的全部可能性并且涉及到了来自不同类型的电影共有1682部影片这些信息已经被经过清理工作后在推荐系统测试中每个用户的评分记录不少于20条这样的处理方式有助于提升实验结果的有效性同时也能保证研究工作的顺利进行

from d2l import mxnet as d2l

from mxnet import gluon, np

import os

import pandas as pd

然后，下载MovieLens
100k数据集并将交互作为DataFrame加载。

#@save

d2l.DA