/stroke-predictor

Neural network to predict strokes.

Primary LanguagePython

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

Use the neural network

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

python3 src/predict.py