MIMIC-III AMA prediction task + examples + tests

docs/api/datasets/pyhealth.datasets.MIMIC3Dataset.rst

@@ -1,15 +1,9 @@
-pyhealth.datasets.MIMIC3Dataset
+pyhealth.datasets.MIMIC3Dataset
 ===================================
 
 The open Medical Information Mart for Intensive Care (MIMIC-III) database, refer to `doc <https://mimic.mit.edu/>`_ for more information. We process this database into well-structured dataset object and give user the **best flexibility and convenience** for supporting modeling and analysis.
 
 .. autoclass:: pyhealth.datasets.MIMIC3Dataset
     :members:
     :undoc-members:
-    :show-inheritance:
-
-
-
-
-
-
+    :show-inheritance:

docs/api/tasks.rst

@@ -205,6 +205,7 @@ Available Tasks
 .. toctree::
     :maxdepth: 3
 
+    AMA Prediction (MIMIC-III) <tasks/pyhealth.tasks.ama_prediction>
     Base Task <tasks/pyhealth.tasks.BaseTask>
     In-Hospital Mortality (MIMIC-IV) <tasks/pyhealth.tasks.InHospitalMortalityMIMIC4>
     MIMIC-III ICD-9 Coding <tasks/pyhealth.tasks.MIMIC3ICD9Coding>

docs/api/tasks/pyhealth.tasks.ama_prediction.rst

@@ -0,0 +1,11 @@
+pyhealth.tasks.ama_prediction
+=================================
+
+Against-medical-advice (AMA) discharge on MIMIC-III using administrative
+features only.  Use with :class:`pyhealth.datasets.MIMIC3Dataset` and
+``tables=[]`` unless you extend the cohort.
+
+.. autoclass:: pyhealth.tasks.ama_prediction.AMAPredictionMIMIC3
+    :members:
+    :undoc-members:
+    :show-inheritance: