Fixed a bug in merge mehotd of fm_predict

myui · Nov 17, 2015 · f90a7f2 · f90a7f2
1 parent 2906b38
commit f90a7f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/java/hivemall/fm/FMPredictGenericUDAF.java b/core/src/main/java/hivemall/fm/FMPredictGenericUDAF.java
@@ -298,8 +298,8 @@ void merge(final double o_ret, @Nullable final Object o_sumVjXj, @Nullable final
                 Object o2 = sumV2X2OI.getListElement(o_sumV2X2, f);
                 double d1 = doubleOI.get(o1);
                 double d2 = doubleOI.get(o2);
-                sumVjXj[f] = d1;
-                sumV2X2[f] = d2;
+                sumVjXj[f] += d1;
+                sumV2X2[f] += d2;
             }
         }