This project focuses on predicting customer churn using an Artificial Neural Network (ANN). The model was developed using TensorFlow/Keras and deployed as a user-friendly web application via Streamlit. The solution leverages robust preprocessing pipelines to handle data efficiently, enabling businesses to identify and retain at-risk customers.
- Customer Churn Prediction: Predict whether a customer is likely to churn based on input features.
- Interactive Web App: User-friendly interface to input data and get instant predictions.
- End-to-End Preprocessing: Automated handling of missing values, normalization, and encoding of categorical variables.
- Model Deployment: Hosted on Streamlit for easy accessibility.
Access the application here: ANN Churn Prediction Web App
- TensorFlow/Keras: For building and training the ANN model.
- Streamlit: For creating an interactive and accessible web application.
- Pandas & NumPy: For data preprocessing and feature engineering.
- Scikit-learn: For splitting data and preprocessing utilities.
- Data Preprocessing:
- Handled missing values and performed feature scaling.
- Encoded categorical variables using one-hot encoding.
- Model Development:
- Designed a feedforward neural network with multiple dense layers.
- Optimized the model using backpropagation and various hyperparameters.
- Evaluated performance using metrics like accuracy and recall.
- Deployment:
- Deployed the trained model as a Streamlit app for real-time predictions.
To run this project locally:
- Clone the repository:
git clone https://github.com/krishnaik06/ANN-CLassification-Churn.git
- Install the required dependencies:
pip install -r requirements.txt
- Run the Streamlit app:
streamlit run app.py
- Open the hosted app or run it locally.
- Enter customer details in the input fields.
- Click on the "Predict" button to get the churn prediction.
- Incorporate additional features to improve prediction accuracy.
- Add a feature to upload bulk data for batch predictions.
- Enhance the user interface for better user experience.
- This project is licensed under the MIT License.