SAMOA-58: was not initializing arrays prior to use

Project: http://git-wip-us.apache.org/repos/asf/incubator-samoa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-samoa/commit/c743b7e0
Tree: http://git-wip-us.apache.org/repos/asf/incubator-samoa/tree/c743b7e0
Diff: http://git-wip-us.apache.org/repos/asf/incubator-samoa/diff/c743b7e0

Branch: refs/heads/master
Commit: c743b7e08d3174d247e534fa7fba2794093828af
Parents: 397de94
Author: edi_bice <[email protected]>
Authored: Mon Feb 22 15:56:51 2016 -0500
Committer: Gianmarco De Francisci Morales <[email protected]>
Committed: Tue Apr 19 11:33:42 2016 +0300

----------------------------------------------------------------------
 .../samoa/evaluation/F1ClassificationPerformanceEvaluator.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-samoa/blob/c743b7e0/samoa-api/src/main/java/org/apache/samoa/evaluation/F1ClassificationPerformanceEvaluator.java
----------------------------------------------------------------------
diff --git 
a/samoa-api/src/main/java/org/apache/samoa/evaluation/F1ClassificationPerformanceEvaluator.java
 
b/samoa-api/src/main/java/org/apache/samoa/evaluation/F1ClassificationPerformanceEvaluator.java
index b59480a..3046360 100644
--- 
a/samoa-api/src/main/java/org/apache/samoa/evaluation/F1ClassificationPerformanceEvaluator.java
+++ 
b/samoa-api/src/main/java/org/apache/samoa/evaluation/F1ClassificationPerformanceEvaluator.java
@@ -33,7 +33,7 @@ public class F1ClassificationPerformanceEvaluator extends 
AbstractMOAObject impl
         ClassificationPerformanceEvaluator {
 
     private static final long serialVersionUID = 1L;
-    protected int numClasses;
+    protected int numClasses = -1;
 
     protected long[] support;
     protected long[] truePos;
@@ -63,6 +63,7 @@ public class F1ClassificationPerformanceEvaluator extends 
AbstractMOAObject impl
 
     @Override
     public void addResult(Instance inst, double[] classVotes) {
+        if (numClasses==-1) reset(inst.numClasses());
         int trueClass = (int) inst.classValue();
         this.support[trueClass] += 1;
         int predictedClass = Utils.maxIndex(classVotes);

Reply via email to