| package io.prediction.output.itemrec |
| |
| import io.prediction.commons.Config |
| import io.prediction.commons.modeldata.ItemRecScore |
| import io.prediction.commons.settings.{ Algo, App, OfflineEval } |
| import com.github.nscala_time.time.Imports._ |
| |
| object ItemRecCFAlgoBatchOutput { |
| private val config = new Config |
| |
| def output(uid: String, n: Int, |
| itypes: Option[Seq[String]], instant: DateTime)( |
| implicit app: App, algo: Algo, |
| offlineEval: Option[OfflineEval]): Seq[(String, Double)] = { |
| val itemRecScores = offlineEval map { _ => |
| config.getModeldataTrainingItemRecScores |
| } getOrElse config.getModeldataItemRecScores |
| itemRecScores.getTopNIidsAndScores(uid, n, itypes) |
| } |
| } |