For a better understanding of the project, please check the Google Colab file π uploaded in this repository. It contains detailed explanations and execution steps to help you grasp the workflow more effectively.
ChurnPredictANN is an Artificial Neural Network (ANN) model designed to predict customer churn in a banking dataset. It analyzes customer demographics, account details, and transaction history to determine whether a customer is likely to leave the bank.
The model is trained on the Churn_Modelling.csv dataset, which includes:
- Features: Customer ID, Credit Score, Age, Geography, Gender, Balance, Tenure, etc.
- Target Variable:
Exited(1 if the customer left, 0 otherwise).
The neural network consists of:
- Input Layer: 12 input features (after preprocessing).
- Two Hidden Layers:
- 6 neurons each, with ReLU activation.
- Output Layer:
- 1 neuron, with Sigmoid activation (binary classification).
- Dropped unnecessary columns (
RowNumber,CustomerId,Surname). - Encoded categorical features:
Genderβ Label Encoding.Geographyβ One-Hot Encoding.
- Feature Scaling: Standardized numerical values using
StandardScaler. - Train-Test Split: 80% training, 20% testing.
- Loss Function:
binary_crossentropy(for binary classification). - Optimizer:
adam(adaptive learning rate). - Metrics:
accuracy. - Batch Size: 64
- Epochs: 130
- Confusion Matrix: Measures true positives, true negatives, false positives, and false negatives.
- Accuracy Score: Evaluates model performance on the test set.
git clone https://github.com/your-username/ChurnPredictANN.git
cd ChurnPredictANN