MVPTeam/kamco-train-api
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

성공시 csv 파일 테이블에 저장 연결

Browse Source
This commit is contained in:
teddy
2026-02-12 21:01:24 +09:00
parent 8ada26448b
commit 5b682c1386
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -22,6 +22,8 @@ public class TrainJobWorker {
private final ModelTrainJobCoreService modelTrainJobCoreService; private final ModelTrainJobCoreService modelTrainJobCoreService;
private final ModelTrainMngCoreService modelTrainMngCoreService; private final ModelTrainMngCoreService modelTrainMngCoreService;
private final ModelTrainMetricsJobService modelTrainMetricsJobService;
private final ModelTestMetricsJobService modelTestMetricsJobService;
private final DockerTrainService dockerTrainService; private final DockerTrainService dockerTrainService;
private final ObjectMapper objectMapper; private final ObjectMapper objectMapper;
@@ -90,8 +92,10 @@ public class TrainJobWorker {
if (isEval) { if (isEval) {
modelTrainMngCoreService.markStep2Success(modelId); modelTrainMngCoreService.markStep2Success(modelId);
modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
} else { } else {
modelTrainMngCoreService.markStep1Success(modelId); modelTrainMngCoreService.markStep1Success(modelId);
modelTestMetricsJobService.findTestValidMetricCsvFiles();
} }
} else { } else {