Skip to content

πŸ’š A heart disease classifier using 4 SVM kernels and decision trees, with PCA, ROC, pruning, grid search cv, confusion matrix, and more

Notifications You must be signed in to change notification settings

avestura/scikit-heart-disease-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

1 Commit
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Heart Disease Classifier

A heart disease classifier using Scikit-Learn.

Methods, approaches and visualizations

  • SVM Classification (linear, poly, rbf and sigmoid kernels)
  • Decision Tree Classification
  • One-Hot encoding
  • Data Scaling
  • Confusion Matrix charts
  • Optimizing parameters via grid search cross validiation
  • PCA
  • ROC Charts
  • Decision surface/region chart
  • Cost complexity pruning
  • Visual Decision Tree after and before pruning

Data Set