task template

pyhealth/tasks/__init__.py

@@ -1,3 +1,4 @@
+from .task_template import TaskTemplate
 from .drug_recommendation import (
  drug_recommendation_eicu_fn,
  drug_recommendation_mimic3_fn,
@@ -29,3 +30,5 @@
  sleep_staging_isruc_fn,
  sleep_staging_shhs_fn,
 )
+from .covid19_cxr_classification import COVID19CXRClassification
+from .medical_transcriptions_classification import MedicalTranscriptionsClassification

pyhealth/tasks/task_template.py

@@ -0,0 +1,14 @@
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from typing import Dict, List
+
+
+@dataclass(frozen=True)
+class TaskTemplate(ABC):
+ task_name: str
+ input_schema: Dict[str, str]
+ output_schema: Dict[str, str]
+
+ @abstractmethod
+ def __call__(self, patient) -> List[Dict]:
+ raise NotImplementedError