Downloads · 30 days
0
Marsill/CustomerSegmentation-kmeans
CustomerSegmentation-kmeans is a machine learning model from Marsill. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This repository contains a trained K-Means clustering model for customer segmentation, along with a scaler to preprocess input features.
Downloads · 30 days
0
Access
Public
Updated Sep 23, 2025
Repo size
4.6 KB
Likes
0
Public
Click a slice to open those files.
.pkl4.6 KB · 64%
From the Hugging Face model README
This repository contains a trained K-Means clustering model for customer segmentation, along with a scaler to preprocess input features.
kmeans_model.pkl – Trained K-Means modelscaler.pkl – StandardScaler used to normalize features before clusteringThe model segments customers into two clusters:
| Cluster | Label | Description |
|---|---|---|
| 0 | High Spender | Customers with high sales and profit |
| 1 | Occasional Buyer | Customers with lower or occasional purchases |
The model was trained on the Superstore dataset.
To predict the cluster of a customer, the model expects a dictionary with the following keys:
TotalSales – Total sales of the customerAvgSales – Average sales per orderTotalProfit – Total profit from the customerAvgProfit – Average profit per orderTotalQuantity – Total quantity purchasedAvgDiscount – Average discount received (0-1)