From 3a45b04edf5975c599c72b436074768a131f4a87 Mon Sep 17 00:00:00 2001 From: chenzhihang <709011834@qq.com> Date: Tue, 24 Sep 2024 11:47:54 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=9E=E9=AA=8C=E6=A8=A1=E5=9E=8B=E5=AF=BC?= =?UTF-8?q?=E5=87=BA=E5=85=83=E6=95=B0=E6=8D=AE=E8=AE=B0=E5=BD=95=E6=8C=87?= =?UTF-8?q?=E6=A0=87=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/ruoyi/platform/service/impl/ModelsServiceImpl.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java index 64aa42a8..06973e08 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java @@ -1136,6 +1136,7 @@ public class ModelsServiceImpl implements ModelsService { } void getMetrics(ModelMetaVo modelMetaVo) throws Exception { + HashMap result = new HashMap<>(); HashMap metrics = modelMetaVo.getMetrics(); JSONArray trainMetrics = (JSONArray) metrics.get("train"); for (int i = 0; i < trainMetrics.size(); i++) { @@ -1143,13 +1144,17 @@ public class ModelsServiceImpl implements ModelsService { String runId = jsonObject.getString("run_id"); List expTrainInfos = aimsService.getExpTrainInfos(modelMetaVo.getTrainTask().getExperimentId(), runId); System.out.print(expTrainInfos); + for (InsMetricInfoVo expTrainInfo :expTrainInfos) { + Map metrics1 = expTrainInfo.getMetrics(); + } } - JSONArray testMetrics = (JSONArray) metrics.get("test"); + JSONArray testMetrics = (JSONArray) metrics.get("evaluate"); for (int i = 0; i < testMetrics.size(); i++) { JSONObject jsonObject = testMetrics.getJSONObject(i); String runId = jsonObject.getString("run_id"); List expTestInfos = aimsService.getExpEvaluateInfos(modelMetaVo.getTrainTask().getExperimentId(), runId); + System.out.print(expTestInfos); } } }