often 5 달 전
부모
커밋
e3e365cb31
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      recommend-model-produce/src/main/python/models/dssm/static_model.py

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

@@ -50,9 +50,9 @@ class StaticModel():
         )
 
         if is_infer:
-            sample_id, left_features, right_features = input
+            left_features, right_features = input
         else:
-            sample_id, left_features, right_features, label = input
+            left_features, right_features, label = input
 
         # 获取相似度和特征向量
         sim_score, left_vec, right_vec = dssm_model(left_features, right_features)