| # Getting data | |
| from sklearn.datasets import load_iris | |
| iris = load_iris() | |
| classes=iris.target_names | |
| X = iris.data | |
| y = iris.target | |
| # Importing SilhouetteVisualizer | |
| from yellowbrick.cluster import SilhouetteVisualizer | |
| # Importing K-Means model | |
| from sklearn.cluster import KMeans | |
| # Creating the silhouette plot | |
| visualizer = SilhouetteVisualizer(KMeans(2, random_state=42), | |
| colors='yellowbrick') | |
| visualizer.fit(X) | |
| # Saving plot in PNG format | |
| visualizer.show(outpath="Silhouette_Plot.png") |