|
|
@@ -21,8 +21,8 @@ class StaticModel():
|
|
|
|
|
|
def create_feeds(self, is_infer=False):
|
|
|
# 定义输入数据占位符
|
|
|
- sample_id = paddle.static.data(
|
|
|
- name="sample_id", shape=[-1, 1], dtype='int64')
|
|
|
+ # sample_id = paddle.static.data(
|
|
|
+ # name="sample_id", shape=[-1, 1], dtype='int64')
|
|
|
|
|
|
left_features = paddle.static.data(
|
|
|
name="left_features", shape=[-1, self.feature_num], dtype='float32')
|
|
|
@@ -30,7 +30,7 @@ class StaticModel():
|
|
|
right_features = paddle.static.data(
|
|
|
name="right_features", shape=[-1, self.feature_num], dtype='float32')
|
|
|
|
|
|
- feeds_list = [sample_id, left_features, right_features]
|
|
|
+ feeds_list = [left_features, right_features]
|
|
|
|
|
|
if not is_infer:
|
|
|
label = paddle.static.data(
|
|
|
@@ -63,7 +63,6 @@ class StaticModel():
|
|
|
|
|
|
if is_infer:
|
|
|
fetch_dict = {
|
|
|
- 'sample_id': sample_id,
|
|
|
'similarity': sim_score,
|
|
|
'left_vector': left_vec,
|
|
|
'right_vector': right_vec
|
often