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); 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); result.put("train", train);
metrics.put("tarinDetail", trainBatchMetrics); metrics.put("tarinDetail", trainBatchMetrics);

} }


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


// 记录验证指标详情 // 记录验证指标详情
HashMap<String, Object> map = aimsService.queryMetricsParams(runId); 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); result.put("evaluate", evaluate);


Loading…
Cancel
Save