What problem does it solve?
PyHealth provides an end-to-end framework for developing, testing, and deploying healthcare AI solutions. It unifies datasets, tasks, models, and evaluation workflows to streamline clinical machine learning—from EHRs to physiological signals, medical images, and clinical notes.
Core Features & Use Cases
- Data loading, task definition, model selection, training, evaluation, and deployment for healthcare AI.
- Predefined clinical prediction tasks (mortality, readmission, length of stay, drug recommendation) and extensible task creation.
- Medical coding translation, preprocessing, and model interpretability to support research, validation, and deployment in clinical settings.
Quick Start
Set up a mortality-prediction workflow by loading MIMIC4, applying the mortality_prediction_mimic4_fn task, and training a Transformer model.