4

我正在使用此处提供的 MovieLens 数据集为电影制作推荐系统:http: //grouplens.org/datasets/movielens/

为了计算这个推荐系统,我在 scala 中使用了 Flink 的 ML 库,特别是 ALS 算法 ( org.apache.flink.ml.recommendation.ALS)。

我首先将电影的评分映射到 a DataSet[(Int, Int, Double)],然后创建 atrainingSet和 a testSet(参见下面的代码)。

我的问题是当我将ALS.fit函数与整个数据集(所有评级)一起使用时没有错误,但如果我只删除一个评级,拟合函数不再起作用,我不明白为什么.

你有什么想法?:)

使用的代码:

Rating.scala

case class Rating(userId: Int, movieId: Int, rating: Double)

预处理.scala

object PreProcessing {

def getRatings(env : ExecutionEnvironment, ratingsPath : String): DataSet[Rating] = {
      env.readCsvFile[(Int, Int, Double)](
      ratingsPath, ignoreFirstLine = true,
      includedFields = Array(0,1,2)).map{r => new Rating(r._1, r._2, r._3)}
}

处理.scala

object Processing {
  private val ratingsPath: String = "Path_to_ratings.csv"

  def main(args: Array[String]) {

    val env = ExecutionEnvironment.getExecutionEnvironment

    val ratings: DataSet[Rating] = PreProcessing.getRatings(env, ratingsPath)

    val trainingSet : DataSet[(Int, Int, Double)] =
    ratings
     .map(r => (r.userId, r.movieId, r.rating))
     .sortPartition(0, Order.ASCENDING)
     .first(ratings.count().toInt)

    val als = ALS()
     .setIterations(10)
     .setNumFactors(10)
     .setBlocks(150)
     .setTemporaryPath("/tmp/tmpALS")

    val parameters = ParameterMap()
     .add(ALS.Lambda, 0.01) // After some tests, this value seems to fit the problem
     .add(ALS.Seed, 42L)

    als.fit(trainingSet, parameters)
  }
}

“但如果我只删除一个评级”

val trainingSet : DataSet[(Int, Int, Double)] =
  ratings
    .map(r => (r.userId, r.movieId, r.rating))
    .sortPartition(0, Order.ASCENDING)
    .first((ratings.count()-1).toInt)

错误 :

06/19/2015 15:00:24 CoGroup(在 org.apache.flink.ml.recommendation.ALS$.updateFactors(ALS.scala:570) 上的 CoGroup)(4/4)切换到 FAILED

java.lang.ArrayIndexOutOfBoundsException:5

在 org.apache.flink.ml.recommendation.ALS$BlockRating.apply(ALS.scala:358)

在 org.apache.flink.ml.recommendation.ALS$$anon$111.coGroup(ALS.scala:635)

在 org.apache.flink.runtime.operators.CoGroupDriver.run(CoGroupDriver.java:152)

...

4

1 回答 1

7

问题是first算子结合setTemporaryPathFlink的ALS实现的参数。为了理解这个问题,让我快速解释一下阻塞 ALS 算法是如何工作的。

交替最小二乘法的分块实现首先将给定的评分矩阵按用户和按项目划分为块。对于这些块,计算路由信息。该路由信息表示哪个用户/项目块分别从哪个项目/用户块接收哪个输入。之后,开始 ALS 迭代。

由于 Flink 的底层执行引擎是一个并行流数据流引擎,它尝试以流水线方式执行尽可能多的数据流部分。这需要让管道的​​所有运营商同时在线。这样做的好处是 Flink 避免了实现可能非常大的中间结果。缺点是可用内存必须在所有运行的算子之间共享。在单个DataSet元素(例如用户/项目块)的大小相当大的 ALS 的情况下,这是不希望的。

为了解决这个问题,如果你设置了一个temporaryPath. 路径定义了可以存储中间结果的位置。因此,如果您定义了一个临时路径,那么ALS首先计算用户块的路由信息​​并将它们写入磁盘,然后计算项目块的路由信息​​并将它们写入磁盘,最后但并非最不重要的是它启动 ALS它从临时路径中读取路由信息的迭代。

用户和项目块的路由信息​​的计算都取决于给定的评级数据集。在您的情况下,当您计算用户路由信息时,它将首先读取评级数据集并first对其应用运算符。运算符从基础数据集中first返回n-arbitrary 元素。现在的问题是 Flink 并没有存储这个first操作的结果来计算 item 的路由信息​​。相反,当你开始计算 item 的路由信息​​时,Flink 会从源头开始重新执行数据流。这意味着它从磁盘读取评级数据集并应用first操作员再次对其进行操作。first在许多情况下,与第一次操作的结果相比,这将为您提供一组不同的评级。因此,生成的路由信息​​不一致而ALS失败。

first您可以通过具体化运算符的结果并将此结果用作算法的输入来规避该问题ALS。该对象FlinkMLTools包含一个方法,该方法persist采用 a DataSet,将其写入给定路径,然后返回一个DataSet读取刚刚写入的新的DataSet. 这允许您分解生成的数据流图。

val firstTrainingSet : DataSet[(Int, Int, Double)] =
  ratings
    .map(r => (r.userId, r.movieId, r.rating))
    .first((ratings.count()-1).toInt)

val trainingSet = FlinkMLTools.persist(firstTrainingSet, "/tmp/tmpALS/training")

val als = ALS()
  .setIterations(10)
  .setNumFactors(10)
  .setBlocks(150)
  .setTemporaryPath("/tmp/tmpALS/")

val parameters = ParameterMap()
  .add(ALS.Lambda, 0.01) // After some tests, this value seems to fit the problem
  .add(ALS.Seed, 42L)

als.fit(trainingSet, parameters)

或者,您可以尝试保持未temporaryPath设置。然后以流水线方式执行所有步骤(路由信息计算和 als 迭代)。这意味着用户和项目路由信息计算都使用来自first操作员的相同输入数据集。

Flink 社区目前正在努力将算子的中间结果保存在内存中。这将允许固定first运算符的结果,以便不会计算两次,因此由于其不确定性而不会给出不同的结果。

于 2015-06-19T16:57:42.810 回答