Skip to content
Permalink
main
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

10751062-YL-s1/README.md

Go to file
 
 
Cannot retrieve contributors at this time
18 lines (16 sloc) 1.52 KB
Raw Blame
Open in GitHub Desktop

6006CEM_Machine-learning_Coursework

Stroke is caused when a blood vessel in the brain bursts or the blood supply is blocked, preventing blood and oxygen from reaching the brain. It is the second leading cause of death worldwide and places a significant burden on individuals and national health care systems. Stroke has a wide range of underlying causes. Therefore, the main goal of this project is to apply the techniques of machine learning to existing datasets to make effective stroke predictions based on a wide range of relevant features and to make some recommendations. The dataset selected for this project is from Kaggle: “Stroke Prediction Dataset”The dataset has 12 columns:

  1. id: Unique identifier
  2. gender: Gender of patient
  3. age: The age of the patient
  4. hypertension: Having this disease is marked 1, else 0
  5. heart_disease: Having this disease is marked 1, else 0
  6. ever_married: “Yes” or “No”
  7. work_type: "children", "Govt_jov", "Never_worked", "Private" or "Self-employed"
  8. Residence_type: "Rural" or "Urban"
  9. avg_glucose_level: average glucose level in blood
  10. bmi: body mass index
  11. smoking_status: "formerly smoked", "never smoked", "smokes" or "Unknown"
  12. stroke: 1 if the patient had a stroke or 0 if not

The URL of the dataset: https://www.kaggle.com/datasets/fedesoriano/stroke-prediction-dataset.

The libraries required for this project are: pandas, numpy, matplotlib, warnings, seaborn, sklearn, imblearn, xgboost. These libraries need to be pre-installed to ensure that the program runs!