Upload 2 files

README.md

@@ -0,0 +1,55 @@
+---
+
+tags:
+
+\- machine-learning
+
+\- clustering
+
+\- k-means
+
+\- customer-segmentation
+
+
+
+
+\# Customer Segmentation K-Means Model
+
+
+
+This repository contains a K-Means clustering model for customer segmentation.
+
+
+
+\## Model Description
+
+The model segments customers based on three features: `Age`, `Annual Income (k$)`, and `Spending Score (1-100)`. K-Means clustering was chosen due to its simplicity, efficiency for this type of unsupervised task, and the interpretability of the resulting clusters. The optimal number of clusters was determined using the Elbow Method.
+
+
+
+\## Training Data
+
+The model was trained on a dataset of \*\*200 customers\*\*, with features: `Age`, `Annual Income (k$)`, and `Spending Score (1-100)`.
+
+
+
+\## Usage
+
+To use this model for prediction, you need to provide input in the format:
+
+`{"age": <int>, "annual\_income": <float>, "spending\_score": <float>}`
+
+
+
+\## Example
+
+\*\*Input:\*\*
+
+`{"age": 30, "annual\_income": 50, "spending\_score": 60}`
+
+
+
+\*\*Output (example):\*\*
+
+`{"cluster\_id": 1}`
+

requirements.txt

@@ -0,0 +1,4 @@
+pandas
+scikit-learn
+joblib
+flask # or fastapi, uvicorn