Heart Disease Prediction: Machine Learning with UCI DatasetHeart Disease Prediction: Machine Learning with UCI Dataset
The network for creativity
Join 1.25M professional creatives like you
Connect with clients, get discovered, and run your business 100% commission-free
Creatives on Contra have earned over $150M and we are just getting started
Back to feedPost
Tobechukwu's avatar
Tobechukwu Samuel
• 18h
❤️ Heart Disease Prediction using Machine Learning
📌 Problem Statement
Can we predict whether a patient has heart disease using clinical features?
Early detection of heart disease can save lives, making this a critical real-world machine learning problem.
📊 Dataset
Source: UCI Heart Disease Dataset
Includes medical attributes such as:
Age
Cholesterol levels
Chest pain type
Maximum heart rate
Blood pressure
⚙️ Project Workflow
Data Exploration (EDA)
Data Preprocessing
Model Training
Model Evaluation
Hyperparameter Tuning
🤖 Models Used
Logistic Regression
K-Nearest Neighbors (KNN)
Random Forest Classifier
📈 Model Performance
MetricScore (%)Accuracy73.47%Precision83.00%Recall74.95%F1 Score73.36%
🔍 Key Insights
Logistic Regression performed best after tuning
Cross-validation revealed a drop in performance → highlighting generalization challenges
High precision suggests the model is effective at identifying positive cases
Real-world medical datasets are complex and rarely achieve extremely high accuracy
🧠 Lessons Learned
Accuracy alone is not enough — precision and recall matter more in healthcare
Overfitting can give misleading results without proper validation
Simpler models (like Logistic Regression) can outperform complex ones
🛠 Tools & Libraries
Python
Pandas
NumPy
Scikit-learn
Matplotlib
Seaborn
🚀 Future Improvements
Feature engineering
Try advanced models (XGBoost, LightGBM)
Improve recall (important for medical predictions)
Deploy model as a web app
📎 Project Notebook
Check the full implementation in the Jupyter Notebook.
🙌 Acknowledgements
UCI Machine Learning Repository
🔬 Feature Importance (Logistic Regression)
The model coefficients reveal which features most influence heart disease prediction.
🔺 Features Increasing Risk
ca (number of major vessels) → Strongest predictor
oldpeak (ST depression) → Indicates heart stress during exercise
exang (exercise-induced angina) → Associated with higher risk
restecg abnormalities → Signals irregular heart activity
🔻 Features Decreasing Risk
Certain chest pain types (non-anginal)
Some dataset-specific patterns
Gender-related differences (model-specific behavior)
💡 Insight
The model heavily relies on cardiovascular stress indicators and blood flow patterns, which aligns with real-world medical understanding of heart disease.
Machine Learning EngineeringData Analysisscikit-learnpandasPythonAI Engineer
Post image
Back to feed
The network for creativity
Join 1.25M professional creatives like you
Connect with clients, get discovered, and run your business 100% commission-free
Creatives on Contra have earned over $150M and we are just getting started