Heart attack prediction models #141
Open
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What are the changes?
Added and organized heart attack prediction work using classical machine learning and deep learning approaches. This includes cleaned data, baseline models, and an extended neural network pipeline.
Who worked on the changes?
Ananth: Developed models, analysis, and evaluation metrics. Did have discussions with Juweria on how we would both proceed.
New Components/Features
Heart_attack_model_Ananth.ipynb: Logistic Regression and Random Forest with EDA
deep_learning_heart_attack.ipynb: PyTorch-based DeepHeartNet model + feature selection
heart_attack_dataset.csv: Final cleaned dataset used across notebooks
Enhanced Components/Features
Added visuals: feature importance, ROC curves, confusion matrices
TensorBoard logging for neural net training
Added summary output cells and hyperparameter tuning logic
Additional Notes
Classical models gave 50–53% F1
Neural net tuning improved results slightly (~61% F1 best so far)
Results suggest feature quality or label imbalance limits performance
TODOs
Documentation for the deep learning model