From f90a7f2bcdc16589ea3eb6359247faad56697202 Mon Sep 17 00:00:00 2001 From: myui Date: Tue, 17 Nov 2015 19:41:24 +0900 Subject: [PATCH] Fixed a bug in `merge` mehotd of `fm_predict` --- core/src/main/java/hivemall/fm/FMPredictGenericUDAF.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/src/main/java/hivemall/fm/FMPredictGenericUDAF.java b/core/src/main/java/hivemall/fm/FMPredictGenericUDAF.java index 69dec01e..83b97294 100644 --- 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; } }