|
@@ -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 = {
|