mongospark

    // 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()