首页 > 代码库 > 第三篇：一个Spark推荐系统引擎的实现
第三篇：一个Spark推荐系统引擎的实现
前言
经过2节对MovieLens数据集的学习，想必读者对MovieLens数据集认识的不错了；同时也顺带回顾了些Spark编程技巧，Python数据分析技巧。
而本节将是让人兴奋的一节，它将实现一个基于Spark的推荐系统引擎。
关于推荐算法相关的知识，请读者先自行学习，本文仅仅介绍基于ALS矩阵分解推荐引擎的Spark实现。
PS: 全文示例将采用Scala语言。
第一步：提取有效特征
1. 首先，启动spark-shell并分配足够内存：
2. 载入用户对影片的评级数据：
1 // 载入评级数据2 val rawData = http://www.mamicode.com/sc.textFile("/home/kylin/ml-100k/u.data")3 // 展示一条记录4 rawData.first()
结果为：
3. 切分记录并返回新的RDD：
1 // 格式化数据集2 val rawRatings = rawData.map(_.split("\t").take(3))3 // 展示一条记录4 rawRatings.first()
4. 接下来需要将评分矩阵RDD转化为Rating格式的RDD：
1 // 导入rating类2 import org.apache.spark.mllib.recommendation.Rating3 // 将评分矩阵RDD中每行记录转换为Rating类型4 val ratings = rawRatings.map { case Array(user, movie, rating) => Rating(user.toInt, movie.toInt, rating.toDouble) }
这是因为MLlib的ALS推荐系统算法包只支持Rating格式的数据集。
第二步：训练推荐模型
接下来可以进行ALS推荐系统模型训练了。MLlib中的ALS算法接收三个参数：
a) rank：对应的是隐因子的个数，这个值设置越高越准，但是也会产生更多的计算量。一般将这个值设置为10-200；
b) iterations：对应迭代次数，一般设置个10就够了；
c) lambda：该参数控制正则化过程，其值越高，正则化程度就越深。一般设置为0.01。
1. 首先，执行以下代码，启动ALS训练：
1 // 导入ALS推荐系统算法包2 import org.apache.spark.mllib.recommendation.ALS3 // 启动ALS矩阵分解4 val model = ALS.train(ratings, 50, 10, 0.01)
这步将会使用ALS矩阵分解算法，对评分矩阵进行分解，且隐特征个数设置为50，迭代10次，正则化参数设为了0.01。
相对其他步骤，训练耗费的时间最多。运行结果如下：
2. 返回类型为MatrixFactorizationModel对象，它将结果分别保存到两个(id,factor)RDD里面，分别名为userFeatures和productFeatures。
也就是评分矩阵分解后的两个子矩阵：
上面展示了id为4的用户的“隐因子向量”。请注意ALS实现的操作都是延迟性的转换操作。
第三步：使用ALS推荐模型
1. 预测用户789对物品123的评分：
2. 为用户789推荐前10个物品：
1 val userId = 7892 val K = 103 4 // 获取推荐列表5 val topKRecs = model.recommendProducts(userId, K)6 // 打印推荐列表7 println(topKRecs.mkString("\n"))
结果为：
3. 初步检验推荐效果
获取到各个用户的推荐列表后，想必大家都想先看看用户评分最高的电影，和给他推荐的电影是不是有相似。
3.1 创建电影id - 电影名字典：
1 // 导入电影数据集2 val movies = sc.textFile("/home/kylin/ml-100k/u.item")3 // 建立电影id - 电影名字典4 val titles = movies.map(line => line.split("\\|").take(2)).map(array => (array(0).toInt, array(1))).collectAsMap()5 // 查看id为123的电影名6 titles(123)
结果为：
这样后面就可以根据电影的id找到电影名了。
3.2 获取某用户的所有观影记录并打印：
1 // 建立用户名-其他RDD，并仅获取用户789的记录2 val moviesForUser = ratings.keyBy(_.user).lookup(789)3 // 获取用户评分最高的10部电影，并打印电影名和评分值4 moviesForUser.sortBy(-_.rating).take(10).map(rating => (titles(rating.product), rating.rating)).foreach(println)
结果为：
3.3 获取某用户推荐列表并打印：
读者可以自行对比这两组列表是否有相似性。
第四步：物品推荐
很多时候还有另一种需求：就是给定一个物品，找到它的所有相似物品。
遗憾的是MLlib里面竟然没有包含内置的函数，需要自己用jblas库来实现 = =#。
1. 导入jblas库矩阵类，并创建一个余弦相似度计量函数：
1 // 导入jblas库中的矩阵类2 import org.jblas.DoubleMatrix3 // 定义相似度函数4 def cosineSimilarity(vec1: DoubleMatrix, vec2: DoubleMatrix): Double = {5 vec1.dot(vec2) / (vec1.norm2() * vec2.norm2())6 }
2. 接下来获取物品(本例以物品567为例)的因子特征向量，并将它转换为jblas的矩阵格式：
1 // 选定id为567的电影2 val itemId = 5673 // 获取该物品的隐因子向量4 val itemFactor = model.productFeatures.lookup(itemId).head5 // 将该向量转换为jblas矩阵类型6 val itemVector = new DoubleMatrix(itemFactor)
3. 计算物品567和所有其他物品的相似度：
1 // 计算电影567与其他电影的相似度 2 val sims = model.productFeatures.map{ case (id, factor) => 3 val factorVector = new DoubleMatrix(factor) 4 val sim = cosineSimilarity(factorVector, itemVector) 5 (id, sim) 6 } 7 // 获取与电影567最相似的10部电影 8 val sortedSims = sims.top(K)(Ordering.by[(Int, Double), Double] { case (id, similarity) => similarity }) 9 // 打印结果10 println(sortedSims.mkString("\n"))
结果为：
其中0.999999当然就是自己跟自己的相似度了。
4. 查看推荐结果：
1 // 打印电影567的影片名2 println(titles(567))3 // 获取和电影567最相似的11部电影(含567自己)4 val sortedSims2 = sims.top(K + 1)(Ordering.by[(Int, Double), Double] { case (id, similarity) => similarity })5 // 再打印和电影567最相似的10部电影6 sortedSims2.slice(1, 11).map{ case (id, sim) => (titles(id), sim) }.mkString("\n")
结果为：
看看，这些电影是不是和567相似？
第五步：推荐效果评估
在Spark的ALS推荐系统中，最常用到的两个推荐指标分别为MSE和MAPK。其中MSE就是均方误差，是基于评分矩阵的推荐系统的必用指标。那么MAPK又是什么呢？
它称为K值平均准确率，最多用于TopN推荐中，它表示数据集范围内K个推荐物品与实际用户购买物品的吻合度。具体公式请读者自行参考有关文档。
本文推荐系统就是一个[基于用户-物品评分矩阵的TopN推荐系统]，下面步骤分别用来获取本文推荐系统中的这两个指标。
PS：记得先要导入jblas库。
1. 首先计算MSE和RMSE：
1 // 创建用户id-影片id RDD 2 val usersProducts = ratings.map{ case Rating(user, product, rating) => (user, product)} 3 // 创建(用户id,影片id) - 预测评分RDD 4 val predictions = model.predict(usersProducts).map{ 5 case Rating(user, product, rating) => ((user, product), rating) 6 } 7 // 创建用户-影片实际评分RDD，并将其与上面创建的预测评分RDD join起来 8 val ratingsAndPredictions = ratings.map{ 9 case Rating(user, product, rating) => ((user, product), rating)10 }.join(predictions)11 12 // 导入RegressionMetrics类13 import org.apache.spark.mllib.evaluation.RegressionMetrics14 // 创建预测评分-实际评分RDD15 val predictedAndTrue = ratingsAndPredictions.map { case ((user, product), (actual, predicted)) => (actual, predicted) }16 // 创建RegressionMetrics对象17 val regressionMetrics = new RegressionMetrics(predictedAndTrue)18 19 // 打印MSE和RMSE20 println("Mean Squared Error = " + regressionMetrics.meanSquaredError)21 println("Root Mean Squared Error = " + regressionMetrics.rootMeanSquaredError)
基本原理是将实际评分-预测评分扔到RegressionMetrics类里，该类提供了mse和rmse成员，可直接输出获取。
结果为：
2. 计算MAPK：
// 创建电影隐因子RDD，并将它广播出去val itemFactors = model.productFeatures.map { case (id, factor) => factor }.collect()val itemMatrix = new DoubleMatrix(itemFactors)val imBroadcast = sc.broadcast(itemMatrix) // 创建用户id - 推荐列表RDDval allRecs = model.userFeatures.map{ case (userId, array) => val userVector = new DoubleMatrix(array) val scores = imBroadcast.value.mmul(userVector) val sortedWithId = scores.data.zipWithIndex.sortBy(-_._1) val recommendedIds = sortedWithId.map(_._2 + 1).toSeq (userId, recommendedIds)} // 创建用户 - 电影评分ID集RDDval userMovies = ratings.map{ case Rating(user, product, rating) => (user, product) }.groupBy(_._1) // 导入RankingMetrics类import org.apache.spark.mllib.evaluation.RankingMetrics// 创建实际评分ID集-预测评分ID集 RDDval predictedAndTrueForRanking = allRecs.join(userMovies).map{ case (userId, (predicted, actualWithIds)) => val actual = actualWithIds.map(_._2) (predicted.toArray, actual.toArray)}// 创建RankingMetrics对象val rankingMetrics = new RankingMetrics(predictedAndTrueForRanking)// 打印MAPKprintln("Mean Average Precision = " + rankingMetrics.meanAveragePrecision)
结果为：
比较坑的是不能设置K，也就是说，计算的实际是MAP...... 正如属性名：meanAveragePrecision。
小结
感觉MLlib的推荐系统真的很一般，一方面支持的类型少 - 只支持ALS；另一方面支持的推荐系统算子也少，连输出个RMSE指标都要写好几行代码，太不方便了。
唯一的好处是因为接近底层，所以可以让使用者看到些实现的细节，对原理更加清晰。
第三篇：一个Spark推荐系统引擎的实现