Bladeren bron

delete unuseful log

often 5 maanden geleden
bovenliggende
commit
ea30143260

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

@@ -5,13 +5,17 @@ class DSSMReader(MultiSlotDataGenerator):
     def __init__(self):
         super(DSSMReader, self).__init__()
         self.feature_dim = 5  # 设置特征维度
-
+        self.data_count = 0  # 添加计数器
     def init(self, config=None):
         pass
 
     def line_process(self, line):
         try:
+            self.data_count += 1
+            if self.data_count % 100 == 0:  # 每100条打印一次
+                print(f"Processing line {self.data_count}: {line[:100]}...")  # 打印前100个字符
             # 按tab分割样本的各个字段
+            
             sample_values = line.rstrip('\n').split('\t')
             if len(sample_values) == 6: # 训练格式
                 sample_id, label, vid_left,vid_right,left_features, right_features = sample_values

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

@@ -65,6 +65,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)