What problem does it solve?
PyHealth provides a unified toolkit to streamline the development, testing, and deployment of healthcare AI models using clinical data and standardized datasets.
Core Features & Use Cases
- Data loading from major healthcare datasets (MIMIC, eICU, OMOP) with a consistent interface.
- Task definition and dataset transformation to task-specific samples.
- Model selection and training pipeline with evaluation and deployment readiness.
- Use cases include mortality/readmission/LOS prediction, medical coding translation, EEG/ECG processing, imaging and clinical text analysis.
- Real-world workflow: load data, set a task, split data, train a model (e.g., Transformer), evaluate, and interpret.
Quick Start
Load the MIMIC-IV dataset, set mortality_prediction_mimic4_fn as the task, train a Transformer model, and evaluate on a held-out test set.