often 5 miesięcy temu
rodzic
commit
ef6e1c4a56

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

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

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

@@ -35,9 +35,9 @@ class StaticModel():
                 name="right_features", shape=[-1, self.feature_num], dtype='float32')
             feeds_list.append(right_features)
         else:
-            sample_id = paddle.static.data(
-                name="sample_id", shape=[-1, 1], dtype='int64')
-            feeds_list.append(sample_id)     
+            #sample_id = paddle.static.data(
+            #    name="sample_id", shape=[-1, 1], dtype='int64')
+            #feeds_list.append(sample_id)     
             left_features = paddle.static.data(
                 name="left_features", shape=[-1, self.feature_num], dtype='float32')
             feeds_list.append(left_features)
@@ -56,7 +56,7 @@ class StaticModel():
         )
         
         if is_infer:
-            sample_id,left_features = input
+            left_features = input
             left_vec = dssm_model(left_features,None,is_infer=True)
             self.inference_target_var = left_vec
             fetch_dict = {