Browse Source

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

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

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

@@ -1028,7 +1028,7 @@ public class ModelsServiceImpl implements ModelsService {

ModelDependency1 modelDependency1 = modelDependency1Dao.queryByRepoAndVersion(model.getRepoId(), model.getIdentifier(), model.getVersion());
ModelMetaVo modelMetaVo = JSON.parseObject(modelDependency1.getMeta(), ModelMetaVo.class);
HashMap<String, Object> metrics = modelMetaVo.getMetrics();
HashMap<String, Object> metrics = modelMetaVo.getMetricsParams();
if (Constant.Type_Train.equals(type)) {
batchMetrics.add(metrics.get("tarinDetail"));
} else {
@@ -1271,7 +1271,7 @@ public class ModelsServiceImpl implements ModelsService {

}
result.put("train", train);
result.put("tarinDetail", trainBatchMetrics);
metrics.put("tarinDetail", trainBatchMetrics);

}

@@ -1293,10 +1293,11 @@ public class ModelsServiceImpl implements ModelsService {

}
result.put("evaluate", evaluate);
result.put("evaluateDetail", evaluateBatchMetrics);
metrics.put("evaluateDetail", evaluateBatchMetrics);

}
modelMetaVo.setMetrics(result);
modelMetaVo.setMetricsParams(metrics);
}

}

Loading…
Cancel
Save