CODEWITHSUNDEEP
apache-sparkmachine-learningspark-streamingapache-spark-mllib
Subhod Lagade
Subhod Lagade

Reputation: 113

Mllib ALS model predict Error

Facing error for model.predict method MatrixFactorizationModel saying while trying to compile in 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)}

Error : not enough arguments for method predict: (user: Int, product: Int)Double.

Upvotes: 0

Views: 393

Answers (1)

Xiangrui Meng
Xiangrui Meng

Reputation: 46

MatrixFactorizationModel.predict() takes either an RDD of (user, product) pairs or a single user and product pair as two input arguments. Please check the API docs.

In your code, you should change it to:

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

Upvotes: 2

Related Questions