(TIL) Sklearn: Pipeline diagram

less than 1 minute read

Estimators can be displayed with a HTML representation when shown in a jupyter notebook. This can be useful to diagnose or visualize a Pipeline with many estimators. This visualization is activated by setting the display option in sklearn.set_config:

from sklearn import set_config

# displays HTML representation in a jupyter context

Model diagram

Via towards data science and scikit-learn.org.