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); } } }