PyHealth Tasks

Clinical Tasks

Standardized clinical prediction task definitions — from mortality and readmission to drug recommendation, sleep staging, and genomic analysis. Call dataset.set_task() and you're ready to train.

Actively growing — new tasks added regularly. View planned additions →

Browse Tasks

Full API reference →

Output type

Input modality

Loading tasks…

Need a custom task?

Defining a new clinical prediction task in PyHealth takes minutes. Specify your input features and output label schema, and dataset.set_task() handles the rest — including batching, tokenization, and dataloader creation.

Custom Task Tutorial → View Source on GitHub →