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