Customer Churn Prediction ML Pipeline

🤖 ML Project

Customer Churn Prediction ML Pipeline

Scikit-learn pipeline predicting member churn with 89% accuracy.

Project Overview

Client

Internal Project

Industry

Machine Learning / Healthcare

Timeline

2 months (2024)

My Role

ML Engineer

Developed a machine learning pipeline to predict member churn using historical activity data, enabling proactive retention strategies.

The Challenge

The healthcare platform was losing members without early warning:

✗

No visibility into churn risk factors

✗

Reactive approach to member retention

✗

Manual analysis of member behavior

✗

Inconsistent intervention strategies

✗

High false-positive rates in existing rules

They needed a data-driven approach to predict and prevent churn.

My Solution

I built an end-to-end ML pipeline:

1

Feature Engineering

Extracted 50+ features from activity logs, plan usage, and engagement data.

2

Model Training

Tested multiple algorithms (Random Forest, XGBoost, Logistic Regression).

3

API Deployment

FastAPI endpoint for real-time predictions.

4

Monitoring

Model performance tracking and drift detection.

Key Features

🎯

89% Accuracy

High-precision predictions with low false positives.

âš¡

Real-Time

Sub-100ms prediction latency.

📊

Explainable

SHAP values for feature importance.

🔄

Auto-Retrain

Scheduled model updates with new data.

Tech Stack

ML Framework

Scikit-learnXGBoostPandas

API

FastAPIPydantic

Data

PostgreSQLApache Airflow

MLOps

MLflowDockerAWS SageMaker

Screenshots

Model

Model Performance

Features

Feature Importance

Dashboard

Prediction Dashboard

Results & Impact

89%

Accuracy

Achieved

50+

Features

Engineered

<100ms

Latency

Per Prediction

30%

Churn

Reduction

Key Achievements

✓

Achieved 89% prediction accuracy

✓

Reduced churn by 30% through early intervention

✓

Sub-100ms prediction latency

✓

Automated weekly model retraining

Interested in Something Similar?

I help businesses build robust backend systems, membership platforms, and automation tools.

More Projects