| @@ -1009,15 +1009,19 @@ public class ModelsServiceImpl implements ModelsService { | |||||
| branch.put("metrics", trainMetrics); | branch.put("metrics", trainMetrics); | ||||
| ArrayList<String> metrics_names = new ArrayList<>(); | 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); | branch.put("metrics_names", metrics_names); | ||||
| } else { | } else { | ||||
| Map<String, Object> evaluateMetrics = (Map<String, Object>) metrics.get(Constant.Type_Evaluate); | Map<String, Object> evaluateMetrics = (Map<String, Object>) metrics.get(Constant.Type_Evaluate); | ||||
| ArrayList<String> metrics_names = new ArrayList<>(); | 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_names", metrics_names); | ||||
| branch.put("metrics", evaluateMetrics); | branch.put("metrics", evaluateMetrics); | ||||