In this project I have trained a ML model to predict lifetime value of customers and divided them into 3 categories :
Gold,
Silver
and Bronze.
For better accuracy I have tune hyper parameters of multiple classification models and choose the best among them. Elbow method was used for K-means clustering. Finally tuned version of xgboost model is used with 88.9% accuracy.
Python
- Google Colab
- Pandas
- Numpy
- Scikit-learn
- XGBoost
- Pickle
- RandomizedSearchCV
- Random Forest
- K means
Online Retail Dataset from UCI Machine Repository was used. Link