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)}

错误:方法predict：(user: Int，product: Int)Double的参数不足。

Answer 1

MatrixFactorizationModel.predict()使用(user, product)对的RDD或单个用户和产品对作为两个输入参数。请查看docs接口。

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

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