dssm train

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

@@ -45,7 +45,7 @@ class DSSMReader(MultiSlotDataGenerator):
                 
                 # 构建输出列表
                 output = []
-                output.append(("sample_id", [float(sample_id)]))  # 样本ID
+                # output.append(("sample_id", [float(sample_id)]))  # 样本ID
                 output.append(("left_features", left_features))   # 左视频特征
 
                 

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

@@ -91,7 +91,6 @@ class StaticModel():
         )
         
         if is_infer:
-            print("input: {}".format(input))
             left_features = input[0]
             left_vec = dssm_model(left_features,right_features=None,is_infer=True)
             self.inference_target_var = left_vec