changed input of rnn

rvandewater · Oct 9, 2023 · ce0831c · ce0831c
1 parent 869a611
commit ce0831c
Showing 1 changed file with 49 additions and 50 deletions.
diff --git a/configs/prediction_models/RNNpytorch.gin b/configs/prediction_models/RNNpytorch.gin
@@ -125,7 +125,55 @@ PredictionDatasetTFTpytorch.target=[
                 "MissingIndicator_48",
                 "label",
             ]
-PredictionDatasetTFTpytorch.time_varying_unknown_categoricals=["MissingIndicator_1",
+PredictionDatasetTFTpytorch.time_varying_unknown_reals=["alb",
+                "alp",
+                "alt",
+                "ast",
+                "be",
+                "bicar",
+                "bili",
+                "bili_dir",
+                "bnd",
+                "bun",
+                "ca",
+                "cai",
+                "ck",
+                "ckmb",
+                "cl",
+                "crea",
+                "crp",
+                "dbp",
+                "fgn",
+                "fio2",
+                "glu",
+                "hgb",
+                "hr",
+                "inr_pt",
+                "k",
+                "lact",
+                "lymph",
+                "map",
+                "mch",
+                "mchc",
+                "mcv",
+                "methb",
+                "mg",
+                "na",
+                "neut",
+                "o2sat",
+                "pco2",
+                "ph",
+                "phos",
+                "plt",
+                "po2",
+                "ptt",
+                "resp",
+                "sbp",
+                "temp",
+                "tnt",
+                "urine",
+                "wbc",
+                "MissingIndicator_1",
                 "MissingIndicator_2",
                 "MissingIndicator_3",
                 "MissingIndicator_4",
@@ -175,52 +223,3 @@ PredictionDatasetTFTpytorch.time_varying_unknown_categoricals=["MissingIndicator
                 "MissingIndicator_48",
                 "label",
             ]
-PredictionDatasetTFTpytorch.time_varying_unknown_reals=["alb",
-                "alp",
-                "alt",
-                "ast",
-                "be",
-                "bicar",
-                "bili",
-                "bili_dir",
-                "bnd",
-                "bun",
-                "ca",
-                "cai",
-                "ck",
-                "ckmb",
-                "cl",
-                "crea",
-                "crp",
-                "dbp",
-                "fgn",
-                "fio2",
-                "glu",
-                "hgb",
-                "hr",
-                "inr_pt",
-                "k",
-                "lact",
-                "lymph",
-                "map",
-                "mch",
-                "mchc",
-                "mcv",
-                "methb",
-                "mg",
-                "na",
-                "neut",
-                "o2sat",
-                "pco2",
-                "ph",
-                "phos",
-                "plt",
-                "po2",
-                "ptt",
-                "resp",
-                "sbp",
-                "temp",
-                "tnt",
-                "urine",
-                "wbc",]
-