blob: d242243e6fda0549e1683eb6abd9fff7b88e586e [file] [log] [blame]
package io.prediction.output.itemrec
import io.prediction.commons.modeldata.ItemRecScore
import io.prediction.commons.settings.{ App, Algo, OfflineEval }
import com.github.nscala_time.time.Imports._
object ItemRecCFAlgoOutput extends ItemRecAlgoOutput {
override def output(uid: String, n: Int, itypes: Option[Seq[String]],
instant: DateTime)(implicit app: App, algo: Algo, offlineEval: Option[OfflineEval]): Seq[(String, Double)] = {
/** FIXME(someone). Ugly hack for realtime algorithms. **/
if (algo.infoid == "pio-itemrec-single-featurebased_realtime") {
ItemRecAlgoRealtimeOutput.output(uid, n, itypes, instant)
} else {
ItemRecCFAlgoBatchOutput.output(uid, n, itypes, instant)
}
}
}