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 8ce5073f..514a0d36 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 @@ -1162,28 +1162,32 @@ public class ModelsServiceImpl implements ModelsService { HashMap metrics = modelMetaVo.getMetrics(); JSONArray trainMetrics = (JSONArray) metrics.get("train"); - for (int i = 0; i < trainMetrics.size(); i++) { - JSONObject jsonObject = trainMetrics.getJSONObject(i); - String runId = jsonObject.getString("run_id"); - List expTrainInfos = aimsService.getExpTrainInfos1(true, modelMetaVo.getTrainTask().getExperimentId(), runId); - for (InsMetricInfoVo expTrainInfo : expTrainInfos) { - Map metrics1 = expTrainInfo.getMetrics(); - train.putAll(metrics1); + if (trainMetrics != null) { + for (int i = 0; i < trainMetrics.size(); i++) { + JSONObject jsonObject = trainMetrics.getJSONObject(i); + String runId = jsonObject.getString("run_id"); + List expTrainInfos = aimsService.getExpTrainInfos1(true, modelMetaVo.getTrainTask().getExperimentId(), runId); + for (InsMetricInfoVo expTrainInfo : expTrainInfos) { + Map metrics1 = expTrainInfo.getMetrics(); + train.putAll(metrics1); + } } + result.put("train", train); } - result.put("train", train); 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.getExpTrainInfos1(false, modelMetaVo.getTrainTask().getExperimentId(), runId); - for (InsMetricInfoVo expTestInfo : expTestInfos) { - Map metrics1 = expTestInfo.getMetrics(); - evaluate.putAll(metrics1); + if (testMetrics != null) { + for (int i = 0; i < testMetrics.size(); i++) { + JSONObject jsonObject = testMetrics.getJSONObject(i); + String runId = jsonObject.getString("run_id"); + List expTestInfos = aimsService.getExpTrainInfos1(false, modelMetaVo.getTrainTask().getExperimentId(), runId); + for (InsMetricInfoVo expTestInfo : expTestInfos) { + Map metrics1 = expTestInfo.getMetrics(); + evaluate.putAll(metrics1); + } } + result.put("evaluate", evaluate); } - result.put("evaluate", evaluate); modelMetaVo.setMetrics(result); } }