#机器学习# 文章列表 浅入浅出:PageRank算法 使用 TextRank 算法为文本生成关键字和摘要 基于物品的协同过滤 如何使用MapReduce实现基于物品的协同过滤(1) 如何使用MapReduce实现基于物品的协同过滤(2) 浅入浅出:K近邻算法 使用mahout下的朴素贝叶斯分类器对新闻分类 使用Affinity Propagation进行聚类 K-medoids聚类 矩阵分解在推荐系统中的应用:NMF和经典SVD实战 使用特征递归消除筛选特征 如何分配权重 比较NMF、PCA和VQ 方差和协方差 基于SVD的协同过滤 逻辑斯谛回归代码实现 隐语义模型和NMF(非负矩阵分解) 使用PCA处理MNIST数据集 使用GBDT选取特征 基于贝叶斯的文本分类系统的数据库设计 在hadoop1.2.1上安装mahout 0.9 Hadoop 2.4 实现Kmeans聚类算法 在Iris数据集上对比PCA、LDA、NMF 基于贝叶斯的文本分类实战 单层决策树 Logistic regression(逻辑斯蒂回归) 基于用户的协同过滤 词袋模型与文档-词矩阵 如何实现拼音与汉字的互相转换 梯度下降法 如何判定相似度 MovieLens数据集介绍 基于KNN的文本分类实战 Jasper文本分类系列博客阅读摘录 使用 Mean Shift进行聚类 朴素贝叶斯的三个常用模型:高斯、多项式、伯努利 使用决策树处理iris数据集 浅入浅出:从Kmeans到Kmeans++ 如何持久化scikit-learn中训练好的模型 浅入浅出:DBSCAN聚类算法(1) 浅入浅出:DBSCAN聚类算法(2) 2015阿里移动推荐算法比赛第一赛季总结 爬山算法 使用朴素贝叶斯分类器划分邮件 层次聚类 基于MapReduce的频繁项集挖掘 搜狗实体关系提取比赛

MovieLens数据集介绍


#机器学习#


2014-11-20

MovieLens数据集保存了用户对电影的评分。基于这个数据集,我们可以测试一些推荐算法、评分预测算法。

MovieLens 100k

该数据集记录了943个用户对1682部电影的共100,000个评分,每个用户至少对20部电影进行了评分。

**文件u.info**保存了该数据集的概要:

943 users
1682 items
100000 ratings

**文件u.item**保存了item的信息,也就是电影的信息,共1682部电影,其id依次是1、2、......、1682。文件中每一行保存了一部电影的信息,格式如下:

movie id | movie title | release date | video release date |
IMDb URL | unknown | Action | Adventure | Animation |
Children's | Comedy | Crime | Documentary | Drama | Fantasy |
Film-Noir | Horror | Musical | Mystery | Romance | Sci-Fi |
Thriller | War | Western |

注意,最后19个字段保存的是该电影的类型,一个字段对应一个类型,值为0代表不属于该类型,值为1代表属于该类型,类型信息保存在文件u.genre中。

随便浏览了一下u.item文件,发现基本是20世纪90年代的电影。

**文件u.genre**保存了电影的类型信息。

**文件u.user**保存了用户的信息,共有943个用户,其id依次是1、2、......、943。文件中每一行保存了一个用户的信息,格式如下:

user id | age | gender | occupation | zip code

**文件u.occupation**保存了用户职业的集合。

下面介绍数据集的主要文件。

**文件u.data**保存了所有的评分记录,每一行是一个用户对一部电影的评分,共有100000条记录。当然,如果某用户没有对某电影评分,则不会包含在该文件中。评分的分值在1到5之间,就是1、2、3、5这5个评分。每一行格式如下:

user id | item id | rating | timestamp

其中,item id就是电影的id,时间戳timestamp是评分时间。我转换了下时间戳,也是在20世纪90年代。

文件u1.base文件u1.test放在一起就是文件u.data。将u.data按照80%/20%的比例分成u1.baseu1.test,可以将u1.base作为训练集,u1.test作为测试集。u2u3u4u5系列文件和u1类似。u1u2u3u4u5的测试集是不相交的,它们可以用来做**(5折交叉验证)5 fold cross validation**。

**文件ua.base文件ua.test**也是由u.data拆分而来,在ua.test中包含了每个用户对10部电影的评分,从u.data去掉ua.test得到ua.baseub.baseub.test也使用了同样的生成方法。另外,ua.testub.test是不相交的。

MovieLens 1M

该数据集保存的是6040个用户对3952部电影的1000209个评分记录。具体可以参考其README文件。

MovieLens 10M

71567个用户,10681部电影,10000054条评分记录,同时多了个用户为电影设置的标签。具体可以阅读其中的README.html

Tag Genome

该数据集下有三个数据文件。

**movies.dat:**其每一行的格式是:

<MovieID><Title><MoviePopularity>

MoviePopularity是在MovieLens中对该电影的评分次数。

**tag.dat:**每一行的格式是:

<TagID><Tag><TagPopularity> 

<TagPopularity> 是使用该Tag的用户数,一个用户最多算1次。

tag_relevance.dat:每一行的格式是:

<MovieID><TagID><Relevance>

<Relevance>的值在0和1之间,值越大,Tag与Movie的关联性越强。


( 本文完 )