PUBDEV-1579: Fix weighted mean, wasn't reducing the sum of weights. W…

…ould only work for small data (1 chunk).
h2oai · Jun 30, 2015 · e4e260f · e4e260f
1 parent 10dcdb9
commit e4e260f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/h2o-core/src/main/java/water/util/FrameUtils.java b/h2o-core/src/main/java/water/util/FrameUtils.java
@@ -241,6 +241,9 @@ public  double weightedMean() {
         _wsum += w;
       }
     }
-    @Override public void reduce(WeightedMean mrt) { _wresponse += mrt._wresponse; }
+    @Override public void reduce(WeightedMean mrt) {
+      _wresponse += mrt._wresponse;
+      _wsum += mrt._wsum;
+    }
   }
 }