diff --git a/pom.xml b/pom.xml index c47c24c0..8b0c7853 100644 --- a/pom.xml +++ b/pom.xml @@ -32,7 +32,7 @@ be accompanied by a bump in version number, regardless of how minor the change. --> - 0.9.2 + 0.9.3 diff --git a/src/main/java/quickml/supervised/inspection/RandomForestDumper.java b/src/main/java/quickml/supervised/inspection/RandomForestDumper.java index c95464e1..20ce832d 100644 --- 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 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 index c14e280a..b0301ddb 100644 --- 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); }