blob: 6468e4bbf8ef00158b22117362110f7f7b99995b [file] [log] [blame]
package org.template.recommendation
import io.prediction.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