- Published on
- An Avestura Project:
Heart Disease Classifier
- Authors
- ← Back to projects vitrine
- Name
- Aryan Ebrahimpour
- GitHub
- @avestura
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
-
Documentation: https://archive.ics.uci.edu/ml/datasets/Heart+Disease
-
Download: Download