search

geffy / tffm

TensorFlow implementation of an arbitrary order Factorization Machine
MIT License
780 stars 176 forks source link

is the output of model the same as original FM? #47

Closed kdqzzxxcc closed 5 years ago

kdqzzxxcc commented 5 years ago

is the self.outputs in the core.py the same as:

from the original FM model paper?

I have trained a model using tffm, and I want to use saved model to inference by c++ code, so I need to rewrite a output by myself, I find the output in your model is a little confusing, so does it have a equation for order 2 FM model like above one? or is it the same equation?

kdqzzxxcc commented 5 years ago

I've carefully read the code, found the equation is the same, close the issue

Bihaqo commented 5 years ago

Hi, it computes it by using different formulas (complicated and unpublished, long story..), but should yield the same numbers as the formula in the original paper (there are tests for that).

On Wed, 27 Mar 2019 at 09:41, Deqiang Kong notifications@github.com wrote:

Closed #47 https://github.com/geffy/tffm/issues/47.

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/geffy/tffm/issues/47#event-2232462541, or mute the thread https://github.com/notifications/unsubscribe-auth/AFXq6VmNEvm8ZmiJC5m-9cIFMZmSQN71ks5vazy4gaJpZM4cNIng .