|
@@ -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
|
|
|
|
|
|
|