package org.example.recommendation

import org.apache.predictionio.controller.EngineFactory
import org.apache.predictionio.controller.Engine

case class Query(
  user: String,
  num: Int
)

case class PredictedResult(
  itemScores: Array[ItemScore]
)

case class ActualResult(
  ratings: Array[Rating]
)

case class ItemScore(
  item: String,
  score: Double
)

object RecommendationEngine extends EngineFactory {
  def apply() = {
    new Engine(
      classOf[DataSource],
      classOf[Preparator],
      Map("als" -> classOf[ALSAlgorithm]),
      classOf[Serving])
  }
}