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 4713b099..ee82c9b0 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 @@ -986,34 +986,42 @@ public class ModelsServiceImpl implements ModelsService { String meta = modelDependency1Dao.getMeta(identifier, owner, (String) branch.get("name")); if (StringUtils.isNotEmpty(meta)) { ModelMetaVo modelMetaVo = JSON.parseObject(meta, ModelMetaVo.class); - if (modelMetaVo.getParams() != null) { - HashMap params = modelMetaVo.getParams(); - branch.put("params", params); + if (Constant.Source_Add.equals(modelMetaVo.getModelSource())) { + branch.put("params", null); + branch.put("params_names", new ArrayList<>()); + branch.put("metrics", null); + branch.put("metrics_names", new ArrayList<>()); + } else { + if (modelMetaVo.getParams() != null) { + HashMap params = modelMetaVo.getParams(); + branch.put("params", params); - ArrayList params_names = new ArrayList<>(); - for (String key : params.keySet()) { - params_names.add(key); - } - branch.put("params_names", params_names); - } - if (modelMetaVo.getMetrics() != null) { - HashMap metrics = modelMetaVo.getMetrics(); - if (Constant.Type_Train.equals(type)) { - Map trainMetrics = (Map) metrics.get(Constant.Type_Train); - ArrayList metrics_names = new ArrayList<>(); - for (String key : trainMetrics.keySet()) { - metrics_names.add(key); + ArrayList params_names = new ArrayList<>(); + for (String key : params.keySet()) { + params_names.add(key); } - branch.put("metrics_names", metrics_names); - branch.put("metrics",trainMetrics); - } else { - Map evaluateMetrics = (Map) metrics.get(Constant.Type_Evaluate); - ArrayList metrics_names = new ArrayList<>(); - for (String key : evaluateMetrics.keySet()) { - metrics_names.add(key); + branch.put("params_names", params_names); + } + if (modelMetaVo.getMetrics() != null) { + HashMap metrics = modelMetaVo.getMetrics(); + if (Constant.Type_Train.equals(type)) { + Map trainMetrics = (Map) metrics.get(Constant.Type_Train); + branch.put("metrics", trainMetrics); + + ArrayList metrics_names = new ArrayList<>(); + 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); + } + branch.put("metrics_names", metrics_names); + branch.put("metrics", evaluateMetrics); } - branch.put("metrics_names", metrics_names); - branch.put("metrics",evaluateMetrics); } } }