updating defaults and notebook

deepskies · Nov 5, 2024 · e22db60 · e22db60
1 parent 998b276
commit e22db60
Show file tree

Hide file tree

Showing 3 changed files with 338 additions and 1,456 deletions.
diff --git a/deepuq/train/train.py b/deepuq/train/train.py
@@ -40,7 +40,7 @@ def train_DER(
     DEVICE,
     COEFF: float,
     loss_type: str,
-    norm_params: dict,
+    norm_params: dict = {},
     model_name: str = "DER",
     EPOCHS: float = 100,
     out_dir: str = "./DeepUQResources/",
@@ -485,7 +485,7 @@ def train_DE(
     DEVICE,
     loss_type: str,
     n_models: float,
-    norm_params: dict,
+    norm_params: dict = {},
     model_name: str = "DE",
     BETA: float = 0.5,
     EPOCHS: float = 100,

diff --git a/deepuq/utils/defaults.py b/deepuq/utils/defaults.py
@@ -8,7 +8,7 @@
         "data_engine": "DataLoader",
         "data_dimension": "2D",
         "data_injection": "input",
-        "size_df": 1000,
+        "size_df": 5000,
         "noise_level": "low",
         "normalize": False,
         "uniform": False,
@@ -21,7 +21,7 @@
         "model_engine": "DE",
         "model_type": "DE",
         "loss_type": "bnll_loss",
-        "n_models": 5,
+        "n_models": 15,
         "init_lr": 0.001,
         "BETA": 0.5,
         "n_epochs": 100,
@@ -31,7 +31,8 @@
         "plot_inline": False,
         "plot_savefig": False,
         "save_chk_random_seed_init": False,
-        "rs_list": [41, 42],
+        "rs_list": [42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52,
+                    53, 54, 55, 56],
         "save_n_hidden": False,
         "n_hidden": 64,
         "save_data_size": False,
@@ -48,7 +49,7 @@
         "data_engine": "DataLoader",
         "data_dimension": "2D",
         "data_injection": "output",
-        "size_df": 1000,
+        "size_df": 5000,
         "noise_level": "low",
         "normalize": False,
         "uniform": False,

diff --git a/notebooks/DeepEnsemble_train_and_validate.ipynb b/notebooks/DeepEnsemble_train_and_validate.ipynb