학습실행 step1 할 때 best epoch 업데이트

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);
+  }
 }

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);
 }

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();
+  }
 }