|
@@ -65,7 +65,6 @@ public class PAIModel {
|
|
|
featureKeys.add(featureKeyTuple);
|
|
featureKeys.add(featureKeyTuple);
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- LOGGER.info("featureKeys: {}", JSON.toJSONString(featureKeys));
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
private void initPredictClient() {
|
|
private void initPredictClient() {
|
|
@@ -130,7 +129,7 @@ public class PAIModel {
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
TFResponse response = predictClient.predict(request);
|
|
TFResponse response = predictClient.predict(request);
|
|
|
- Map<String, PredictProtos.ArrayProto> outputsMap = response.getOutputsMap();
|
|
|
|
|
|
|
+ // Map<String, PredictProtos.ArrayProto> outputsMap = response.getOutputsMap();
|
|
|
// LOGGER.info("response keys: {}", outputsMap.keySet());
|
|
// LOGGER.info("response keys: {}", outputsMap.keySet());
|
|
|
Map<String, List<Float>> resultMap = new HashMap<>();
|
|
Map<String, List<Float>> resultMap = new HashMap<>();
|
|
|
for (String scoreKey : scoreKeys) {
|
|
for (String scoreKey : scoreKeys) {
|