| package org.example.recommendation | |
| import org.apache.predictionio.controller.PPreparator | |
| import org.apache.spark.SparkContext | |
| import org.apache.spark.SparkContext._ | |
| import org.apache.spark.rdd.RDD | |
| class Preparator | |
| extends PPreparator[TrainingData, PreparedData] { | |
| def prepare(sc: SparkContext, trainingData: TrainingData): PreparedData = { | |
| new PreparedData(ratings = trainingData.ratings) | |
| } | |
| } | |
| class PreparedData( | |
| val ratings: RDD[Rating] | |
| ) extends Serializable |