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