|
@@ -77,7 +77,7 @@ class StaticModel():
|
|
def net(self, input, is_infer=False):
|
|
def net(self, input, is_infer=False):
|
|
# 创建模型实例
|
|
# 创建模型实例
|
|
dssm_model = DSSMLayer(
|
|
dssm_model = DSSMLayer(
|
|
- tag_feature_nums=self.feature_nums,
|
|
|
|
|
|
+ tag_feature_nums=self.tag_feature_nums,
|
|
stat_features_num = self.stat_features_num,
|
|
stat_features_num = self.stat_features_num,
|
|
stat_features_num_embeddings = self.stat_features_num_embeddings,
|
|
stat_features_num_embeddings = self.stat_features_num_embeddings,
|
|
embedding_dim=self.embedding_dim,
|
|
embedding_dim=self.embedding_dim,
|