Fixed score calculation w.r.t. bias values in predict()

myui · Oct 18, 2013 · 0130cfe · 0130cfe
1 parent e7759d9
commit 0130cfe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/main/hivemall/classifier/BinaryOnlineClassifierUDTF.java b/src/main/hivemall/classifier/BinaryOnlineClassifierUDTF.java
@@ -200,7 +200,7 @@ protected float predict(final List<?> features) {
         if(biasKey != null) {
             WeightValue biasWeight = weights.get(biasKey);
             if(biasWeight != null) {
-                score += (biasWeight.getValue() + bias);
+                score += (biasWeight.getValue() * bias);
             }
         }
 

diff --git a/target/hivemall.jar b/target/hivemall.jar