/flask_ml_app

Flask App with PostgreSQL database & Sklearn machine learning model

Primary LanguagePython

Flask Machine Learning App

Project Description

The objective of this project was to become familiar with postgreSQL databases, and their interactions with a Web App on the Flask framework.

The objective of the machine learning model (Sklearn) is to predict wages based on years of experience. The model used a linear regression, previously trained on a dataset, then dumped in pickle format.

Prediction results are stored in the postgreSQL database.