From ce0831cb53229fffab73c97f5e6ad3997f78036a Mon Sep 17 00:00:00 2001 From: "youssef.mecky" Date: Mon, 9 Oct 2023 12:19:53 +0200 Subject: [PATCH] changed input of rnn --- configs/prediction_models/RNNpytorch.gin | 99 ++++++++++++------------ 1 file changed, 49 insertions(+), 50 deletions(-) diff --git a/configs/prediction_models/RNNpytorch.gin b/configs/prediction_models/RNNpytorch.gin index ad049c6d..63c1de4e 100644 --- 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",] -