select-dataset-list api solarPanelCnt 추가, spotless 적용
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package com.kamco.cd.training.model;
|
||||
|
||||
import com.kamco.cd.training.common.dto.MonitorDto;
|
||||
import com.kamco.cd.training.common.enums.ModelType;
|
||||
import com.kamco.cd.training.common.service.SystemMonitorService;
|
||||
import com.kamco.cd.training.config.api.ApiResponseDto;
|
||||
import com.kamco.cd.training.dataset.dto.DatasetDto;
|
||||
@@ -143,9 +144,9 @@ public class ModelTrainMngApiController {
|
||||
@Parameter(
|
||||
description = "모델 구분",
|
||||
example = "",
|
||||
schema = @Schema(allowableValues = {"G1", "G2", "G3"}))
|
||||
schema = @Schema(allowableValues = {"G1", "G2", "G3", "G4"}))
|
||||
@RequestParam
|
||||
String modelType,
|
||||
ModelType modelType,
|
||||
@Parameter(
|
||||
description = "선택 구분",
|
||||
example = "",
|
||||
@@ -153,7 +154,7 @@ public class ModelTrainMngApiController {
|
||||
@RequestParam
|
||||
String selectType) {
|
||||
DatasetReq req = new DatasetReq();
|
||||
req.setModelNo(modelType);
|
||||
req.setModelNo(modelType.getId());
|
||||
req.setDataType(selectType);
|
||||
return ApiResponseDto.ok(modelTrainMngService.getDatasetSelectList(req));
|
||||
}
|
||||
|
||||
@@ -51,10 +51,10 @@ public class ModelTrainMngService {
|
||||
|
||||
@Value("${train.docker.response_dir}")
|
||||
private String responseDir;
|
||||
|
||||
@Value("${train.docker.symbolic_link_dir}")
|
||||
private String symbolicDir;
|
||||
|
||||
|
||||
/**
|
||||
* 모델학습 조회
|
||||
*
|
||||
@@ -324,6 +324,8 @@ public class ModelTrainMngService {
|
||||
public List<SelectDataSet> getDatasetSelectList(DatasetReq req) {
|
||||
if (req.getModelNo().equals(ModelType.G1.getId())) {
|
||||
return modelTrainMngCoreService.getDatasetSelectG1List(req);
|
||||
} else if (req.getModelNo().equals(ModelType.G4.getId())) {
|
||||
return modelTrainMngCoreService.getDatasetSelectG4List(req);
|
||||
} else {
|
||||
return modelTrainMngCoreService.getDatasetSelectG2G3List(req);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user