diff --git a/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java index cffd16e..5098c4b 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java @@ -34,7 +34,7 @@ public class HyperParamCoreService { ModelHyperParamEntity entity = new ModelHyperParamEntity(); entity.setHyperVer(firstVersion); - + entity.setTotalUseCnt(entity.getTotalUseCnt() == null ? 1 : entity.getTotalUseCnt() + 1); applyHyperParam(entity, createReq); // user @@ -172,7 +172,7 @@ public class HyperParamCoreService { } else { entity.setCropSize("256,256"); } - // entity.setCropSize(src.getCropSize()); + entity.setCropSize(src.getCropSize()); // Important entity.setModelType(model); // 20250212 modeltype추가 diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java index bb56dbd..1563b7c 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java @@ -104,6 +104,9 @@ public class ModelTrainMngCoreService { if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) { throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND); } + // 하이퍼 파라미터 사용회수 업데이트 + hyperParamEntity.setTotalUseCnt( + hyperParamEntity.getTotalUseCnt() == null ? 1 : hyperParamEntity.getTotalUseCnt() + 1); String modelVer = String.join( diff --git a/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java b/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java index d0cd472..fb250b1 100644 --- a/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java +++ b/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java @@ -310,6 +310,9 @@ public class ModelHyperParamEntity { @Column(name = "default_param") private Boolean isDefault = false; + @Column(name = "total_use_cnt") + private Integer totalUseCnt = 0; + public HyperParamDto.Basic toDto() { return new HyperParamDto.Basic( this.modelType, diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java index 68214a9..b5dd84e 100644 --- a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java +++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java @@ -296,14 +296,8 @@ public class TrainJobService { e); // 런타임 예외로 래핑하되, 메시지에 핵심 정보 포함 - throw new IllegalStateException( - "tmp dataset build failed: modelUuid=" - + modelUuid - + ", modelId=" - + modelId - + ", tmpRaw=" - + raw, - e); + throw new CustomApiException( + "INTERNAL_SERVER_ERROR", HttpStatus.INTERNAL_SERVER_ERROR, "임시 데이터셋 생성에 실패했습니다."); } return modelUuid; }