pyhealth.tasks.ChestXray14MultilabelClassification

class pyhealth.tasks.ChestXray14MultilabelClassification(code_mapping=None)

Bases: BaseTask

A PyHealth task class for multilabel classification of all fourteen diseases in the ChestXray14 dataset.

task_name

The name of the task.

Type:

str

input_schema

The schema for the task input.

Type:

Dict[str, str]

output_schema

The schema for the task output.

Type:

Dict[str, str]

Examples

>>> from pyhealth.datasets import ChestXray14Dataset
>>> from pyhealth.tasks import ChestXray14MultilabelClassification
>>> dataset = ChestXray14Dataset(root="/path/to/chestxray14")
>>> task = ChestXray14MultilabelClassification()
>>> samples = dataset.set_task(task)

task_name: str = 'ChestXray14MultilabelClassification'

input_schema: Dict[str, str] = {'image': 'image'}

output_schema: Dict[str, str] = {'labels': 'multilabel'}

pre_filter(df)

Return type:

LazyFrame