전이학습 상세 - before dataset 추가

This commit is contained in:
2026-02-20 16:05:29 +09:00
parent 564a99448c
commit 83859bb9fe
7 changed files with 44 additions and 1 deletions

View File

@@ -179,6 +179,7 @@ public class ModelTrainDetailDto {
private ModelConfigDto.Basic etcConfig;
private TransferHyperSummary modelTrainHyper;
private List<SelectDataSet> modelTrainDataset;
private List<SelectDataSet> beforeTrainDataset;
}
@Getter

View File

@@ -47,6 +47,8 @@ public class ModelTrainMngDto {
private ZonedDateTime packingStrtDttm;
private ZonedDateTime packingEndDttm;
private Long beforeModelId;
public String getStatusName() {
if (this.statusCd == null || this.statusCd.isBlank()) return null;
try {

View File

@@ -96,10 +96,34 @@ public class ModelTrainDetailService {
dataSets = mngCoreService.getDatasetSelectG2G3List(datasetReq);
}
DatasetReq beforeDatasetReq = new DatasetReq();
List<Long> beforeDatasetIds = new ArrayList<>();
List<SelectDataSet> beforeDataSets = new ArrayList<>();
Long beforeModelId = modelInfo.getBeforeModelId();
if (beforeModelId != null) {
Basic beforeInfo = modelTrainDetailCoreService.findByModelBeforeId(beforeModelId);
List<MappingDataset> beforeDatasets =
modelTrainDetailCoreService.getByModelMappingDataset(beforeInfo.getUuid());
for (MappingDataset before : beforeDatasets) {
beforeDatasetIds.add(before.getDatasetId());
}
beforeDatasetReq.setIds(beforeDatasetIds);
beforeDatasetReq.setModelNo(modelInfo.getModelNo());
if (beforeInfo.getModelNo().equals(ModelType.G1.getId())) {
beforeDataSets = mngCoreService.getDatasetSelectG1List(beforeDatasetReq);
} else {
beforeDataSets = mngCoreService.getDatasetSelectG2G3List(beforeDatasetReq);
}
}
TransferDetailDto transferDetailDto = new TransferDetailDto();
transferDetailDto.setEtcConfig(configInfo);
transferDetailDto.setModelTrainHyper(hyperSummary);
transferDetailDto.setModelTrainDataset(dataSets);
transferDetailDto.setBeforeTrainDataset(beforeDataSets);
return transferDetailDto;
}