| # Getting data | |
| from sklearn.datasets import load_iris | |
| iris = load_iris() | |
| classes=iris.target_names | |
| X = iris.data | |
| y = iris.target | |
| # Importing KElbowVisualizer | |
| from yellowbrick.cluster import KElbowVisualizer | |
| # Importing K-Means model | |
| from sklearn.cluster import KMeans | |
| # Creating the Elbow plot | |
| visualizer = KElbowVisualizer(KMeans(), k=(2,11)) | |
| visualizer.fit(X) | |
| # Saving plot in PNG format | |
| visualizer.show(outpath="Elbow_Plot.png") |