하이퍼파라미터 기능 추가
This commit is contained in:
@@ -52,7 +52,8 @@ public class ModelTrainService {
|
||||
}
|
||||
|
||||
// 3. 하이퍼파라미터 목록
|
||||
List<ModelMngDto.HyperParamInfo> hyperParams = hyperParamCoreService.findAllActiveHyperParams();
|
||||
List<ModelMngDto.HyperParamInfo> hyperParams =
|
||||
null; // hyperParamCoreService.findAllActiveHyperParams();
|
||||
|
||||
// 4. 데이터셋 목록
|
||||
List<ModelMngDto.DatasetInfo> datasets = datasetCoreService.findAllActiveDatasetsForTraining();
|
||||
@@ -65,26 +66,6 @@ public class ModelTrainService {
|
||||
.build();
|
||||
}
|
||||
|
||||
/**
|
||||
* 하이퍼파라미터 등록
|
||||
*
|
||||
* @param createReq 등록 요청
|
||||
* @return 생성된 버전명
|
||||
*/
|
||||
@Transactional
|
||||
public String createHyperParam(ModelMngDto.HyperParamCreateReq createReq) {
|
||||
// 신규 버전 추가 시 baseHyperVer가 없으면 H1으로 설정
|
||||
if (createReq.getBaseHyperVer() == null || createReq.getBaseHyperVer().isEmpty()) {
|
||||
String firstVersion = hyperParamCoreService.getFirstHyperParamVersion();
|
||||
createReq.setBaseHyperVer(firstVersion);
|
||||
log.info("baseHyperVer가 없어 첫 번째 버전으로 설정: {}", firstVersion);
|
||||
}
|
||||
|
||||
String newVersion = hyperParamCoreService.createHyperParam(createReq);
|
||||
log.info("하이퍼파라미터 등록 완료: {}", newVersion);
|
||||
return newVersion;
|
||||
}
|
||||
|
||||
/**
|
||||
* 하이퍼파라미터 단건 조회
|
||||
*
|
||||
@@ -245,10 +226,10 @@ public class ModelTrainService {
|
||||
.lastEpoch(entity.getLastCheckpointEpoch())
|
||||
.totalEpoch(entity.getEpochCnt())
|
||||
.checkpointPath(entity.getCheckpointPath())
|
||||
.failedAt(
|
||||
entity.getStopDttm() != null
|
||||
? entity.getStopDttm().atZone(java.time.ZoneId.systemDefault())
|
||||
: null)
|
||||
// .failedAt(
|
||||
// entity.getStopDttm() != null
|
||||
// ? entity.getStopDttm().atZone(java.time.ZoneId.systemDefault())
|
||||
// : null)
|
||||
.build();
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user