@nivdul
#DV14 #MLwithSpark
// Evaluate the model
val userMovies = test.map {
case Rating(user, movie, rate) => (user, movie)
}
val predictions = model.predict(userMovies).map {
case Rating(user, movie, rate) => ((user, movie), rate)
}
!
val ratesAndPreds = test.map {
case Rating(user, movie, rate) => ((user, movie), rate)
}.join(predictions)
//measuring the Mean Squared Error of rating prediction
val MSE = ratesAndPreds.map { case ((user, movie), (r1, r2)) =>
val err = (r1 - r2)
err * err
}.mean()