Save sklearn pipeline diagram

Question

I want to save a pipeline displayed with the set_config(display="diagram") command, see below:

Is there a better way to do it than by taking a screenshot?

Treillage

Perhaps this may help... – Prichard

Thanks @amiola, I tried but it gives me the equivalent of what is shown in my Jupyter Notebook, but on a web page. Printing the content of that web page formats it with page number, url and other things that usually come in the frame of printing a web page. Unfortunately I cannot use that in a report. – Treillage

Answer 1

from sklearn.utils import estimator_html_repr

with open("pipeline.html", "w") as f:
    f.write(estimator_html_repr(pipeline))

Concoction

This does not answer the question since the OP clearly asks how to create an image without having to take a screenshot. I might add: could it even be saved as a vector graphic? – Flanna