- 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.pyusingpython main_train.py. - Currently implemented models, (switch between them using the cli option
--modelas one of [logistic,random_forest]):- Logistic Regression
- Random Forest Classifiers
sklearnmatplotlibscipypandasnumpyseaborn



