We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 02a37c7 + ed78604 commit e74853eCopy full SHA for e74853e
models/rank/fat_deepffm/net.py
@@ -193,7 +193,7 @@ def __init__(self,
193
if acts[i] == 'relu':
194
act = paddle.nn.ReLU()
195
self.add_sublayer('act_%d' % i, act)
196
- self._mlp_layers.append(linear)
+ self._mlp_layers.append(act)
197
198
def forward(self, feat_embeddings):
199
y_dnn = paddle.reshape(feat_embeddings, [feat_embeddings.shape[0], -1])
0 commit comments