|
@@ -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:")
|