모델학습 데이터셋 선택 목록 수정

src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java

@@ -2,12 +2,14 @@ package com.kamco.cd.training.model.service;
 
 import com.kamco.cd.training.common.dto.HyperParam;
 import com.kamco.cd.training.common.enums.HyperParamSelectType;
+import com.kamco.cd.training.dataset.dto.DatasetDto.SelectDataSet;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.model.dto.ModelConfigDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq;
 import com.kamco.cd.training.postgres.core.HyperParamCoreService;
 import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService;
+import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -85,4 +87,19 @@ public class ModelTrainMngService {
   public ModelConfigDto.Basic getModelConfigByModelId(UUID uuid) {
     return modelTrainMngCoreService.findModelConfigByModelId(uuid);
   }
+
+  /**
+   * 모델별 데이터셋 목록 조회
+   *
+   * @param modelType
+   * @param selectType
+   * @return
+   */
+  public List<SelectDataSet> getDatasetSelectList(String modelType, String selectType) {
+    if (modelType.equals("G1")) {
+      return modelTrainMngCoreService.getDatasetSelectM1List(modelType, selectType);
+    } else {
+      return modelTrainMngCoreService.getDatasetSelectM2M3List(modelType, selectType);
+    }
+  }
 }