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 f7b743eb..fb84362b 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 @@ -296,7 +296,7 @@ public class ExperimentServiceImpl implements ExperimentService { ExperimentIns insert = experimentInsService.insert(experimentIns); //插入到模型依赖关系表 if (dependendcy != null && trainInfo != null) { - insertModelDependencyNew(dependendcy, trainInfo, output, insert.getId(), experiment.getName(), experiment.getGlobalParam()); + insertModelDependencyNew(dependendcy, trainInfo, output, insert.getId(), experiment.getId(), experiment.getName(), experiment.getGlobalParam()); } Map datasetDependendcy = (Map) converMap2.get("dataset_dependency"); @@ -490,7 +490,7 @@ public class ExperimentServiceImpl implements ExperimentService { } - private void insertModelDependencyNew(Map dependendcy, Map trainInfo, Map output, Integer experimentInsId, String experimentName, String globalParam) throws Exception { + private void insertModelDependencyNew(Map dependendcy, Map trainInfo, Map output, Integer experimentInsId, Integer experimentId, String experimentName, String globalParam) throws Exception { Iterator> dependendcyIterator = dependendcy.entrySet().iterator(); Map modelTrain = (Map) trainInfo.get("model_train"); Map modelEvaluate = (Map) trainInfo.get("model_evaluate"); @@ -521,7 +521,7 @@ public class ExperimentServiceImpl implements ExperimentService { Map imagesMap = (Map) modelTrainMap.get("image"); modelMetaVo.setImage(imagesMap.get("name")); - //处理训练参数 todo + //处理训练参数 HashMap trainParam = new HashMap<>(); JSONArray jsonArray = JSONArray.parseArray(globalParam); for (int i = 0; i < jsonArray.size(); i++) { @@ -569,6 +569,7 @@ public class ExperimentServiceImpl implements ExperimentService { trainTask.setTaskId(sourceTaskId); trainTask.setInsId(experimentInsId); trainTask.setName(experimentName); + trainTask.setExperimentId(experimentId); modelMetaVo.setTrainTask(trainTask); //处理test数据 @@ -594,7 +595,6 @@ public class ExperimentServiceImpl implements ExperimentService { //处理target数据 LoginUser loginUser = SecurityUtils.getLoginUser(); String gitLinkUsername = loginUser.getSysUser().getGitLinkUsername(); -// String gitLinkUsername = "fanshuai"; modelMetaVo.setOwner(gitLinkUsername); if (target != null) { for (int i = 0; i < target.size(); i++) {