From 45567c8322de6b978c88d896b7c1882f90a3f669 Mon Sep 17 00:00:00 2001 From: Takuya Kitazawa Date: Wed, 1 Mar 2017 19:01:13 +0900 Subject: [PATCH] Fix wrong getV() arg on ffm_predict --- core/src/main/java/hivemall/fm/FFMPredictUDF.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/src/main/java/hivemall/fm/FFMPredictUDF.java b/core/src/main/java/hivemall/fm/FFMPredictUDF.java index 2de5170cf..48745d948 100644 --- 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++) {