pytorch-fm icon indicating copy to clipboard operation
pytorch-fm copied to clipboard

Published 20 hours ago • verified publisher icon rixwew

network structure difference with deepFM paper

Open Pengtai9928 opened this issue 3 years ago • 0 comments

I noticed that for the deemFM model, the implemented mlp_dim is (16, 16). However in the original deepFM paper, the network structure specified is 400-400-400 (screenshot below). May I ask why is the network structure different?

image

Pengtai9928 avatar Dec 08 '21 08:12 Pengtai9928