소스 검색

feat:bugfix

zhaohaipeng 3 주 전
부모
커밋
68376f6640
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/score/model/PAIModel.java

+ 2 - 0
recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/score/model/PAIModel.java

@@ -126,6 +126,8 @@ public class PAIModel {
                     request.addFeed(featureName, TFDataType.DT_DOUBLE, shape, doubleFeatureArr.get(featureName));
                 }
             }
+            request.addFetch("y_return_n_uv");
+            request.addFetch("probs_is_share");
             TFResponse response = predictClient.predict(request);
             // Map<String, PredictProtos.ArrayProto> outputsMap = response.getOutputsMap();
             // LOGGER.info("response keys: {}", outputsMap.keySet());