What problem does it solve?
PyHealth turns messy clinical data work into a structured, reusable pipeline for training and evaluating healthcare deep-learning models.
Core Features & Use Cases
- Unified clinical pipeline: Build end-to-end workflows using the Dataset β Task β Model β Trainer β Metrics pattern to avoid plumbing mistakes.
- Multi-domain dataset support: Load EHR, physiological signals, and imaging datasets such as MIMIC-III/IV, eICU, OMOP, SleepEDF, SHHS/ISRUC, and ChestXray14.
- Task-appropriate modeling: Train models for mortality, readmission, length-of-stay, drug recommendation, sleep staging, ICD/ATC coding, and EEG event prediction with task-matched
monitor metrics.
- Medical code utilities: Perform ICD/ATC/NDC/RxNorm lookups and cross-mapping (including cross-version mapping like ICD-9 β ICD-10 and NDC β RxNorm) for feature engineering and interpretation.
Quick Start
Use the pyhealth skill to build a mortality prediction run on MIMIC-III by loading the dataset, setting a mortality task, choosing a Transformer model, training with Trainer, and reporting PR-AUC/ROC-AUC metrics.