diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java index 0bbf41ad..041e1ad9 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/service/impl/ActiveLearnInsServiceImpl.java @@ -298,12 +298,15 @@ public class ActiveLearnInsServiceImpl implements ActiveLearnInsService { switch ((String) dataMap.get("framework_type")) { case Constant.AL_PYTORCH: { ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model_weights.pth"); + break; } case Constant.AL_SKLEARN: { ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/final_model.joblib"); + break; } case Constant.AL_KERAS: { ins.setResultPath(endpoint + "/" + directoryPath + "/final_checkpoint/model.h5"); + break; } }