// Get user recommendations | |
import sqlContext.implicits._ | |
val unratedMovies = movieRatings.filter(s"user_id != $userId").select("movie_id").distinct().map(r => | |
(userId, r.getAs[Int]("movie_id"))).toDF("user_id", "movie_id") | |
val recommendations = combinedModel.transform(unratedMovies) | |
// Convert the recommendations into UserMovieRatings | |
val userRecommendations = recommendations.map(r => | |
UserMovieRating(0, r.getAs[Int]("movie_id"), r.getAs[Float]("prediction").toInt)).toDF() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment