Bladeren bron

add infer code

often 6 maanden geleden
bovenliggende
commit
f92cb172b2
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      recommend-model-produce/src/main/python/models/dssm/static_model.py

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

@@ -36,7 +36,7 @@ class StaticModel():
         else:
             sample_id = paddle.static.data(
                 name="sample_id", shape=[-1, 1], dtype='int64')
-            feeds_list.append(label)     
+            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)