So you’ve built a bunch of models for your project (or hackathon). A few of them are even giving you a really good AUC-ROC score. But how do you go about choosing one? Sure you can get into the nitty gritty of it but wouldn’t it help if you could just visualize them and pick the best one from there?
Yellowbrick is a suite of visualization tools that extend the scikit-learn API to enable acceleration and ease of model selection for data scientists. In simple words, this Python package combines the power of scikit-learn with the capabilities of matplotlib to generate intuitive visualizations of your models.
This set of tools are called Visualizers, which are basically objects that learn from the data and then create visualizations so you can gain a deeper insight into the model selection process. Some of the most popular visualizers are listed below:
Yellowbrick is compatible with Python 2.7 as well but the developers recommend using Python 3.5 to utilise this package to its maximum potential. Of course you will also need to install scikit-learn and matplotlib as well. To install Yellowbrick, use the pip method:
$ pip install yellowbrick
To see example of Yellowbrick in action and to replicate what the developers have demonstrated, head over to the GitHub page here.
I love this package! Adding visualization to any part of the data science process always helps in my opinion. We have previously covered the ANN visualiser which illustrates the artificial neural network you build and we have also seen the deep learning process being visualized.
Yellowbrick will really help you, as a data scientist, dig deeper into your model understanding and also save you time by portraying the process in the form of a neat diagram. I urge you to try out this library and let us know your experience using it.
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
Thanks Pranav. Do you know of libraries in R that are similar to yellowbrick?
Hi Venkat, I have only used the diverse ggplot2 library in R so haven't come across anything that's similar to Yellowbricks. Will update you if I find something.