Mllib ALS модель предсказать Ошибка

Question

Облицовка ошибки для метода model.predict MatrixFactorizationModel говоря, пытаясь скомпилировать в scala.

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, product: Int) Double.

Answer 1

MatrixFactorizationModel.predict() принимает либо RDD (user, product) пары, либо пару одного пользователя и продукта как два входных аргумента. Проверьте API docs.

В коде, вы должны изменить его на:

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