(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

set_config(display='diagram')
# displays HTML representation in a jupyter context
model

Model diagram

Via towards data science and scikit-learn.org.

Comments