src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java

@@ -8,6 +8,8 @@ import com.kamco.cd.training.model.dto.ModelConfigDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto.Basic;
 import com.kamco.cd.training.model.service.ModelTrainMngService;
+import com.kamco.cd.training.train.service.ModelTestMetricsJobService;
+import com.kamco.cd.training.train.service.ModelTrainMetricsJobService;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.media.Content;
@@ -16,6 +18,7 @@ import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.responses.ApiResponses;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.Valid;
+import java.io.IOException;
 import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -35,6 +38,8 @@ import org.springframework.web.bind.annotation.RestController;
 @RequestMapping("/api/models")
 public class ModelTrainMngApiController {
   private final ModelTrainMngService modelTrainMngService;
+  private final ModelTrainMetricsJobService modelTrainMetricsJobService;
+  private final ModelTestMetricsJobService modelTestMetricsJobService;
 
   @Operation(summary = "모델학습 목록 조회", description = "모델학습 목록 조회 API")
   @ApiResponses(
@@ -167,4 +172,44 @@ public class ModelTrainMngApiController {
   public ApiResponseDto<Long> findModelStep1InProgressCnt() {
     return ApiResponseDto.ok(modelTrainMngService.findModelStep1InProgressCnt());
   }
+
+  @Operation(
+      summary = "스케줄러 findTrainValidMetricCsvFiles",
+      description = "스케줄러 findTrainValidMetricCsvFiles")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "검색 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = Long.class))),
+        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @GetMapping("/schedule-trainvalid")
+  public ApiResponseDto<Long> findTrainValidMetricCsvFiles() {
+    modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
+    return ApiResponseDto.ok(null);
+  }
+
+  @Operation(summary = "스케줄러 findTestMetricCsvFiles", description = "스케줄러 findTestMetricCsvFiles")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "검색 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = Long.class))),
+        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @GetMapping("/schedule-test")
+  public ApiResponseDto<Long> findTestValidMetricCsvFiles() throws IOException {
+    modelTestMetricsJobService.findTestValidMetricCsvFiles();
+    return ApiResponseDto.ok(null);
+  }
 }

src/main/java/com/kamco/cd/training/train/service/TrainJobWorker.java

@@ -100,10 +100,10 @@ public class TrainJobWorker {
 
         if (isEval) {
           modelTrainMngCoreService.markStep2Success(modelId);
-          modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
+          modelTestMetricsJobService.findTestValidMetricCsvFiles();
         } else {
           modelTrainMngCoreService.markStep1Success(modelId);
-          modelTestMetricsJobService.findTestValidMetricCsvFiles();
+          modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
         }
 
       } else {