瀏覽代碼

add infer code

often 5 月之前
父節點
當前提交
f92cb172b2
共有 1 個文件被更改,包括 1 次插入1 次删除
  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)