New Function
This is a new function introduced in PyCaret 2.0x.
AutoML
This function returns the best model out of all models created in the current active environment based on metric defined in optimize parameter. Run this code at the end of your script.
This function is only available in pycaret.classification and pycaret.regression modules.
Example
Code
# loading dataset from pycaret.datasets import get_data data = get_data('diabetes') # initializing setup from pycaret.classification import * clf1 = setup(data, target = 'Class variable') # compare all baseline models and select top 5 top5 = compare_models(n_select = 5) # tune top 5 base models tuned_top5 = [tune_model(i) for i in top5] # ensemble top 5 tuned models bagged_top5 = [ensemble_model(i) for i in tuned_top5] # blend top 5 base models blender = blend_models(estimator_list = top5) # select best model best = automl(optimize = 'Recall')
Output
No output. Best performing model returned.
Try this next
Was this page helpful?
GitHub