PUBDEV-1778: Fix major bug in DRF scoring - wrong metrics during mode…

…l building (were correct at the end).
h2oai · Jul 28, 2015 · dad2c4b · dad2c4b
1 parent 02f79e8
commit dad2c4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-algos/src/main/java/hex/tree/drf/DRFModel.java b/h2o-algos/src/main/java/hex/tree/drf/DRFModel.java
@@ -34,7 +34,7 @@ public static class DRFOutput extends SharedTreeModel.SharedTreeOutput {
    *  subclass scoring logic. */
   @Override protected double[] score0(double data[], double preds[], double weight, double offset) {
     super.score0(data, preds, weight, offset);
-    int N = _parms._ntrees;
+    int N = _output._ntrees;
     if (_output.nclasses() == 1) { // regression - compute avg over all trees
       preds[0] /= N;
     } else { // classification