avestura / scikit-heart-disease-classifier Public

Notifications You must be signed in to change notification settings
Fork 0
Star 3

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

3 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
notebook.ipynb		notebook.ipynb
processed.cleveland.data		processed.cleveland.data
readme.md		readme.md

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

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 rbf-kernel pca pruning confusion-matrix roc decision-tree decision-tree-classifier svm-classifier linear-kernel one-hot-encoding grid-search-cross-validation cost-complexity-pruning poly-kernel sigmoid-kernel

Report repository

Languages

Jupyter Notebook 100.0%