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

svm svm-classifier decision-tree decision-tree-classifier pca roc pruning grid-search-cross-validation confusion-matrix one-hot-encoding cost-complexity-pruning linear-kernel poly-kernel rbf-kernel sigmoid-kernel

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

About

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

svm svm-classifier decision-tree decision-tree-classifier pca roc pruning grid-search-cross-validation confusion-matrix one-hot-encoding cost-complexity-pruning linear-kernel poly-kernel rbf-kernel sigmoid-kernel

Languages

Language:Jupyter Notebook 100.0%