dssm train

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

@@ -68,9 +68,9 @@ class StaticModel():
             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.input_dim], dtype='float32')
             feeds_list.append(left_features)
@@ -91,12 +91,11 @@ class StaticModel():
         )
         
         if is_infer:
-            vid, left_features = input
+            left_features = input[0]
             left_vec = dssm_model(left_features,right_features=None,is_infer=True)
             self.inference_target_var = left_vec
             fetch_dict = {
-                'left_vector': left_vec,
-                'vid': vid
+                'left_vector': left_vec
             }
             return fetch_dict
         else: