test

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

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

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

@@ -53,6 +53,9 @@ class StaticModel():
             left_features, right_features = input
         else:
             left_features, right_features, label = input
+            paddle.static.Print(left_features, message="lqc left data feature shape:")
+            paddle.static.Print(right_features, message="lqc right data feature shape:")
+            paddle.static.Print(label, message="lqc label feature shape:")
 
         # 获取相似度和特征向量
         sim_score, left_vec, right_vec = dssm_model(left_features, right_features)