Fix wrong getV() arg on ffm_predict

apache · Mar 1, 2017 · 45567c8 · 45567c8
1 parent 1e83eb5
commit 45567c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/java/hivemall/fm/FFMPredictUDF.java b/core/src/main/java/hivemall/fm/FFMPredictUDF.java
@@ -155,7 +155,7 @@ private static double predict(@Nonnull final Feature[] x,
                 if (!model.getV(ei, jField, vij)) {
                     continue;
                 }
-                if (!model.getV(ej, iField, vij)) {
+                if (!model.getV(ej, iField, vji)) {
                     continue;
                 }
                 for (int f = 0; f < factors; f++) {