浏览代码

delete unuseful log

often 5 月之前
父节点
当前提交
ed368ac99b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      recommend-model-produce/src/main/python/models/dssm/net.py

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

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