하이퍼 파라미터 사용회수 카운트 기능 추가 및 조회 수정

This commit is contained in:
2026-02-23 15:19:40 +09:00
parent fd63824edc
commit e6caea05b3
4 changed files with 10 additions and 10 deletions

View File

@@ -34,7 +34,7 @@ public class HyperParamCoreService {
ModelHyperParamEntity entity = new ModelHyperParamEntity(); ModelHyperParamEntity entity = new ModelHyperParamEntity();
entity.setHyperVer(firstVersion); entity.setHyperVer(firstVersion);
entity.setTotalUseCnt(entity.getTotalUseCnt() == null ? 1 : entity.getTotalUseCnt() + 1);
applyHyperParam(entity, createReq); applyHyperParam(entity, createReq);
// user // user
@@ -172,7 +172,7 @@ public class HyperParamCoreService {
} else { } else {
entity.setCropSize("256,256"); entity.setCropSize("256,256");
} }
// entity.setCropSize(src.getCropSize()); entity.setCropSize(src.getCropSize());
// Important // Important
entity.setModelType(model); // 20250212 modeltype추가 entity.setModelType(model); // 20250212 modeltype추가

View File

@@ -104,6 +104,9 @@ public class ModelTrainMngCoreService {
if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) { if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) {
throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND); throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND);
} }
// 하이퍼 파라미터 사용회수 업데이트
hyperParamEntity.setTotalUseCnt(
hyperParamEntity.getTotalUseCnt() == null ? 1 : hyperParamEntity.getTotalUseCnt() + 1);
String modelVer = String modelVer =
String.join( String.join(

View File

@@ -310,6 +310,9 @@ public class ModelHyperParamEntity {
@Column(name = "default_param") @Column(name = "default_param")
private Boolean isDefault = false; private Boolean isDefault = false;
@Column(name = "total_use_cnt")
private Integer totalUseCnt = 0;
public HyperParamDto.Basic toDto() { public HyperParamDto.Basic toDto() {
return new HyperParamDto.Basic( return new HyperParamDto.Basic(
this.modelType, this.modelType,

View File

@@ -296,14 +296,8 @@ public class TrainJobService {
e); e);
// 런타임 예외로 래핑하되, 메시지에 핵심 정보 포함 // 런타임 예외로 래핑하되, 메시지에 핵심 정보 포함
throw new IllegalStateException( throw new CustomApiException(
"tmp dataset build failed: modelUuid=" "INTERNAL_SERVER_ERROR", HttpStatus.INTERNAL_SERVER_ERROR, "임시 데이터셋 생성에 실패했습니다.");
+ modelUuid
+ ", modelId="
+ modelId
+ ", tmpRaw="
+ raw,
e);
} }
return modelUuid; return modelUuid;
} }