This project demonstrates a Customer Churn Prediction Model using XGBoost and the JFrogML platform. It showcases multiple deployment strategies and training approaches for production-ready churn prediction systems.
Before starting, ensure you have:
- Python 3.9-3.11 installed
- JFrog account (Get started for free)
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
│ 💻 Local ML │ -> │ 🏗️ Build │ -> │ 🚀 Deploy │
│ App Code │ │ Process │ │ ML Serving │
│ (or GitHub) │ │(w/ Training Job)│ │ API Endpoint │
└─────────────────┘ └─────────────────┘ └─────────────────┘
Local IDE FrogML CLI FrogML CLI
Complete workflow: 🚀 Remote Training & Deployment Guide
Best for: Standardized, replicable, production-ready workflows with integrated training and serving
churn_model_new/
├── main/ # Main directory containing core code
│ ├── __init__.py # Python package initialization
│ ├── model.py # FrogMLModel with churn prediction logic
│ ├── data.csv # Training dataset
│ └── conda.yml # Environment dependencies
├── tests/ # Integration tests
│ └── it/
│ └── test_churn.py # Integration test for churn model
├── test_model_code_locally.py # Script to test the model locally
├── test_live_endpoint.py # Script to test live deployment endpoint
├── test_batch_endpoint.py # Script to test batch inference endpoint
├── README.md # This documentation
└── remote-training-and-deployment.md # Deployment guide
- Choose your deployment path from the guides above
- Follow the step-by-step instructions in your chosen guide
- Customize the model for your specific churn prediction needs
- Scale up with larger datasets and more complex feature engineering
Ready to get started? Pick a guide above and begin your churn prediction journey! 🚀