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