|
|
|
@@ -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; |
|
|
|
} |
|
|
|
} |
|
|
|
|