This project was developed as my final project during my internship at Aman Holding.
It is a Streamlit web application that predicts whether a person is at risk of stroke based on healthcare data, while also providing interactive visualizations, dataset insights, and model explainability using SHAP values.
-
Uses a healthcare dataset from Kaggle for training and evaluation.
-
Provides interactive visualizations to explore dataset patterns and insights.
-
Integrates a trained Machine Learning model to predict stroke likelihood.
-
Uses SHAP values to explain the model's predictions.
-
Includes a Jupyter Notebook documenting:
- Data preprocessing
- Feature engineering
- Model selection
- Model evaluation
- Stroke Prediction: Enter patient health attributes and receive a prediction.
- Visual Analytics: Explore dataset statistics and visualizations.
- Explainable AI: SHAP values help explain why the model made a prediction.
- End-to-End Pipeline: From raw data → preprocessing → model → deployment.
- Python 3.8+
- NumPy
- Pandas
- Scikit-learn
- SHAP
- Matplotlib
- Seaborn
- Plotly
- Streamlit
- Joblib
- PIL
Clone the repository:
git clone https://github.com/Seif-Elmezaien/Stroke-Detection.git
cd Stroke-Detectionpip install -r requirements.txtRun the Streamlit application:
streamlit run app.pyThe application will open in your browser.
Stroke-Detection/
├── demo/
│ └── stroke_demo.gif
├── Notebook/
├── app.py
├── requirements.txt
└── README.md
This project was developed for educational purposes as part of my internship at Aman Holding. The dataset is publicly available on Kaggle.
The application is not intended for medical diagnosis or to replace professional medical advice.
