|
|
|
@@ -581,7 +581,7 @@ public class ExperimentServiceImpl implements ExperimentService { |
|
|
|
if (trainDatasetList != null) { |
|
|
|
List<NewDatasetVo> trainDatasets = new ArrayList<>(); |
|
|
|
for (Map<String, Object> dataset : trainDatasetList) { |
|
|
|
if(!dataset.containsKey("dataset_ref")) { |
|
|
|
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")); |
|
|
|
@@ -631,13 +631,15 @@ public class ExperimentServiceImpl implements ExperimentService { |
|
|
|
List<Map<String, Object>> testDatasetList = (List<Map<String, Object>>) evaluateMap.get("datasets"); |
|
|
|
List<NewDatasetVo> testDatasets = new ArrayList<>(); |
|
|
|
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); |
|
|
|
} |
|
|
|
|