What problem does it solve?
PyHealth accelerates end-to-end healthcare AI development by providing datasets, task definitions, models, training pipelines, evaluation metrics, and interpretability tools for clinical data.
Core Features & Use Cases
- Data access and normalization: supports EHR datasets such as MIMIC-III/IV, eICU, and OMOP, with standardized event-patient-visit structures.
- Task definitions and sample generation: 20+ predefined clinical prediction tasks and easy custom task creation.
- Model zoo: 33+ architectures including RETAIN, Transformer, GNN, SafeDrug, GAMENet for healthcare.
- Training, evaluation, calibration, and interpretability: built-in Trainer, metrics, uncertainty quantification, and explanations tools.
- Deployment readiness: supports calibration, monitoring, and reproducible pipelines for clinical use.
- Documentation and references: comprehensive references for datasets, preprocessing, models, and tasks.
Quick Start
Install the library, load a healthcare dataset (e.g., MIMIC4), set a mortality prediction task, and train a Transformer model to evaluate its performance.