blob: 8f2f7e4b9dd9b5932b1c2446c746dd25b33e074e [file] [log] [blame]
package org.template.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