mithun-ai commited on
Commit
ba55893
·
verified ·
1 Parent(s): 4166c95

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +68 -0
README.md CHANGED
@@ -1,3 +1,71 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+ - customer-segmentation
5
+ - machine-learning
6
+ - clustering
7
+ - dbscan
8
+ - scikit-learn
9
  ---
10
+
11
+ # Customer Segmentation - DBSCAN
12
+
13
+ ## Overview
14
+
15
+ This model performs customer segmentation using the DBSCAN clustering algorithm. It identifies groups of customers based on similarities in demographic and behavioral features while also detecting customers that do not belong to any major group.
16
+
17
+ ## Model Type
18
+
19
+ DBSCAN Clustering
20
+
21
+ ## Features Used
22
+
23
+ The model was trained using the following features:
24
+
25
+ - Gender
26
+ - Ever Married
27
+ - Age
28
+ - Graduated
29
+ - Profession
30
+ - Work Experience
31
+ - Spending Score
32
+ - Family Size
33
+ - Age Group
34
+ - Family Category
35
+
36
+ ## Output
37
+
38
+ The model assigns a cluster label to each customer.
39
+
40
+ Example:
41
+
42
+ - Cluster 0
43
+ - Cluster 1
44
+ - Cluster 2
45
+ - -1 (Noise / Outlier)
46
+
47
+ A label of `-1` indicates that the customer was identified as an outlier and does not belong to any major cluster.
48
+
49
+ ## Files Required
50
+
51
+ - `dbscan_model.joblib`
52
+ - `scaler.joblib`
53
+
54
+ ## Usage
55
+
56
+ ```python
57
+ import joblib
58
+ import numpy as np
59
+ scaler = joblib.load("scaler.joblib")
60
+ model = joblib.load("dbscan_model.joblib")
61
+ customer = np.array([[...]])
62
+ customer_scaled = scaler.transform(customer)
63
+ # DBSCAN does not support direct prediction on new samples.
64
+ # The model is primarily intended for clustering datasets.
65
+ labels = model.fit_predict(customer_scaled)
66
+ print(labels)
67
+ ```
68
+
69
+ ## Author
70
+
71
+ Mithun