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