Browse Source

优化导出数据集模型版本

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

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

@@ -1308,13 +1308,13 @@ public class ModelsServiceImpl implements ModelsService {

// 记录训练指标详情
HashMap<String, Object> map = aimsService.queryMetricsParams(runId);
List<Map<String, Object>> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params"));
trainBatchMetrics.add(batchMetric);

if (map != null && !map.isEmpty()) {
List<Map<String, Object>> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params"));
trainBatchMetrics.add(batchMetric);
}
}
result.put("train", train);
metrics.put("tarinDetail", trainBatchMetrics);

}

JSONArray testMetrics = (JSONArray) metrics.get("evaluate");
@@ -1330,8 +1330,10 @@ public class ModelsServiceImpl implements ModelsService {

// 记录验证指标详情
HashMap<String, Object> map = aimsService.queryMetricsParams(runId);
List<Map<String, Object>> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params"));
evaluateBatchMetrics.add(batchMetric);
if (map != null && !map.isEmpty()) {
List<Map<String, Object>> batchMetric = aimsService.getBatchMetric((String) map.get("run_hash"), (String) map.get("params"));
evaluateBatchMetrics.add(batchMetric);
}

}
result.put("evaluate", evaluate);


Loading…
Cancel
Save