Fix bug related to missing attribute values

sanity · Jul 19, 2015 · 156b536 · 156b536
1 parent 968d0ee
commit 156b536
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pom.xml b/pom.xml
@@ -32,7 +32,7 @@
         be accompanied by a bump in version number, regardless of how minor the change.
     -->
 
-    <version>0.9.2</version>
+    <version>0.9.3</version>
 
 
     <repositories>

diff --git a/src/main/java/quickml/supervised/inspection/RandomForestDumper.java b/src/main/java/quickml/supervised/inspection/RandomForestDumper.java
@@ -68,7 +68,7 @@ public void summarizeModel(PrintStream out, RandomDecisionForest forest) {
 
     }
 
-    private static class TreeSummary {
+    public static class TreeSummary {
         private int splits;
         private Map<String, AttributeSummary> attributes = new HashMap<>();
 

diff --git a/src/main/java/quickml/supervised/tree/decisionTree/reducers/DTCatBranchReducer.java b/src/main/java/quickml/supervised/tree/decisionTree/reducers/DTCatBranchReducer.java
@@ -51,7 +51,7 @@ else if (acceptableMissingValue)
                 continue;
 
             if (cc == null) {
-                cc = new ClassificationCounter(attrVal);
+                cc = new ClassificationCounter(attrVal != null ? attrVal : MISSING_VALUE);
                 Serializable newKey = (attrVal != null) ? attrVal : MISSING_VALUE;
                 result.put(newKey, cc);
             }