liutongsz/fibo-rule
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

模型取值逻辑修改

Browse Source
This commit is contained in:
杜刚
2025-02-11 22:49:14 +08:00
parent a8aece404d
commit 554286c88c
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ddp/ddp-common/ddp-service/src/main/java/com/fibo/ddp/common/service/strategyx/aimodel/PMMLExecutor/impl/PMMLExecutorRFImpl.java
View File

@@ -82,12 +82,24 @@ public class PMMLExecutorRFImpl implements PMMLExecutor {
FieldName targetFieldName = targetField.getName(); FieldName targetFieldName = targetField.getName();
Object targetFieldValue = results.get(targetFieldName); 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; 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; return value_1;
} }