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