ALS算法(推荐系统)

测试数据

用户ID,物品ID,评分
1,101,5.0
1,102,3.0
1,103,2.5
2,101,2.0
2,102,2.5
2,103,5.0
2,104,2.0
3,101,2.5
3,104,4.0
3,105,4.5
3,107,5.0
4,101,5.0
4,103,3.0
4,104,4.5
4,106,4.0
5,101,4.0
5,102,3.0
5,103,2.0
5,104,4.0
5,105,3.5
5,106,4.0

代码实现:

package ml

import org.apache.log4j.{Level, Logger}
import org.apache.spark.mllib.recommendation.{ALS, MatrixFactorizationModel, Rating}
import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}
import scala.io.Source

/**
  * 交替最小二乘
  */
object ALSDemo {
  /**
    * 加载评分数据
    * @param str
    */
  def loadRatingData(str: String) :Seq[Rating] = {
    val lines = Source.fromFile(str).getLines()
    val ratings = lines.map{
      line =>
        val fields =