Browse Source

优化主动学习

pull/227/head
chenzhihang 10 months ago
parent
commit
765f37ab7c
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java

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

@@ -298,12 +298,15 @@ public class ActiveLearnInsServiceImpl implements ActiveLearnInsService {
switch ((String) dataMap.get("framework_type")) { switch ((String) dataMap.get("framework_type")) {
case Constant.AL_PYTORCH: { case Constant.AL_PYTORCH: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth"); ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth");
break;
} }
case Constant.AL_SKLEARN: { case Constant.AL_SKLEARN: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model.joblib"); ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model.joblib");
break;
} }
case Constant.AL_KERAS: { case Constant.AL_KERAS: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/model.h5"); ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/model.h5");
break;
} }
} }




Loading…
Cancel
Save