avestura / 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

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