This project predicts customer churn using an Artificial Neural Network (ANN) and provides an interactive web interface built with Streamlit. Users can input customer details and instantly get a churn prediction.
🔗 Live App: 👉 https://churnmodellingusingann-jxyqbzwvxtc9ndrnwtqbo9.streamlit.app/
Customer churn prediction helps businesses identify customers who are likely to leave. In this project:
- An ANN model is trained on banking customer data
- The trained model predicts whether a customer will churn or not
- A Streamlit web app is used for real-time predictions
- Preprocessing objects and trained models are saved and reused
- ANN-based churn prediction
- Interactive Streamlit UI
- Real-time predictions
- Data preprocessing (encoding & scaling)
- Saved model and encoders (
.pkl,.h5,.keras) - User-friendly web deployment
- Python
- NumPy
- Pandas
- Scikit-learn
- TensorFlow / Keras
- Streamlit
- Jupyter Notebook
churn_modelling_usingANN/
│
├── BankChurners.csv # Dataset
├── Churn_Modelling.csv # Dataset
│
├── app.py # Streamlit application
├── experiments.ipynb # Model training & experiments
├── predictions.ipynb # Prediction testing
│
├── model.h5 # Trained ANN model
├── model.keras # Keras model format
│
├── scaler.pkl # StandardScaler
├── ohe_geo.pkl # One-Hot Encoder (Geography)
├── label_encoder_gender.pkl # Label Encoder (Gender)
│
├── requirements.txt # Dependencies
├── req.txt # Alternate requirements file
└── .devcontainer/ # Dev container setup
Clone the repository:
https://github.com/JahnaviSingh2005/churn_modelling_usingANN.gitMove into the project directory:
cd churn_modelling_usingANNInstall dependencies:
pip install -r requirements.txtstreamlit run app.pyThe model uses the following inputs:
- Credit Score
- Geography
- Gender
- Age
- Tenure
- Balance
- Number of Products
- Has Credit Card
- Is Active Member
- Estimated Salary
- 0 → Customer will NOT churn
- 1 → Customer WILL churn
The result is displayed instantly on the Streamlit interface.
The application is deployed using Streamlit Cloud.
🔗 Live App Link: https://churnmodellingusingann-jxyqbzwvxtc9ndrnwtqbo9.streamlit.app/
- Model performance comparison with ML algorithms
- Hyperparameter tuning
- Explainable AI (SHAP / LIME)
- Improved UI/UX
- Dataset expansion
Jahnavi Singh B.Tech Student | Machine Learning & AI Enthusiast
Just say the word 🚀