模型取值逻辑修改 ,使用原来的取值逻辑

This commit is contained in:
2025-02-17 18:00:09 +08:00
parent 74103d4663
commit c35000738c

View File

@@ -76,6 +76,9 @@ public class PMMLExecutorRFImpl implements PMMLExecutor {
}
Map<FieldName, ?> results = evaluator.evaluate(arguments);
System.out.println("pmml_predict_results: " + results);
List<TargetField> 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");
}