A machine learning project that segments retail customers into meaningful groups using K-Means Clustering and customer behavior analysis. The project includes exploratory data analysis, feature engineering, dimensionality reduction, cluster evaluation, and business recommendations for targeted marketing.
Customer segmentation helps businesses understand different customer groups based on purchasing behavior and demographics. This project applies unsupervised machine learning techniques to identify customer segments that can support personalized marketing strategies and improve customer engagement.
- Exploratory Data Analysis (EDA)
- Data preprocessing and scaling
- K-Means clustering
- Optimal cluster selection (Elbow Method & Silhouette Score)
- PCA visualization
- Cluster profiling
- Customer behavior analysis
- Business recommendations
- Python
- Pandas
- NumPy
- Matplotlib
- Seaborn
- Scikit-learn
- Jupyter Notebook
Mall Customers Dataset
Features include:
- Customer ID
- Gender
- Age
- Annual Income
- Spending Score
- K-Means Clustering
- Elbow Method
- Silhouette Score
- Davies-Bouldin Index
Dataset
│
▼
Data Cleaning
│
▼
Exploratory Data Analysis
│
▼
Feature Scaling
│
▼
K-Means Clustering
│
▼
Cluster Evaluation
│
▼
PCA Visualization
│
▼
Customer Insights
customer-segmentation/
│── customer_segmentation.ipynb
│── README.md
│── requirements.txt
│── dataset/
│── images/
└── outputs/
- Identified distinct customer groups based on spending patterns.
- Compared clustering performance using multiple evaluation metrics.
- Visualized customer clusters using PCA.
- Generated actionable business recommendations for targeted marketing.
git clone https://github.com/YOUR_USERNAME/customer-segmentation.gitcd customer-segmentationpip install -r requirements.txtjupyter notebookRun:
customer_segmentation.ipynb
- Hierarchical Clustering
- DBSCAN Clustering
- Interactive Streamlit Dashboard
- Real-world retail dataset integration
- Customer lifetime value prediction
This project is intended for educational and research purposes.
Rithvik Jaswal
Computer Science Engineering Student
Interested in Data Science, Machine Learning, Artificial Intelligence, and Business Analytics.
⭐ If you found this project useful, consider giving it a star.