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