Skip to content

matheusbbritto/Customer-Churn-Prediction-Benchmarking-ML-Deep-Learning-Models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

5 Commits
Β 
Β 
Β 
Β 

Repository files navigation

πŸ” Customer Churn Prediction

Benchmarking of Machine Learning and Deep Learning models for customer churn prediction in the telecom industry, with automatic hyperparameter optimization via Optuna.

πŸ“Œ About the Project
This project investigates the predictive power of five distinct approaches to identify customers at risk of churning (service cancellation), using a real-world telecom dataset. The full pipeline covers exploratory data analysis, preprocessing, feature engineering, model training, hyperparameter optimization, and comparative evaluation.

πŸ§ͺ Models Evaluated
MLP
Random Forest
XGBoostGradient
TabTabular
TabPFN2

πŸ—‚οΈ Dataset

Source: Telco Customer Churn
Target: Churn β€” Yes / No

βš™οΈ Pipeline

  1. DATA LOADING

  2. PREPROCESSING
    β”œβ”€β”€ Convert TotalCharges to numeric (coerce errors)
    β”œβ”€β”€ Encode target variable: Churn β†’ binary (0/1)
    β”œβ”€β”€ One-hot encoding of categorical features
    β”œβ”€β”€ Standardization with StandardScaler
    └── Train / Validation / Test split

  3. EXPLORATORY DATA ANALYSIS └── Distribution plots for tenure, MonthlyCharges, TotalCharges

  4. FEATURE ENGINEERING β€” Bucketing β”œβ”€β”€ MonthlyCharges β†’ <22 | 22–68 | 68–105 | 105+
    β”œβ”€β”€ tenure β†’ 0–3 | 4–69 | 70+
    └── TotalCharges β†’ <200 | 200–2000 | 2000+

  5. CLASS BALANCING
    └── RandomOverSampler

  6. MODEL TRAINING + HYPERPARAMETER OPTIMIZATION (Optuna)

  7. EVALUATION β”œβ”€β”€ Metrics: Accuracy, Balanced Accuracy, Precision, Recall, F1, AUC-ROC, KS Statistic
    β”œβ”€β”€ Confusion Matrix per model
    └── ROC Curve comparison

πŸ“Š Results

image

About

Benchmarking of ML and Deep Learning models (MLP, Random Forest, XGBoost, STab, TabPFN2) for customer churn prediction in the telecom industry.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors