Browse Source

自动机器学习优化测试

pull/225/head
chenzhihang 9 months ago
parent
commit
bd643d235c
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java

+ 2
- 1
ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/MachineLearnServiceImpl.java View File

@@ -183,6 +183,7 @@ public class MachineLearnServiceImpl implements MachineLearnService {
Map<String, Object> param_output = (Map<String, Object>) output.get("param_output"); Map<String, Object> param_output = (Map<String, Object>) output.get("param_output");
List output1 = (ArrayList) param_output.values().toArray()[0]; List output1 = (ArrayList) param_output.values().toArray()[0];
Map<String, String> output2 = (Map<String, String>) output1.get(0); Map<String, String> output2 = (Map<String, String>) output1.get(0);
String s = output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/".substring("data/".length());
String outputPath = minioEndpoint + "/" + output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/"; String outputPath = minioEndpoint + "/" + output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/";


switch (machineLearn.getType()) { switch (machineLearn.getType()) {
@@ -200,7 +201,7 @@ public class MachineLearnServiceImpl implements MachineLearnService {
} }
case Constant.ML_TextClassification: { case Constant.ML_TextClassification: {
machineLearnIns.setModelPath(outputPath + "saved_dict/" + modelType + ".ckpt"); machineLearnIns.setModelPath(outputPath + "saved_dict/" + modelType + ".ckpt");
machineLearnIns.setRunHistoryPath(outputPath + "log/" + modelType);
machineLearnIns.setRunHistoryPath(output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/".substring("data/".length()) + "log/" + modelType);
machineLearnIns.setResultPath(outputPath + "log/" + modelType + "/result.txt"); machineLearnIns.setResultPath(outputPath + "log/" + modelType + "/result.txt");
break; break;
} }


Loading…
Cancel
Save