What problem does it solve?
PyHealth provides a comprehensive healthcare AI toolkit to build, train, and deploy clinical prediction models on real-world EHR, imaging, and text data.
Core Features & Use Cases
- Datasets and task orchestration: access MIMIC-III/IV, eICU, OMOP, Sleep EEG, and clinical notes; set up sample datasets with task functions to standardize inputs and outputs.
- End-to-end workflow: preprocessing, model selection (RETAIN, Transformer, GNN), training, evaluation, calibration, and interpretability, enabling rapid prototyping and production-ready deployment.
- Task-driven collaboration: predefined clinical prediction tasks (mortality, readmission, LOS, drug recommendation) and support for custom task creation to align with clinical questions.
Quick Start
Set up a dataset (e.g., MIMIC-IV), apply the mortality_prediction_mimic4_fn task, train a Transformer model, and evaluate on a held-out test set.