Complete Evaluation
diff --git a/src/main/scala/CompleteEvaluation.scala b/src/main/scala/CompleteEvaluation.scala
new file mode 100644
index 0000000..e168872
--- /dev/null
+++ b/src/main/scala/CompleteEvaluation.scala
@@ -0,0 +1,13 @@
+package org.template.classification
+
+import io.prediction.controller.Evaluation
+import io.prediction.controller.MetricEvaluator
+
+object CompleteEvaluation extends Evaluation {
+ engineEvaluator = (
+ ClassificationEngine(),
+ MetricEvaluator(
+ metric = Accuracy(),
+ otherMetrics = Seq(Precision(0.0), Precision(1.0), Precision(2.0)),
+ outputPath = "best.json"))
+}
diff --git a/src/main/scala/PrecisionEvaluation.scala b/src/main/scala/PrecisionEvaluation.scala
index 6e53265..de80b76 100644
--- a/src/main/scala/PrecisionEvaluation.scala
+++ b/src/main/scala/PrecisionEvaluation.scala
@@ -6,6 +6,8 @@
case class Precision(label: Double)
extends OptionAverageMetric[EmptyEvaluationInfo, Query, PredictedResult, ActualResult] {
+ override def header: String = s"Precision(label = $label)"
+
def calculate(query: Query, predicted: PredictedResult, actual: ActualResult)
: Option[Double] = {
if (predicted.label == label) {
@@ -23,5 +25,3 @@
object PrecisionEvaluation extends Evaluation {
engineMetric = (ClassificationEngine(), new Precision(label = 1.0))
}
-
-
