Browse Source

自动机器学习优化测试

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

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

@@ -183,7 +183,7 @@ public class MachineLearnServiceImpl implements MachineLearnService {
Map<String, Object> param_output = (Map<String, Object>) output.get("param_output");
List output1 = (ArrayList) param_output.values().toArray()[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 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")) + "/";

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


Loading…
Cancel
Save