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 431077b4..216824f7 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 @@ -988,7 +988,8 @@ public class ModelsServiceImpl implements ModelsService { ModelMetaVo modelMetaVo = JSON.parseObject(meta, ModelMetaVo.class); if (modelMetaVo.getParams() != null) { HashMap params = modelMetaVo.getParams(); - branch.putAll(params); + branch.put("params", params); + ArrayList params_names = new ArrayList<>(); for (String key : params.keySet()) { params_names.add(key); @@ -999,10 +1000,20 @@ public class ModelsServiceImpl implements ModelsService { HashMap metrics = modelMetaVo.getMetrics(); if (Constant.Type_Train.equals(type)) { Map trainMetrics = (Map) metrics.get(Constant.Type_Train); - branch.putAll(trainMetrics); + ArrayList metrics_names = new ArrayList<>(); + for (String key : trainMetrics.keySet()) { + metrics_names.add(key); + } + branch.put("metrics_names", metrics_names); + branch.put("metrics",trainMetrics); } else { Map evaluateMetrics = (Map) metrics.get(Constant.Type_Evaluate); - branch.putAll(evaluateMetrics); + ArrayList metrics_names = new ArrayList<>(); + for (String key : evaluateMetrics.keySet()) { + metrics_names.add(key); + } + branch.put("metrics_names", metrics_names); + branch.put("metrics",evaluateMetrics); } } }