Stroke prediction

Dataset from Kaggle

Tested with Python 3.10.2

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

Train the neural network

Edit src/train.py for hyperparameters

python3 src/train.py

Edit src/predict.py for the inputs (look at src/dataset_maps.py for some mappings)

python3 src/predict.py