Customer Segmentation Model
This repository contains a trained K-Means clustering model for customer segmentation, along with a scaler to preprocess input features.
Files
kmeans_model.pklβ Trained K-Means modelscaler.pklβ StandardScaler used to normalize features before clustering
Description
The 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.
Input Features
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)
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support