This document demonstrates how to perform k-means clustering in Python using the scikit-learn library. It imports necessary libraries like Pandas, Matplotlib, Seaborn, and scikit-learn. It loads passenger data, preprocesses it to extract features for clustering, and applies k-means clustering with different values of k. Elbow method is used to find the optimal number of clusters. Finally, it visualizes the clustered data and cluster centroids.
This document demonstrates how to perform k-means clustering in Python using the scikit-learn library. It imports necessary libraries like Pandas, Matplotlib, Seaborn, and scikit-learn. It loads passenger data, preprocesses it to extract features for clustering, and applies k-means clustering with different values of k. Elbow method is used to find the optimal number of clusters. Finally, it visualizes the clustered data and cluster centroids.
This document demonstrates how to perform k-means clustering in Python using the scikit-learn library. It imports necessary libraries like Pandas, Matplotlib, Seaborn, and scikit-learn. It loads passenger data, preprocesses it to extract features for clustering, and applies k-means clustering with different values of k. Elbow method is used to find the optimal number of clusters. Finally, it visualizes the clustered data and cluster centroids.
This document demonstrates how to perform k-means clustering in Python using the scikit-learn library. It imports necessary libraries like Pandas, Matplotlib, Seaborn, and scikit-learn. It loads passenger data, preprocesses it to extract features for clustering, and applies k-means clustering with different values of k. Elbow method is used to find the optimal number of clusters. Finally, it visualizes the clustered data and cluster centroids.
Download as TXT, PDF, TXT or read online from Scribd
Download as txt, pdf, or txt
You are on page 1/ 2
unsupervised learning algorithm - k means clustering -in python - using scikit
#Pandas is a Python library. Pandas is used to analyze data.
import pandas as pd
#Matplotlib is a comprehensive library for creating static, animated, and
interactive visualizations in Python. import matplotlib.pyplot as plt
#Seaborn is more comfortable in handling Pandas data frames, seaborn uses
matplotlib to draw its plots. import seaborn as sns #Scikit-learn is probably the most useful library for machine learning in Python. The sklearn library contains a lot of efficient tools for machine learning and statistical modeling including classification, regression, clustering and dimensionality reduction from sklearn.cluster import KMeans
#style - whitegrid - graph will be in white with box
#why elbow to find how many number of clusters are there in the given data plt.title('The Elbow Method') plt.xlabel('Number of clusters') plt.ylabel('WCSS') plt.show() #kmeans++ - to iterate values kmeans = KMeans(n_clusters = 2, init= 'k-means++')
#y_kmeans is to predict which values will be fit to which cluster
y_kmeans = kmeans.fit_predict(X)
#again plotting an cluster
plt.figure(figsize=(15,7))
#here Y values will be plot,The scatter() method in the matplotlib library is used to draw a scatter plot plt.scatter(X[y_kmeans == 0, 0], X[y_kmeans == 0,1], color = 'yellow', label = 'Cluster 1', s=15) plt.scatter(X[y_kmeans == 1, 0], X[y_kmeans == 1,1], color = 'blue', label = 'Cluster 2', s=15)
plt.scatter(kmeans.cluster_centers_[:, 0], kmeans.cluster_centers_[:,1], color =
