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

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();
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추가

View File

@@ -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(

View File

@@ -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,

View File

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