What problem does it solve?
PyHealth helps you turn messy healthcare data into reusable deep-learning pipelines without hand-wiring the entire workflow. It reduces the friction of loading EHR, signal, or imaging datasets, matching them to the right prediction task, and training a clinically meaningful model with the right evaluation metrics.
Core Features & Use Cases
- Dataset loading: Work with clinical sources such as MIMIC-III, MIMIC-IV, eICU, OMOP, SleepEDF, ChestXray14, and EHRShot.
- Task selection: Set up common healthcare objectives like mortality prediction, readmission, length of stay, drug recommendation, sleep staging, ICD coding, and EEG event detection.
- Model and training workflow: Pair datasets with PyHealth models, train through the unified trainer, and monitor the correct metric for binary, multiclass, or multilabel outputs.
- Medical code utilities: Look up ICD, ATC, NDC, RxNorm, and CCS codes or cross-map between coding systems for feature engineering and interpretation.
- Use case: A researcher can load MIMIC-III, define a mortality task, train a Transformer or RETAIN model, and evaluate the result with clinically appropriate metrics in a single pipeline.
Quick Start
Ask me to build, troubleshoot, or explain a PyHealth pipeline for your healthcare dataset, task, model, and metric, and I will produce the correct end-to-end setup.