package org.example.recommendation.evaluation;

import org.apache.predictionio.controller.Engine;
import org.apache.predictionio.controller.java.JavaEvaluation;
import org.apache.predictionio.core.BaseAlgorithm;
import org.example.recommendation.Algorithm;
import org.example.recommendation.DataSource;
import org.example.recommendation.PredictedResult;
import org.example.recommendation.Preparator;
import org.example.recommendation.PreparedData;
import org.example.recommendation.Query;
import org.example.recommendation.Serving;

import java.util.Collections;

public class EvaluationSpec extends JavaEvaluation {
    public EvaluationSpec() {
        this.setEngineMetric(
                new Engine<>(
                        DataSource.class,
                        Preparator.class,
                        Collections.<String, Class<? extends BaseAlgorithm<PreparedData, ?, Query, PredictedResult>>>singletonMap("algo", Algorithm.class),
                        Serving.class
                ),
                new PrecisionMetric()
        );
    }
}