blob: 02a9ce58fd1b30dd9d65e0ad2c62d030bd576fbc [file] [log] [blame]
package org.template.recommendation
import io.prediction.controller.PPreparator
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, items = trainingData.items)
}
// HOWTO: added items(movies) list to prepared data to have possiblity to sort
// them in predict stage.
class PreparedData(val ratings: RDD[Rating], val items: RDD[(String, Item)])
extends Serializable