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"
}