From 941eee1bf38dc215f8f07cf1aac7e54d8071663f Mon Sep 17 00:00:00 2001 From: sladesha Date: Fri, 3 Aug 2018 15:34:08 +0800 Subject: [PATCH] fix bug --- FFM/FFM.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/FFM/FFM.py b/FFM/FFM.py index 6945855..f4fe321 100644 --- a/FFM/FFM.py +++ b/FFM/FFM.py @@ -57,10 +57,11 @@ def inference(self): # shape of [None, 1] self.field_aware_interaction_terms = tf.constant(0, dtype='float32') # build dict to find f, key of feature,value of field + # fix bug : vifj instead of vifi , the same as vjfi for i in range(self.p): for j in range(i+1,self.p): self.field_aware_interaction_terms += tf.multiply( - tf.reduce_sum(tf.multiply(v[i,self.feature2field[i]], v[j,self.feature2field[j]])), + tf.reduce_sum(tf.multiply(v[i,self.feature2field[j]], v[j,self.feature2field[i]])), tf.multiply(self.X[:,i], self.X[:,j]) ) # shape of [None, 2]