|
@@ -196,6 +196,7 @@ public class PAIModelV1 {
|
|
|
request.addFeed(entry.getKey(), TFDataType.DT_STRING, new long[]{items.size()}, entry.getValue());
|
|
|
}
|
|
|
request.addFetch("probs");
|
|
|
+ LOGGER.info("PAIModelV1 request={}", request.getRequest().toString());
|
|
|
TFResponse response = client.predict(request);
|
|
|
List<Float> result = response.getFloatVals("probs");
|
|
|
if (!CollectionUtils.isEmpty(result)) {
|