Browse Source

实验实例成功运行结束后记录指标数据

dev-czh
chenzhihang 1 year ago
parent
commit
803eea2054
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ExperimentInstanceStatusTask.java

+ 2
- 2
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/scheduling/ExperimentInstanceStatusTask.java View File

@@ -59,7 +59,7 @@ public class ExperimentInstanceStatusTask {
if (trainMetricRecords != null && !trainMetricRecords.isEmpty()) {
for (Map<String, Object> trainMetricRecord : trainMetricRecords) {
String taskId = (String) trainMetricRecord.get("task_id");
if (taskId.startsWith("model-export")) {
if (taskId.startsWith("model-train")) {
String runId = (String) trainMetricRecord.get("run_id");
List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(true, experimentIns.getExperimentId(), runId);
for (InsMetricInfoVo expTrainInfo : expTrainInfos) {
@@ -73,7 +73,7 @@ public class ExperimentInstanceStatusTask {
if (evaluateMetricRecords != null && !evaluateMetricRecords.isEmpty()) {
for (Map<String, Object> evaluateMetricRecord : evaluateMetricRecords) {
String taskId = (String) evaluateMetricRecord.get("task_id");
if (taskId.startsWith("model-export")) {
if (taskId.startsWith("model-evaluate")) {
String runId = (String) evaluateMetricRecord.get("run_id");
List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(false, experimentIns.getExperimentId(), runId);
for (InsMetricInfoVo expTrainInfo : expTrainInfos) {


Loading…
Cancel
Save