iae-bora/ml-api

API para o modelo de machine learning desenvolvido para o projeto IAe, Bora?

About   |    Technologies   |    Machine learning models   |    Solution architecture   |    Running locally   |    Running with Docker

📖 About

API developed in Python which predicts the categories of places that an user should visit based on his answers. This API runs K-Means and Decision Tree Classifier as machine learning models to predict the output.

⚒️ Technologies

• Flask
• Scikit-learn

👨‍💻 Machine learning models

• K-Means
• Decision Tree Classifier

🏙️ Solution architecture

This repository is represented by the number 3 in the architecture, if you want to view the other repositories of the solution, click on one of the items below:

• (1) Web Crawler
• (4) Back-End
• (5) Front-End
• (6) Mobile

💻 Running locally

1. Install the dependencies

$ pip install -r requirements.txt

2. Run the application

$ python wsgi.py

3. The application will start running in localhost:8080.

🐳 Running with Docker

1. Run the following commands on terminal

$ docker build -t <CONTAINER-NAME>:latest .
$ docker run -p 8080:8080 <CONTAINER-NAME>:latest

2. The application will start running in localhost:8080.