zhaohaipeng 3 nedēļas atpakaļ
vecāks
revīzija
68376f6640

+ 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());