atdokmeci commited on
Commit
b4a9ad2
·
verified ·
1 Parent(s): d02b763

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -0
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Bank Customer Churn Prediction Model Card
2
+
3
+ ## Model Details
4
+
5
+ - **Architecture:** Artificial Neural Network (ANN) with 2 hidden layers (12 and 6 units, ReLU activation), output layer (1 unit, sigmoid activation).
6
+ - **Framework:** Keras (TensorFlow backend)
7
+ - **Input Features:** 14 normalized and engineered features:
8
+ - CreditScore, Gender, Age, Tenure, Balance, NumOfProducts, HasCrCard, IsActiveMember, EstimatedSalary, BalanceSalaryRatio, TenureByAge, Geography_France, Geography_Germany, Geography_Spain
9
+ - **Output:** Binary classification (Exited: 0 = retained, 1 = churned)
10
+
11
+ ## Intended Use
12
+
13
+ - Predict whether a bank customer will churn (exit) based on their profile and account activity.
14
+ - Useful for financial institutions to identify at-risk customers and take retention actions.
15
+
16
+ ## Training Data
17
+
18
+ - **Dataset:** Custom bank churn dataset (`churn.csv`)
19
+ - **Size:** 10,000 samples
20
+ - **Split:** 80% train, 20% test
21
+ - **Preprocessing:** Feature engineering (BalanceSalaryRatio, TenureByAge), categorical encoding, min-max scaling.
22
+
23
+ ## Metrics
24
+
25
+ - **Loss:** Binary cross-entropy
26
+ - **Accuracy:** ~81% on test set
27
+ - **Evaluation:** Confusion matrix, classification report
28
+
29
+ ## Limitations
30
+
31
+ - Model trained on a specific dataset; may not generalize to other banks or regions.
32
+ - Sensitive to feature distribution and preprocessing steps.
33
+ - Does not explain feature importance.