What problem does it solve?
PyHealth provides a comprehensive, scalable toolkit to develop, test, and deploy machine learning models for healthcare data, reducing the time to translate clinical data into actionable AI solutions.
Core Features & Use Cases
- Comprehensive data handling for Electronic Health Records (EHR), clinical tasks (mortality, readmission, length of stay), medical coding (ICD/NDC/ATC), and multi-modal data (signals, imaging, text).
- End-to-end pipelines from data loading to model training, evaluation, and interpretability, with modular components for datasets, tasks, models, and trainer.
- Real-world use cases include mortality risk prediction, hospital readmission forecasting, drug recommendation with safety constraints, and clinical code translation to harmonize multi-site studies.
Quick Start
Install PyHealth and load a healthcare dataset, define a mortality task, and train a baseline model to start evaluating clinical predictions.