/Machine-Learning-with-flask

Performing basic machine learning algorithm using Flask framework.

Primary LanguageJupyter Notebook

Machine learning model on Flask

This is a basic project which runs a machine learning model on the flask. Here, I have implemented an SVM classifier to predict the hear attack condition in the patient. The dataset is available on the archive folder. For more details about the dataset check out Kaggle.

Model used for classification

SVM

You can check that in the heart_attack.ipynb Got accuracy of 82% with F1 score of 81. Model weights are stored in finalized_model.sav

You can read my blog on SVM here.

Flask Integration

Created an front end page with html. The flask is used to run these pages. model_prediction.py is where all the post requests are processed and prediction is made.

Screenshots

Home Results

Run app.py