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<String> 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<String, Object> evaluateMetrics = (Map<String, Object>) metrics.get(Constant.Type_Evaluate);
                             ArrayList<String> 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);