From 30b6b1582050cb2c74649e7283d18ecebb9ca0b7 Mon Sep 17 00:00:00 2001 From: chenzhihang <709011834@qq.com> Date: Fri, 18 Oct 2024 15:50:29 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=9F=A5=E8=AF=A2=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E7=89=88=E6=9C=AC=E6=8C=87=E6=A0=87=E5=AF=B9=E6=AF=94?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../platform/service/impl/ModelsServiceImpl.java | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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 734e39e7..3742478d 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 @@ -1009,15 +1009,19 @@ public class ModelsServiceImpl implements ModelsService { branch.put("metrics", trainMetrics); ArrayList metrics_names = new ArrayList<>(); - for (String key : trainMetrics.keySet()) { - metrics_names.add(key); + if (trainMetrics != null) { + for (String key : trainMetrics.keySet()) { + metrics_names.add(key); + } } branch.put("metrics_names", metrics_names); } else { Map evaluateMetrics = (Map) metrics.get(Constant.Type_Evaluate); ArrayList metrics_names = new ArrayList<>(); - for (String key : evaluateMetrics.keySet()) { - metrics_names.add(key); + if (evaluateMetrics != null) { + for (String key : evaluateMetrics.keySet()) { + metrics_names.add(key); + } } branch.put("metrics_names", metrics_names); branch.put("metrics", evaluateMetrics);