The dataset originates from a reputable medical source and comprises multiple features significant to cardiovascular health, such as age, gender, blood pressure, cholesterol levels, and lifestyle choices. The data is preprocessed and transformed to ensure optimal model performance.
The project follows a structured approach:
- Data Preprocessing: Cleaning, normalizing, and transforming data to prepare for analysis.
- Exploratory Analysis: Uncovering trends and patterns through statistical methods and visualization.
- Model Building: Implementing machine learning algorithms including Random Forest, AdaBoost, and XGBoost.
- Evaluation: Assessing models using precision, recall, F1-score, and ROC-AUC metrics.
The repository includes a Jupyter notebook with detailed code annotations. Key configurations:
- Python Version: 3.9
- Key Libraries: scikit-learn, pandas, numpy, matplotlib, seaborn, xgboost
The notebook begins with data loading, followed by an initial examination of the dataset's characteristics.
Data quality assessment and transformations are meticulously performed to enhance model reliability.
Strategic feature engineering is applied, including BMI calculation and interaction terms to improve predictive power.
An in-depth EDA is conducted, offering visual and statistical insights into the dataset, enhancing understanding of the underlying health factors.
The project demonstrates the application of advanced machine learning models:
- Random Forest
- AdaBoost
- XGBoost
Each model undergoes thorough tuning and evaluation, ensuring robustness and accuracy.
SHAP (SHapley Additive exPlanations) values are calculated for an in-depth understanding of feature importance.