diff --git a/ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java b/ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java index 29335b6..437564d 100644 --- a/ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java +++ b/ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java @@ -76,6 +76,9 @@ public class PMMLExecutorRFImpl implements PMMLExecutor { } Map results = evaluator.evaluate(arguments); + + System.out.println("pmml_predict_results: " + results); + List targetFields = evaluator.getTargetFields(); TargetField targetField = targetFields.get(0); @@ -92,10 +95,10 @@ public class PMMLExecutorRFImpl implements PMMLExecutor { value_1 = tmpnum.doubleValue(); } - // TODO 不知道啥含义 先注释掉 返回结果 - // if (targetFieldValue instanceof ProbabilityDistribution) { - // value_1 = ((ProbabilityDistribution) targetFieldValue).getValue("1"); - // } + // TODO 模型取值逻辑待确认 + if (targetFieldValue instanceof ProbabilityDistribution) { + value_1 = ((ProbabilityDistribution) targetFieldValue).getValue("1"); + }