Update some comments

amplab · May 9, 2015 · 22d79ec · 22d79ec
1 parent 8e422e3
commit 22d79ec
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/scala/evaluation/MeanAveragePrecisionEvaluator.scala b/src/main/scala/evaluation/MeanAveragePrecisionEvaluator.scala
@@ -3,6 +3,10 @@ package evaluation
 import org.apache.spark.rdd.RDD
 import org.apache.spark.SparkContext._
 
+/**
+ * MeanAveragePrecision (MAP) calculator based on MATLAB code
+ * from the encoding eval toolkit at http://www.robots.ox.ac.uk/~vgg/software/enceval_toolkit/
+ */
 object MeanAveragePrecisionEvaluator {
 
   /**
@@ -59,7 +63,7 @@ object MeanAveragePrecisionEvaluator {
   }
 
   /**
-   * Get average precision at 10 levels given precisions and recalls
+   * Get average precision at 11 levels given precisions and recalls
    */
   private def getAP(precisions: Array[Double], recalls: Array[Double]) = {
     var ap = 0.0