often 5 bulan lalu
induk
melakukan
b541087075

+ 1 - 1
recommend-model-produce/src/main/python/models/dssm/net.py

@@ -12,7 +12,7 @@ class DSSMLayer(nn.Layer):
         self.embedding_dim = embedding_dim
         self.output_dim = output_dim
         # 第一层的输入维度是所有特征的embedding拼接
-        self.hidden_layers = [feature_num * embedding_dim] + hidden_layers + [output_dim]
+        self.hidden_layers = [self.feature_num * embedding_dim] + hidden_layers + [output_dim]
         self.hidden_acts = hidden_acts