/scikit-heart-disease-classifier

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

Primary LanguageJupyter Notebook

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