An open API service indexing awesome lists of open source software.

https://github.com/developerwilliams/customer-churn-prediction

This code trains a basic Random Forest model for churn prediction.
https://github.com/developerwilliams/customer-churn-prediction

Last synced: 4 months ago
JSON representation

This code trains a basic Random Forest model for churn prediction.

Awesome Lists containing this project

README

          

# Customer Churn Prediction

## Overview
This project builds a machine learning model to predict customer churn based on historical data. It includes data preprocessing, model training, and performance evaluation.

## Features
- **Data Preprocessing**: Encodes categorical data and scales features.
- **Modeling**: Trains a Random Forest and Logistic Regression model.
- **Evaluation**: Evaluates models using accuracy, precision, recall, and AUC.

## Installation
1. Install dependencies: `pip install -r requirements.txt`
2. Run `python train.py --data customer_data.csv --output model.pkl`

## Technologies Used
- `scikit-learn` for machine learning
- `pandas` and `numpy` for data handling
- `matplotlib` for visualization