From 49d2acb392eac432ef400c38daf2423cdfb57ad1 Mon Sep 17 00:00:00 2001
From: chenzhihang <709011834@qq.com>
Date: Fri, 27 Sep 2024 16:45:58 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=A8=A1=E5=9E=8B=E7=89=88?=
 =?UTF-8?q?=E6=9C=AC=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ModelsServiceImpl.java       | 36 ++++++++++---------
 1 file changed, 20 insertions(+), 16 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 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<String, Object> 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<InsMetricInfoVo> 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<InsMetricInfoVo> 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<InsMetricInfoVo> 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<InsMetricInfoVo> 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);
     }
 }