often 5 月之前
父節點
當前提交
73fc266bb4
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      recommend-model-produce/src/main/python/models/dssm/static_model.py

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

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