فهرست منبع

add infer code

often 5 ماه پیش
والد
کامیت
8e1672bc99
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

@@ -56,7 +56,7 @@ class StaticModel():
         )
         
         if is_infer:
-            left_features = input
+            left_features = input[0]
             left_vec = dssm_model(left_features,None,is_infer=True)
             self.inference_target_var = left_vec
             fetch_dict = {