模型取值逻辑修改

This commit is contained in:
2025-02-11 22:49:14 +08:00
parent a8aece404d
commit 554286c88c

View File

@@ -82,12 +82,24 @@ public class PMMLExecutorRFImpl implements PMMLExecutor {
FieldName targetFieldName = targetField.getName();
Object targetFieldValue = results.get(targetFieldName);
System.out.println("target: " + targetFieldName.getValue() + ", value: " + targetFieldValue);
System.out.println("target: " + targetFieldName.getValue() + ", value: " + targetFieldValue+",valueType="+targetFieldValue.getClass());
double value_1 = 0.0f;
if (targetFieldValue instanceof ProbabilityDistribution) {
value_1 = ((ProbabilityDistribution) targetFieldValue).getValue("1");
// 2025-02-11 新增
if(targetFieldValue instanceof Number){
Number tmpnum = (Number)targetFieldValue;
value_1 = tmpnum.doubleValue();
}
// TODO 不知道啥含义 先注释掉 返回结果
// if (targetFieldValue instanceof ProbabilityDistribution) {
// value_1 = ((ProbabilityDistribution) targetFieldValue).getValue("1");
// }
return value_1;
}