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