#35 · Primary category: Classical Machine Learning Frameworks

scikit-plot

data-science machine-learning plot plotting scikit-learn visualization

An intuitive library to add plotting functionality to scikit-learn objects.

Project last updated:08/20/24

GitHub Stars

2.4K

Forks

285

Contributors

13

License

MIT

Why we included this project

Anyone who has trained a scikit-learn model and then wrestled with raw matplotlib to look at the results knows how much extra work that step costs. Scikit-plot turns the usual model-evaluation charts into one-liners: ROC and precision-recall curves, confusion matrices with class labels, calibration and learning curves, feature-importance and elbow plots all come straight from a fitted estimator and test data. The library deliberately stays small, wrapping scikit-learn and matplotlib instead of replacing them, so it drops into an existing workflow without touching how you train. Since the helpers take predicted probabilities rather than scikit-learn objects, they also work on classifiers built outside scikit-learn. For anyone doing classical supervised learning and wanting consistent, presentable figures across experiments, it earns a permanent spot in the notebook.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category