From 51e47223d5ca48de4061f0a99f28ee65bbbed4ad Mon Sep 17 00:00:00 2001
From: dengzedong <dengzedong@email>
Date: 星期三, 05 二月 2025 11:37:06 +0800
Subject: [PATCH] 删除double类型的模型输出特殊处理，double转为长度为1的double数组
---
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/impl/PredictModelHandlerImpl.java | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/impl/PredictModelHandlerImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/impl/PredictModelHandlerImpl.java
index 368ad04..dd80a49 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/impl/PredictModelHandlerImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/predict/impl/PredictModelHandlerImpl.java
@@ -108,7 +108,6 @@
List<MmItemOutputEntity> itemOutputList = mmItemOutputService.getByItemid(predictModel.getItemid());
Map<MmItemOutputEntity, double[]> predictMatrixs = new HashMap<>();
- Map<MmItemOutputEntity, Double> predictDoubleValues = new HashMap<>();
for (MmItemOutputEntity output : itemOutputList) {
if (!modelResult.containsKey(output.getResultstr())) {
continue;
@@ -129,14 +128,13 @@
break;
case D:
Double temp3 = (Double) modelResult.get(output.getResultstr());
- predictDoubleValues.put(output, temp3);
+ predictMatrixs.put(output, new double[]{temp3});
break;
default:
break;
}
}
result.setPredictMatrixs(predictMatrixs);
- result.setPredictDoubleValues(predictDoubleValues);
result.setModelResult(modelResult);
result.setPredictTime(predictTime);
} catch (ModelResultErrorException ex) {
--
Gitblit v1.9.3