What problem does it solve?
This Skill streamlines the creation and management of clinical/healthcare deep-learning pipelines, making it easier to build, train, and evaluate models for healthcare applications.
Core Features & Use Cases
- Unified Pipeline: Provides a modular 5-stage pipeline (Dataset → Task → Model → Trainer → Metrics) for building deep-learning pipelines.
- Dataset Support: Offers support for a variety of datasets including MIMIC, eICU, OMOP, SleepEDF, ChestXray14, and EHRShot.
- Modeling: Supports various models such as Transformer, RETAIN, GAMENet, SafeDrug, MICRON, StageNet, AdaCare, CNN/RNN/MLP.
- Use Case: For a hospital looking to predict patient mortality based on EHR data, PyHealth can be used to load the data, define the task, instantiate the model, train the model, and compute clinical metrics.
Quick Start
Use the pyhealth skill to train a model for mortality prediction on MIMIC-III data.