often 5 mesi fa
parent
commit
0ae4accafc

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

@@ -26,7 +26,7 @@ class DSSMReader(MultiSlotDataGenerator):
                 # 验证特征维度
                 if len(left_features) != self.feature_dim or len(right_features) != self.feature_dim:
                     return None
-                
+                sys.stderr.write(f"left_features length: {len(left_features)}\n")
                 # 构建输出列表
                 output = []
                 #output.append(("sample_id", [sample_id]))  # 样本ID

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

@@ -35,6 +35,7 @@ class StaticModel():
 
         # 第一层的输入维度是所有特征的embedding拼接
         self.input_dim = self.tag_feature_num  + self.stat_features_num*3
+        print(f"debug debug data input dim: {self.input_dim}")
 
         
     def _init_hyper_parameters(self):
@@ -95,7 +96,7 @@ class StaticModel():
             return fetch_dict
         else:
             label,left_features, right_features = input
-            print(f"Label shape: {label.shape}")
+            
             # 获取相似度和特征向量
             sim_score, left_vec, right_vec = dssm_model(left_features, right_features)