This project aims to predict customer churn using machine learning algorithms. The project includes data preprocessing, feature engineering, and model evaluation.
This project aims to predict customer churn using machine learning algorithms. The goal is to identify customers who are likely to stop using a service or product, allowing businesses to take proactive measures to retain them. The project includes the following key components:
pandas
, numpy
, scikit-learn
, matplotlib
, seaborn
, PyYAML
git clone https://github.com/your-username/Churn-Prediction.git
pip install -r requirements.txt
python main.py
-c
or --classifier
: specify the classifier to use (default: AdaBoost
)-t
or --test-size
: specify the test size for train-test split (default: 0.2
)-p
or --preprocessing-method
: specify the preprocessing method to use (default: standardization
)python main.py
: run the project with default settingspython main.py -c KNN -t 0.5 -p robust-scaling
: run the project with KNN classifier, 50% test size, and robust scaling preprocessingmain.py
: main entry point for the projectconfig.py
: configuration file for the projectdataset.py
: data loading and preprocessing moduleclassification.py
: classification moduleplot_confusion_matrix.py
: confusion matrix plotting modulerequirements.txt
: required libraries for the project