Browse Source

自动机器学习优化

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

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

@@ -199,7 +199,9 @@ public class MachineLearnServiceImpl implements MachineLearnService {
break;
}
case Constant.ML_TextClassification: {
machineLearnIns.setModelPath(outputPath + "/saved_dict/" + modelType + ".ckpt");
machineLearnIns.setModelPath(outputPath + "saved_dict/" + modelType + ".ckpt");
machineLearnIns.setRunHistoryPath(outputPath + "log/" + modelType);
machineLearnIns.setResultPath(outputPath + "log/" + modelType + "/result.txt");
break;
}
case Constant.ML_VideoClassification: {


Loading…
Cancel
Save