From 69bd5efa5e2849a560e604d0aa608d5492113915 Mon Sep 17 00:00:00 2001
From: dengzedong <dengzedong@email>
Date: 星期二, 12 十一月 2024 15:44:52 +0800
Subject: [PATCH] 模型运行输出配置
---
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java | 43 +++++++++++++++++++++++++++++++------------
1 files changed, 31 insertions(+), 12 deletions(-)
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
index 1fd8c7f..89994ce 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/api/MdkApiImpl.java
@@ -70,6 +70,9 @@
@Override
public MdkPredictModuleRespDTO predictModule(MdkPredictReqDTO reqDTO) {
MdkPredictModuleRespDTO resp = new MdkPredictModuleRespDTO();
+ resp.setPredictTime(reqDTO.getPredictTime());
+ resp.setModuleType(reqDTO.getModuleType());
+
Map<String, MdkPredictItemRespDTO> predictItemRespMap = new HashMap<>();
try {
if (reqDTO.getPredictTime() == null) {
@@ -96,18 +99,34 @@
}
List<ItemVO> predictItemList = mmPredictItemService.getByModuleId(module.getId());
Map<String, PredictResultVO> predictResultMap = predictModuleHandler.predict(predictItemList, reqDTO.getPredictTime(), intervalTime);
- for (Map.Entry<String, PredictResultVO> entry : predictResultMap.entrySet()) {
- List<MdkPredictDataDTO> predictData = entry.getValue().getPredictList().stream().map(t-> {
- MdkPredictDataDTO dto1 = new MdkPredictDataDTO();
- dto1.setDataTime(t.getDataTime());
- dto1.setDataValue(t.getDataValue());
- return dto1;
- }).collect(Collectors.toList());
- MdkPredictItemRespDTO itemResp = new MdkPredictItemRespDTO();
- itemResp.setItemId(entry.getValue().getPredictId());
- itemResp.setPredictData(predictData);
- predictItemRespMap.put(entry.getKey(), itemResp);
- }
+
+// for (Map.Entry<String, PredictResultVO> entry : predictResultMap.entrySet()) {
+// for (Map.Entry<String, List<DataValueVO>> dataListEntry : entry.getValue().getPredictLists().entrySet()) {
+// List<MdkPredictDataDTO> predictData = dataListEntry.getValue().stream().map(t -> {
+// MdkPredictDataDTO dto1 = new MdkPredictDataDTO();
+// dto1.setDataTime(t.getDataTime());
+// dto1.setDataValue(t.getDataValue());
+// return dto1;
+// }).collect(Collectors.toList());
+// MdkPredictItemRespDTO itemResp = new MdkPredictItemRespDTO();
+// itemResp.setItemId(dataListEntry.getKey());
+// itemResp.setPredictData(predictData);
+// predictItemRespMap.put(entry.getKey(), itemResp);
+// }
+// }
+
+// for (Map.Entry<String, PredictResultVO> entry : predictResultMap.entrySet()) {
+// List<MdkPredictDataDTO> predictData = entry.getValue().getPredictList().stream().map(t-> {
+// MdkPredictDataDTO dto1 = new MdkPredictDataDTO();
+// dto1.setDataTime(t.getDataTime());
+// dto1.setDataValue(t.getDataValue());
+// return dto1;
+// }).collect(Collectors.toList());
+// MdkPredictItemRespDTO itemResp = new MdkPredictItemRespDTO();
+// itemResp.setItemId(entry.getValue().getPredictId());
+// itemResp.setPredictData(predictData);
+// predictItemRespMap.put(entry.getKey(), itemResp);
+// }
}
log.info("预测计算结束: " + System.currentTimeMillis());
} catch (Exception ex) {
--
Gitblit v1.9.3