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 b97fd948..f69b7359 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 @@ -1308,13 +1308,13 @@ public class ModelsServiceImpl implements ModelsService { // 记录训练指标详情 HashMap map = aimsService.queryMetricsParams(runId); - List> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params")); - trainBatchMetrics.add(batchMetric); - + if (map != null && !map.isEmpty()) { + List> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params")); + trainBatchMetrics.add(batchMetric); + } } result.put("train", train); metrics.put("tarinDetail", trainBatchMetrics); - } JSONArray testMetrics = (JSONArray) metrics.get("evaluate"); @@ -1330,8 +1330,10 @@ public class ModelsServiceImpl implements ModelsService { // 记录验证指标详情 HashMap map = aimsService.queryMetricsParams(runId); - List> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params")); - evaluateBatchMetrics.add(batchMetric); + if (map != null && !map.isEmpty()) { + List> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params")); + evaluateBatchMetrics.add(batchMetric); + } } result.put("evaluate", evaluate);