often 5 mēneši atpakaļ
vecāks
revīzija
c29366e6e7

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

@@ -23,7 +23,12 @@ class StaticModel():
         # 定义输入数据占位符
         # sample_id = paddle.static.data(
         #    name="sample_id", shape=[-1, 1], dtype='int64')
-        
+       
+        if not is_infer:
+            label = paddle.static.data(
+                name="label", shape=[-1, 1], dtype='float32')
+            feeds_list.append(label)
+                    
         left_features = paddle.static.data(
             name="left_features", shape=[-1, self.feature_num], dtype='float32')
         
@@ -32,10 +37,7 @@ class StaticModel():
 
         feeds_list = [left_features, right_features]
         
-        if not is_infer:
-            label = paddle.static.data(
-                name="label", shape=[-1, 1], dtype='float32')
-            feeds_list.append(label)
+
 
         return feeds_list
 
@@ -49,10 +51,12 @@ class StaticModel():
             hidden_acts=self.hidden_acts
         )
 
+    
+
         if is_infer:
             left_features, right_features = input
         else:
-            left_features, right_features, label = input
+            label,left_features, right_features = 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:")