This Project uses the UCi Myocardial Infarction Complications dataset to train a first machine learning model. https://archive.ics.uci.edu/dataset/579/myocardial+infarction+complications
-
heart_attack_prediction.pyMain script that loads data, cleans it, trains models, and evaluates them -
src/load_data.pyLoads the dataset from UCI using 'ucimlrepo' -
src/preprocess.pySelects the target, drops bad columns, and splits data -
src/train_model.pyBuilds preprocessing and achine learning pipelines -
src/evaluate.pyPrints accuracy, precision, recall, F1, ROC-AUC, and confusion matrix
pip install -r requirements.txt
python heart_attack_prediction.py