전이학습 상세 - before dataset 추가
This commit is contained in:
@@ -179,6 +179,7 @@ public class ModelTrainDetailDto {
|
||||
private ModelConfigDto.Basic etcConfig;
|
||||
private TransferHyperSummary modelTrainHyper;
|
||||
private List<SelectDataSet> modelTrainDataset;
|
||||
private List<SelectDataSet> beforeTrainDataset;
|
||||
}
|
||||
|
||||
@Getter
|
||||
|
||||
@@ -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 {
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user