| @@ -581,7 +581,7 @@ public class ExperimentServiceImpl implements ExperimentService { | |||||
| if (trainDatasetList != null) { | if (trainDatasetList != null) { | ||||
| List<NewDatasetVo> trainDatasets = new ArrayList<>(); | List<NewDatasetVo> trainDatasets = new ArrayList<>(); | ||||
| for (Map<String, Object> dataset : trainDatasetList) { | for (Map<String, Object> dataset : trainDatasetList) { | ||||
| if(!dataset.containsKey("dataset_ref")) { | |||||
| if (!dataset.containsKey("dataset_ref")) { | |||||
| NewDatasetVo newDatasetVo = new NewDatasetVo(); | NewDatasetVo newDatasetVo = new NewDatasetVo(); | ||||
| newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); | newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); | ||||
| newDatasetVo.setName((String) dataset.get("dataset_name")); | newDatasetVo.setName((String) dataset.get("dataset_name")); | ||||
| @@ -631,13 +631,15 @@ public class ExperimentServiceImpl implements ExperimentService { | |||||
| List<Map<String, Object>> testDatasetList = (List<Map<String, Object>>) evaluateMap.get("datasets"); | List<Map<String, Object>> testDatasetList = (List<Map<String, Object>>) evaluateMap.get("datasets"); | ||||
| List<NewDatasetVo> testDatasets = new ArrayList<>(); | List<NewDatasetVo> testDatasets = new ArrayList<>(); | ||||
| for (Map<String, Object> dataset : testDatasetList) { | for (Map<String, Object> dataset : testDatasetList) { | ||||
| NewDatasetVo newDatasetVo = new NewDatasetVo(); | |||||
| newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); | |||||
| newDatasetVo.setName((String) dataset.get("dataset_name")); | |||||
| newDatasetVo.setVersion((String) dataset.get("dataset_version")); | |||||
| newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); | |||||
| newDatasetVo.setOwner((String) dataset.get("owner")); | |||||
| testDatasets.add(newDatasetVo); | |||||
| if (!dataset.containsKey("dataset_ref")) { | |||||
| NewDatasetVo newDatasetVo = new NewDatasetVo(); | |||||
| newDatasetVo.setId(Integer.valueOf((String) dataset.get("dataset_id"))); | |||||
| newDatasetVo.setName((String) dataset.get("dataset_name")); | |||||
| newDatasetVo.setVersion((String) dataset.get("dataset_version")); | |||||
| newDatasetVo.setIdentifier((String) dataset.get("dataset_identifier")); | |||||
| newDatasetVo.setOwner((String) dataset.get("owner")); | |||||
| testDatasets.add(newDatasetVo); | |||||
| } | |||||
| } | } | ||||
| modelMetaVo.setTestDatasets(testDatasets); | modelMetaVo.setTestDatasets(testDatasets); | ||||
| } | } | ||||