|
|
|
@@ -581,13 +581,15 @@ public class ExperimentServiceImpl implements ExperimentService { |
|
|
|
if (trainDatasetList != null) { |
|
|
|
List<NewDatasetVo> trainDatasets = new ArrayList<>(); |
|
|
|
for (Map<String, Object> dataset : trainDatasetList) { |
|
|
|
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")); |
|
|
|
trainDatasets.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")); |
|
|
|
trainDatasets.add(newDatasetVo); |
|
|
|
} |
|
|
|
} |
|
|
|
modelMetaVo.setTrainDatasets(trainDatasets); |
|
|
|
} |
|
|
|
|