[FIXED] How to download pipeline code from Pycaret AutoML into .py files?

Issue

PyCaret seems like a great AutoML tool. It works, fast and simple and I would like to download the generated pipeline code into .py files to double check and if needed to customize some parts. Unfortunately, I don't know how to make it real. Reading the documentation have not helped. Is it possible or not?

Solution

It is not possible to get the underlying code since PyCaret takes care of this for you. But it is up to you as the user to decide the steps that you want your flow to take e.g.

# Setup experiment with user-defined options for preprocessing, etc.
setup(...) 

# Create a model (uses training split only)
model = create_model("lr")

# Tune hyperparameters (user can pass a custom tuning grid if needed)
# Again, uses training split only
tuned = tune_model(model, ...)

# Finalize model (so that the best hyperparameters are retrained on the entire dataset
finalize_model(tuned)

# Any other steps you would like to do.
...

Finally, you can save the entire pipeline as a pkl file for use later

# Saves the model + pipeline as a pkl file
save_model(final, "my_best_model") 

Answered By - Nikhil Gupta