- This is a quickly whipped up, well structured project using a Customer Personality dataset.
- I have conducted a quite in-depth feature extraction (as outlined in
feature_extraction.ipynb
). - Models were tinkered with in
train.ipynb
. - Execute
main_train.py
usingpython main_train.py
. - Currently implemented models, (switch between them using the cli option
--model
as one of [logistic
,random_forest
]):- Logistic Regression
- Random Forest Classifiers
sklearn
matplotlib
scipy
pandas
numpy
seaborn