PyHealth icon indicating copy to clipboard operation
PyHealth copied to clipboard

Add example notebook for Kaggle Chest X-Ray Pneumonia dataset

Open rz-coder opened this issue 1 month ago • 0 comments

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:

  1. Loads and explores the Kaggle dataset
  2. Builds a PyHealth-compatible dataset
  3. Prepares train/val/test datasets for deep learning
  4. Trains a pretrained ResNet18 model using PyHealth’s Trainer
  5. 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:

    1. Run this notebook on Kaggle using the following dataset as input:
      https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia/code
    2. Download the dataset and run this notebook locally on your own machine.

rz-coder avatar Dec 02 '25 23:12 rz-coder