Skip to content

Latest commit

 

History

History
14 lines (10 loc) · 618 Bytes

README.md

File metadata and controls

14 lines (10 loc) · 618 Bytes

ML for Heart Attack Prediction

This repository contains a Jupyter Notebook focusing on exploration/analysis and prediction on the following Kaggle.com dataset: Heart Attack Analysis & Prediction Dataset.

Different methods for data visualization, variables selection, dataset pre-processing and ML prediction model selection are presented in the notebook. Tested models are:

  • Decision Tree
  • Random Forest
  • Support Vector Machines
  • Gradient Boosting
  • XGBoost