blob: 2b3bbab2605f1e23933bddcc2c1f67d16b4711fc [file] [log] [blame]
package io.prediction.examples.experimental.cleanupapp
import io.prediction.controller.P2LAlgorithm
import io.prediction.controller.Params
import org.apache.spark.SparkContext
import org.apache.spark.SparkContext._
import org.apache.spark.rdd.RDD
import grizzled.slf4j.Logger
//case class AlgorithmParams(mult: Int) extends Params
//class Algorithm(val ap: AlgorithmParams)
class Algorithm
extends P2LAlgorithm[TrainingData, Model, Query, PredictedResult] {
@transient lazy val logger = Logger[this.type]
def train(sc: SparkContext, data: TrainingData): Model = {
new Model
}
def predict(model: Model, query: Query): PredictedResult = {
// Prefix the query with the model data
PredictedResult(p = "")
}
}
class Model extends Serializable {
override def toString = "Model"
}