[ENH] Improve and fix some bugs in plot_model

alegonz / baikal

A graph-based functional API for building complex scikit-learn pipelines.

https://baikal.readthedocs.io

BSD 3-Clause "New" or "Revised" License

592 stars 30 forks source link

Closed alegonz closed 4 years ago

alegonz commented 4 years ago

Changes:

Add option to include targets (include_targets).
Fix bug of mangled sub-graphs when there were two or more sub-models.
Fix bug of missing edges when outputs came from a sub-model.
Return dot graph object, so that client code can generate an image and embed it into custom plots.
Refactor.