Recompute overall accuracy

As per #417
wcmc-its · Jul 24, 2020 · 19dadab · 19dadab
1 parent cc80c95
commit 19dadab
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/reciter/engine/ReCiterFeatureGenerator.java b/src/main/java/reciter/engine/ReCiterFeatureGenerator.java
@@ -117,7 +117,8 @@ public ReCiterFeature computeFeatures(UseGoldStandard mode,
         log.info("Precision=" + analysis.getPrecision());
         log.info("Recall=" + analysis.getRecall());
 
-        double accuracy = (analysis.getPrecision() + analysis.getRecall()) / 2.0;
+        double accuracy = ((analysis.getTruePositiveList().size() + analysis.getTrueNegativeList().size()) / (analysis.getTruePositiveList().size() + analysis.getTrueNegativeList().size() + analysis.getFalsePositiveList().size() + analysis.getFalseNegativeList().size()));
+
         log.info("Accuracy=" + accuracy);
 
         log.info("True Positive List [" + analysis.getTruePositiveList().size() + "]: " + analysis.getTruePositiveList());
@@ -127,8 +128,9 @@ public ReCiterFeature computeFeatures(UseGoldStandard mode,
         log.info("\n");
 
         // overall accuracy
-        reCiterFeature.setOverallAccuracy((analysis.getPrecision() + analysis.getRecall()) / 2);
-
+	 reCiterFeature.setOverallAccuracy((analysis.getTruePositiveList().size() + analysis.getTrueNegativeList().size())	/  
+(analysis.getTruePositiveList().size() + analysis.getTrueNegativeList().size() + analysis.getFalsePositiveList().size() + analysis.getFalseNegativeList().size()));
+
         // precision
         reCiterFeature.setPrecision(analysis.getPrecision());