xdl recipe

broadinstitute · Oct 26, 2023 · 8d78ed0 · 8d78ed0
1 parent d48f936
commit 8d78ed0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ml4h/tensormap/mgb/xdl.py b/ml4h/tensormap/mgb/xdl.py
@@ -8,6 +8,8 @@
                                 channel_map={'no_hypertension_icd_bp': 0, 'hypertension_icd_bp': 1})
 hypertension_icd_bp_med = TensorMap(name='hypertension_icd_bp_med', interpretation=Interpretation.CATEGORICAL,
                                     channel_map={'no_hypertension_icd_bp_med': 0, 'hypertension_icd_bp_med': 1})
+hypertension_med = TensorMap(name='start_fu_hypertension_med', interpretation=Interpretation.CATEGORICAL,
+                             channel_map={'no_hypertension_medication':0, 'hypertension_medication': 1})
 
 lvef = TensorMap(name='LVEF', interpretation=Interpretation.CONTINUOUS, channel_map={'LVEF': 0})
 
@@ -17,4 +19,4 @@
 cad = TensorMap(name='cad', interpretation=Interpretation.CATEGORICAL, channel_map={'no_cad': 0, 'cad': 1})
 dm = TensorMap(name='dm', interpretation=Interpretation.CATEGORICAL, channel_map={'no_dm': 0, 'dm': 1})
 hypercholesterolemia = TensorMap(name='hypercholesterolemia', interpretation=Interpretation.CATEGORICAL,
-                                      channel_map={'no_hypercholesterolemia': 0, 'hypercholesterolemia': 1})
+                                 channel_map={'no_hypercholesterolemia': 0, 'hypercholesterolemia': 1})