Browse Source

优化查询模型版本指标对比

dev-czh
chenzhihang 1 year ago
parent
commit
30b6b15820
1 changed files with 8 additions and 4 deletions
  1. +8
    -4
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java

+ 8
- 4
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ModelsServiceImpl.java View File

@@ -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);


Loading…
Cancel
Save