PyHealth
PyHealth copied to clipboard
Add example notebook for Kaggle Chest X-Ray Pneumonia dataset
Note
Who you are:
rzhou19
What type of contribution you are making:
New example / use case of PyHealth.
A high level description of what you’ve done:
This notebook demonstrates a complete end-to-end workflow for training a deep learning model on the Chest X-Ray Pneumonia dataset using the PyHealth framework inside Kaggle
At a high level, the notebook:
- Loads and explores the Kaggle dataset
- Builds a PyHealth-compatible dataset
- Prepares train/val/test datasets for deep learning
- Trains a pretrained ResNet18 model using PyHealth’s Trainer
- Evaluates the model performance
A quick rundown of what files to look at to test the implementation
-
To preview the implementation, you can download thsi commit and view locally for better format.
-
To test the implementation, you can either:
-
Run this notebook on Kaggle using the following dataset as input:
https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia/code - Download the dataset and run this notebook locally on your own machine.
-
Run this notebook on Kaggle using the following dataset as input: