The goal of this project is to build a data-driven model that finds the customer groups that lead to good ROIs (Return on Investment).
Data was prepared for further models' usage using pandas and numpy. Sklearn was used for build models and tuning with small support of scipy. API was made using flask framework.
Firstly install needed packages using pip install -r requirements.txt
. Run the app by executing the cell in app.ipynb
. Then, go to http://0.0.0.0:8080/
. You can do sample prediction using sample_data.csv
file, located in the data
folder.