neural_factorization_machine icon indicating copy to clipboard operation
neural_factorization_machine copied to clipboard

Why do not load the deep layer params for pretrain model?

Open demien opened this issue 7 years ago • 1 comments

As the code it seems do not load the deep layer params if use the pretrain model.

https://github.com/hexiangnan/neural_factorization_machine/blob/master/NeuralFM.py#L202-L217

Is that on purpose which means daily model has a brand new deep layer params?

demien avatar Nov 26 '18 08:11 demien

NFM uses the feature embeddings of FM to init the embedding layer. The deep layers are learned from random init.

On Mon, Nov 26, 2018 at 4:29 PM demien [email protected] wrote:

As the code it seems do not load the deep layer params if use the pretrain model.

https://github.com/hexiangnan/neural_factorization_machine/blob/master/NeuralFM.py#L202-L217

Is that on purpose which means daily model has a brand new deep layer params?

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/hexiangnan/neural_factorization_machine/issues/6, or mute the thread https://github.com/notifications/unsubscribe-auth/ABGxjrLqxNioq9jEV0zVeY0h6FKWAXfzks5uy6ZmgaJpZM4Yyiyn .

hexiangnan avatar Nov 26 '18 12:11 hexiangnan