| # Getting data | |
| from sklearn.datasets import load_breast_cancer | |
| cancer = load_breast_cancer() | |
| classes=cancer.target_names | |
| X = cancer.data | |
| y = cancer.target | |
| # Importing ClassBalance visualizer | |
| from yellowbrick.target import ClassBalance | |
| # Creating the class imbalance plot | |
| visualizer = ClassBalance(labels=classes) | |
| visualizer.fit(y) | |
| # Saving plot in PNG format | |
| visualizer.show(outpath="Class_Imbalance_Plot.png") |