|
|
|
@@ -199,9 +199,9 @@ public class AutoMlServiceImpl implements AutoMlService { |
|
|
|
String outputPath = minioEndpoint + "/" + output2.get("path").replace("{{workflow.name}}", (String) metadata.get("name")) + "/"; |
|
|
|
autoMlIns.setModelPath(outputPath + "save_model.joblib"); |
|
|
|
if (Constant.AutoMl_Classification.equals(autoMl.getTaskType())) { |
|
|
|
autoMlIns.setImgPath(outputPath + "Auto-sklearn_accuracy_over_time.png" + "," + outputPath + "Train_Confusion_Matrix.png" + "," + outputPath + "Test_Confusion_Matrix.png"); |
|
|
|
autoMlIns.setImgPath(outputPath + "Auto-sklearn_metric_over_time.png" + "," + outputPath + "Train_Confusion_Matrix.png" + "," + outputPath + "Test_Confusion_Matrix.png"); |
|
|
|
} else { |
|
|
|
autoMlIns.setImgPath(outputPath + "Auto-sklearn_accuracy_over_time.png" + "," + outputPath + "regression.png"); |
|
|
|
autoMlIns.setImgPath(outputPath + "Auto-sklearn_metric_over_time.png" + "," + outputPath + "regression.png"); |
|
|
|
} |
|
|
|
autoMlIns.setResultPath(outputPath + "result.txt"); |
|
|
|
String seed = autoMl.getSeed() != null ? String.valueOf(autoMl.getSeed()) : "1"; |
|
|
|
|