Skip to content

LionelRoxas/customer-churn-prediction

Repository files navigation

Customer Churn Prediction Project 🚀

Welcome to my latest project on Customer Churn Prediction! This project, deployed using Streamlit, leverages both classical machine learning and generative AI to help banks identify customers at risk of churning. With models like Random Forest, XGBoost, Support Vector Machines, and more, this project predicts churn probability and offers actionable insights for customer retention.

🔍 Key Highlights

  • Multiple Machine Learning Models: Predictions are based on an ensemble of models, including Random Forest, K-nearest Neighbors, and Gradient Boosting, to determine the likelihood of customer churn.
  • Generative AI Integration: Utilizing Llama 3.1 (via Groq), the app generates interpretable explanations for predictions, helping banks understand the reasons behind potential churn.
  • Personalized Customer Outreach: The app also employs Llama 3.1 to generate personalized email content that banks can send to at-risk customers to encourage retention.

🛠️ Live Demo

Imagine being a bank manager with access to a dashboard displaying customer data (such as credit score, balance, etc.). The app predicts churn probability, such as a customer having a 19.1% chance of churning due to limited engagement. It then generates personalized email messages, which banks can use to connect with and retain these at-risk customers.

🏗️ Project Architecture

  1. Data Collection: Downloaded a bank customer dataset from Kaggle.
  2. Data Processing: Cleaned and normalized the data, performing feature engineering to improve model accuracy.
  3. Model Training: Trained multiple models, fine-tuning hyperparameters for optimal results.
  4. Web App Development: Created an interactive web app that displays predictions and generates AI-driven explanations and emails.

🔗 Check It Out

The project is live on Streamlit and hosted via Hex for public access. This transparency aims to inspire fellow students and promote innovation in AI-driven solutions. A special thank you to Faizan Ahmed for his guiding tutorial on Headstarter.

Try the app, explore the code, and see AI in action!