apache-spark - Mllib ALS 模型预测误差

Question

尝试在 scala 中进行编译时，model.predict 方法 MatrixFactorizationModel 遇到错误。

 val usersProducts = ratings.map{ case Rating(user, product, rate) => (user, product,rate) }
 val predictions =   model.predict(usersProducts).map{ case Rating(user, product, rate) => ((user, product), rate)}

错误:方法预测参数不足:(用户:Int，产品:Int)Double。

Answer 1

MatrixFactorizationModel.predict() 采用(用户，产品) 对的 RDD 或单个用户和产品对作为两个输入参数。请查看API docs .

在您的代码中，您应该将其更改为:

val usersProducts = ratings.map{ case Rating(user, product, rate) => (user, product) }