/Stroke-EDA-Predictor

Investigation of the factors that lead to strokes.

Primary LanguageJupyter Notebook

Stroke-EDA-Predictor

Introduction

This is a stroke prediction script written for our Machine Learning final project. The dataset was obtained from Kaggle.

Getting Started

Prerequisites

The following dependencies are required:

  • numpy
  • pandas
  • math
  • seaborn
  • matplotlib
    • pyplot
  • sklearn
    • model_selection
    • preprocessing
    • utils
    • metrics
    • svm
  • imblearn
    • over_sampling

If using Anaconda with Python 3.8+, everything above is included except imblearn. To add this, run:

conda install -c conda-forge imbalanced-learn

Else, individually install these libraries using pip install.

Running The Notebooks

Given the .idea folder, the project can be set up in PyCharm Professional. However, using other IDEs such as Jupyter Notebook or JupyterLab will also suffice.

Contributions

Author File
Chathumini Abeyratna Logistic Regression
Saadiya Allahbaksh SVM
Kevin Karnani RandomForest