diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java index f69d9586..d6232933 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ExperimentServiceImpl.java @@ -473,8 +473,8 @@ public class ExperimentServiceImpl implements ExperimentService { // 拼接需要的参数 Map projectMap = (Map) datasetPreprocessMap.get("project"); Map sourceParams = new HashMap<>(); - sourceParams.put("experiment_name", experimentName); - sourceParams.put("experiment_ins_id", experimentInsId); + sourceParams.put("name", experimentName); + sourceParams.put("ins_id", experimentInsId); sourceParams.put("experiment_id", experimentId); sourceParams.put("train_name", sourceTaskId); sourceParams.put("preprocess_code", projectMap); diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java index 67891f13..5df6e006 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/NewDatasetServiceImpl.java @@ -511,6 +511,7 @@ public class NewDatasetServiceImpl implements NewDatasetService { Map preprocessCode = (Map) sourceMap.get("preprocess_code"); sourceMap.remove("preprocess_code"); datasetVo.setTrainTask(sourceMap); + datasetVo.setProcessingCode(preprocessCode); datasetTempStorage.setState(0); datasetTempStorageService.update(datasetTempStorage); }