What problem does it solve?
PyHealth streamlines the process of building, training, and deploying machine learning models on clinical data by providing standardized datasets, tasks, models, and evaluation workflows tailored to healthcare scenarios.
Core Features & Use Cases
- Healthcare AI toolkit for EHR and clinical ML: supports mortality, readmission, drug recommendation, medical coding, and physiologic-signal modeling using common clinical prediction pipelines.
- Modular data-to-model pipeline: handles dataset loading, task definition (input/output schemas), model selection (baselines to healthcare-specific), training, evaluation, calibration, and interpretability.
- Reference-driven guidance: includes organized documentation for datasets, medical code translation, tasks, models, preprocessing, and training/evaluation best practices.
Quick Start
Use the pyhealth skill to build an ICU mortality prediction pipeline on MIMIC-IV by loading the dataset, selecting the appropriate mortality task, choosing a model such as Transformer or RETAIN, training with the Trainer class, and evaluating with AUROC/AUPRC plus calibration and interpretability checks.