network structure difference with deepFM paper

rixwew / pytorch-fm

Factorization Machine models in PyTorch

MIT License

1.04k stars 225 forks source link

network structure difference with deepFM paper #34

Open Pengtai9928 opened 2 years ago

Pengtai9928 commented 2 years ago

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?