This repository contains a Jupyter Notebook for predicting heart disease using a machine learning model. The dataset used for this project is from Kaggle: Heart Disease Dataset.
Heart disease is a leading cause of death worldwide. Early detection can help in taking preventive measures and treatment to reduce the risk. This project aims to predict the presence of heart disease in a patient based on various medical attributes using machine learning techniques.
The dataset used in this project is publicly available on Kaggle and contains several medical predictors such as age, sex, chest pain type, resting blood pressure, cholesterol level, fasting blood sugar, and more.
- Dataset Link: Heart Disease Dataset
To run this project, you need to have Python and Jupyter Notebook installed on your system. Additionally, install the required Python packages by running:
pip install -r requirements.txt
- Clone the repository:
git clone https://github.com/your-username/heart-disease-prediction.git cd heart-disease-prediction
- Install the required packages:
pip install -r requirements.txt
- Open the Jupyter Notebook:
jupyter notebook heart_disease_prediction.ipynb
- Follow the steps in the notebook to preprocess the data, train the model, and evaluate its performance.
The notebook covers the following steps:
- Data Exploration: Understanding the dataset and visualizing the features.
- Data Preprocessing: Cleaning the data and preparing it for the model.
- Model Training: Training various machine learning models such as Logistic Regression, Decision Tree, Random Forest, and others.
- Model Evaluation: Evaluating the performance of the models using metrics such as accuracy, precision, recall, and F1 score.
The model's performance is evaluated, and the best-performing model is selected based on the evaluation metrics. Details of the results can be found in the notebook.
Contributions are welcome! Please fork the repository and submit a pull request for any enhancements or bug fixes.
This project is licensed under the MIT License. See the LICENSE file for details.