MVPTeam/kamco-train-api
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

전이학습 상세 - before dataset 추가

Browse Source
This commit is contained in:
Gina(박가연)
2026-02-20 16:05:29 +09:00
parent 564a99448c
commit 83859bb9fe
7 changed files with 44 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java
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;
}