https://github.com/vietanhdev/udacity-customer-churn
Course project Predict Customer Churn of ML DevOps Engineer Nanodegree Udacity.
https://github.com/vietanhdev/udacity-customer-churn
Last synced: about 1 year ago
JSON representation
Course project Predict Customer Churn of ML DevOps Engineer Nanodegree Udacity.
- Host: GitHub
- URL: https://github.com/vietanhdev/udacity-customer-churn
- Owner: vietanhdev
- Created: 2022-01-24T08:51:45.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2022-01-24T09:16:09.000Z (over 4 years ago)
- Last Synced: 2025-06-02T16:20:55.811Z (about 1 year ago)
- Language: Python
- Homepage:
- Size: 351 KB
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Predict Customer Churn
Course project **Predict Customer Churn** of ML DevOps Engineer Nanodegree Udacity.
Code in `churn_library.py` completes the process for solving the data science process including:
- EDA
- Feature Engineering (including encoding of categorical variables)
- Model Training
- Prediction
- Model Evaluation
## Environment Setup
- Requirement: Python 3.x
```
pip install requirements.txt
```
## Training & Evaluation
Example running command:
```
python churn_library.py --data="data/bank_data.csv" \
--log_path="churn_library.log" \
--eda_output="eda" \
--model_output="models" \
--evaluation_output="results" \
--feature_importance_output="feature_importance.png" \
--test_size=0.3
```
Adjust parameters for grid search (DEFAULT_PARAMS_GRID) in `constants.py`.
## Run tests
```
python3 tests.py
```
Read `logs/test_log.log` for test log.