Browse Source

优化主动学习

pull/268/head
chenzhihang 8 months ago
parent
commit
4f415719f8
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java

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

@@ -309,15 +309,15 @@ public class ActiveLearnInsServiceImpl implements ActiveLearnInsService {
Map<String, Object> dataMap = (Map<String, Object>) paramMap.get("data");
switch ((String) dataMap.get("framework_type")) {
case Constant.AL_PYTORCH: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth");
ins.setResultPath(endpoint + "/" + directoryPath + "/best_checkpoint/best_model_weights.pth");
break;
}
case Constant.AL_SKLEARN: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model.joblib");
ins.setResultPath(endpoint + "/" + directoryPath + "/best_checkpoint/best_model.joblib");
break;
}
case Constant.AL_KERAS: {
ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/model.h5");
ins.setResultPath(endpoint + "/" + directoryPath + "/best_checkpoint/best_model.h5");
break;
}
}


Loading…
Cancel
Save