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 216824f7..e4f78e13 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 @@ -1028,7 +1028,7 @@ public class ModelsServiceImpl implements ModelsService { ModelDependency1 modelDependency1 = modelDependency1Dao.queryByRepoAndVersion(model.getRepoId(), model.getIdentifier(), model.getVersion()); ModelMetaVo modelMetaVo = JSON.parseObject(modelDependency1.getMeta(), ModelMetaVo.class); - HashMap metrics = modelMetaVo.getMetrics(); + HashMap metrics = modelMetaVo.getMetricsParams(); if (Constant.Type_Train.equals(type)) { batchMetrics.add(metrics.get("tarinDetail")); } else { @@ -1271,7 +1271,7 @@ public class ModelsServiceImpl implements ModelsService { } result.put("train", train); - result.put("tarinDetail", trainBatchMetrics); + metrics.put("tarinDetail", trainBatchMetrics); } @@ -1293,10 +1293,11 @@ public class ModelsServiceImpl implements ModelsService { } result.put("evaluate", evaluate); - result.put("evaluateDetail", evaluateBatchMetrics); + metrics.put("evaluateDetail", evaluateBatchMetrics); } modelMetaVo.setMetrics(result); + modelMetaVo.setMetricsParams(metrics); } }