From 1e62a8b097eecf9ce75ab366def1ad2b69916e9d Mon Sep 17 00:00:00 2001
From: "gayoun.park" <gayoun.park@tf.dabeeo.com>
Date: Fri, 13 Feb 2026 10:15:04 +0900
Subject: [PATCH] =?UTF-8?q?=ED=95=99=EC=8A=B5=EC=8B=A4=ED=96=89=20step1=20?=
 =?UTF-8?q?=ED=95=A0=20=EB=95=8C=20best=20epoch=20=EC=97=85=EB=8D=B0?=
 =?UTF-8?q?=EC=9D=B4=ED=8A=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../core/ModelTrainMetricsJobCoreService.java |  5 +++
 .../ModelTrainMetricsJobRepositoryCustom.java |  2 ++
 .../ModelTrainMetricsJobRepositoryImpl.java   |  9 ++++++
 .../service/ModelTrainMetricsJobService.java  | 32 +++++++++++++++++++
 4 files changed, 48 insertions(+)

diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMetricsJobCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMetricsJobCoreService.java
index d7823b1..3f592d8 100644
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMetricsJobCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMetricsJobCoreService.java
@@ -29,4 +29,9 @@ public class ModelTrainMetricsJobCoreService {
   public void insertModelMetricsValidation(List<Object[]> batchArgs) {
     modelTrainMetricsJobRepository.insertModelMetricsValidation(batchArgs);
   }
+
+  @Transactional
+  public void updateModelSelectedBestEpoch(Long modelId, Integer epoch) {
+    modelTrainMetricsJobRepository.updateModelSelectedBestEpoch(modelId, epoch);
+  }
 }
diff --git a/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryCustom.java b/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryCustom.java
index 67517fe..f4031bf 100644
--- a/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryCustom.java
+++ b/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryCustom.java
@@ -12,4 +12,6 @@ public interface ModelTrainMetricsJobRepositoryCustom {
   void updateModelMetricsTrainSaveYn(Long modelId, String stepNo);
 
   void insertModelMetricsValidation(List<Object[]> batchArgs);
+
+  void updateModelSelectedBestEpoch(Long modelId, Integer epoch);
 }
diff --git a/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java b/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java
index fb5916c..1323d40 100644
--- a/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java
+++ b/src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java
@@ -82,4 +82,13 @@ public class ModelTrainMetricsJobRepositoryImpl extends QuerydslRepositorySuppor
 
     jdbcTemplate.batchUpdate(sql, batchArgs);
   }
+
+  @Override
+  public void updateModelSelectedBestEpoch(Long modelId, Integer epoch) {
+    queryFactory
+        .update(modelMasterEntity)
+        .set(modelMasterEntity.bestEpoch, epoch)
+        .where(modelMasterEntity.id.eq(modelId))
+        .execute();
+  }
 }
diff --git a/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java b/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
index b8a05b0..4487973 100644
--- a/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
@@ -6,9 +6,14 @@ import java.io.BufferedReader;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
+import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import java.util.stream.Stream;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.csv.CSVFormat;
@@ -129,6 +134,33 @@ public class ModelTrainMetricsJobService {
         throw new RuntimeException(e);
       }
 
+      Path responsePath = Paths.get(responseDir + "/" + modelInfo.getUuid());
+      Integer epoch = null;
+      boolean exists;
+      Pattern pattern = Pattern.compile("best_changed_fscore_epoch_(\\d+)\\.pth");
+
+      try (Stream<Path> s = Files.list(responsePath)) {
+        epoch =
+            s.filter(Files::isRegularFile)
+                .map(
+                    p -> {
+                      Matcher matcher = pattern.matcher(p.getFileName().toString());
+                      if (matcher.matches()) {
+                        return Integer.parseInt(matcher.group(1)); // ← 숫자 부분 추출
+                      }
+                      return null;
+                    })
+                .filter(Objects::nonNull)
+                .findFirst()
+                .orElse(null);
+
+      } catch (IOException e) {
+        throw new RuntimeException(e);
+      }
+
+      // best_changed_fscore_epoch_숫자.pth -> 숫자 값 가지고 와서 베스트 에폭에 업데이트 하기
+      modelTrainMetricsJobCoreService.updateModelSelectedBestEpoch(modelInfo.getModelId(), epoch);
+
       modelTrainMetricsJobCoreService.updateModelMetricsTrainSaveYn(
           modelInfo.getModelId(), "step1");
     }