ecg 512

broadinstitute · Dec 13, 2024 · 1eee793 · 1eee793
1 parent 6921c4a
commit 1eee793
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ml4h/tensormap/ukb/ecg.py b/ml4h/tensormap/ukb/ecg.py
@@ -581,6 +581,12 @@ def ecg_rest_section_to_segment(tm, hd5, dependents={}):
     normalization=Standardize(mean=0, std=10),
 )
 
+ecg_rest_median_512 = TensorMap(
+    'ecg_rest_median_512', Interpretation.CONTINUOUS, path_prefix='ukb_ecg_rest', shape=(512, 12), loss='logcosh',
+    activation='linear', tensor_from_file=_make_ecg_rest(), channel_map=ECG_REST_MEDIAN_LEADS,
+    normalization=ZeroMeanStd1(),
+)
+
 ecg_rest_median_raw_10_no_poor = TensorMap(
     'ecg_rest_median_raw_10', Interpretation.CONTINUOUS, path_prefix='ukb_ecg_rest', shape=(600, 12), loss='logcosh', activation='linear',
     tensor_from_file=_make_ecg_rest(skip_poor=True), metrics=['mse', 'mae'], channel_map=ECG_REST_MEDIAN_LEADS, normalization=Standardize(mean=0, std=10),