Browse Source

优化流水线转换bug

dev-complex-computation
chenzhihang 1 year ago
parent
commit
b860e9a4d4
1 changed files with 10 additions and 8 deletions
  1. +10
    -8
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java

+ 10
- 8
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java View File

@@ -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);
}


Loading…
Cancel
Save