sunlabuiuc · caiqile · Apr 17, 2026 · Apr 18, 2026 · Apr 18, 2026 · Apr 19, 2026
diff --git a/docs/api/datasets.rst b/docs/api/datasets.rst
@@ -229,6 +229,7 @@ Available Datasets
     datasets/pyhealth.datasets.eICUDataset
     datasets/pyhealth.datasets.ISRUCDataset
     datasets/pyhealth.datasets.MIMICExtractDataset
+    datasets/pyhealth.datasets.MIMIC4NoteExtDIBHCDataset
     datasets/pyhealth.datasets.OMOPDataset
     datasets/pyhealth.datasets.DREAMTDataset
     datasets/pyhealth.datasets.SHHSDataset

diff --git a/docs/api/datasets/pyhealth.datasets.MIMIC4NoteExtDIBHCDataset.rst b/docs/api/datasets/pyhealth.datasets.MIMIC4NoteExtDIBHCDataset.rst
@@ -0,0 +1,14 @@
+pyhealth.datasets.MIMIC4NoteExtDIBHCDataset
+===================================
+
+The Mimic4Note Extension DischargeInstructions BriefHospitalCourse dataset is used in this paper: https://arxiv.org/pdf/2402.15422 to create AI generated medical note summaries.
+
+.. autoclass:: pyhealth.datasets.MIMIC4NoteExtDIBHCDataset
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+
+
+
+
diff --git a/docs/api/tasks.rst b/docs/api/tasks.rst
@@ -230,3 +230,4 @@ Available Tasks
     Mutation Pathogenicity (COSMIC) <tasks/pyhealth.tasks.MutationPathogenicityPrediction>
     Cancer Survival Prediction (TCGA) <tasks/pyhealth.tasks.CancerSurvivalPrediction>
     Cancer Mutation Burden (TCGA) <tasks/pyhealth.tasks.CancerMutationBurden>
+    MIMIC-IV Note Tasks <tasks/pyhealth.tasks.mimic4_note_ext_dibhc_tasks>
diff --git a/docs/api/tasks/pyhealth.tasks.mimic4_note_ext_dibhc_tasks.rst b/docs/api/tasks/pyhealth.tasks.mimic4_note_ext_dibhc_tasks.rst
@@ -0,0 +1,7 @@
+pyhealth.tasks.mimic4_note_ext_dibhc_tasks
+===================================
+
+.. automodule:: pyhealth.tasks.mimic4_note_ext_dibhc_tasks
+   :members:
+   :undoc-members:
+   :show-inheritance: