Add configs for EDGE estimator

neuroinfo-os · Feb 28, 2019 · cbb8c67 · cbb8c67
1 parent a874c40
commit cbb8c67
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/deep_bottleneck/configs/cohort_13_EDGE/egde_relu.json b/deep_bottleneck/configs/cohort_13_EDGE/egde_relu.json
@@ -0,0 +1,15 @@
+{
+  "epochs": 8000,
+  "batch_size": 256,
+  "architecture": [10, 7, 5, 4, 3],
+  "optimizer": "adam",
+  "learning_rate": 0.0004,
+  "activation_fn": "relu",
+  "model": "models.feedforward",
+  "dataset": "datasets.harmonics",
+  "estimator": "mi_estimator.edge",
+  "discretization_range": 0.07,
+  "callbacks": [],
+  "n_runs": 1,
+  "initial_bias": 0
+}
diff --git a/deep_bottleneck/configs/cohort_13_EDGE/egde_tanh.json b/deep_bottleneck/configs/cohort_13_EDGE/egde_tanh.json
@@ -0,0 +1,15 @@
+{
+  "epochs": 8000,
+  "batch_size": 256,
+  "architecture": [10, 7, 5, 4, 3],
+  "optimizer": "adam",
+  "learning_rate": 0.0004,
+  "activation_fn": "tanh",
+  "model": "models.feedforward",
+  "dataset": "datasets.harmonics",
+  "estimator": "mi_estimator.edge",
+  "discretization_range": 0.07,
+  "callbacks": [],
+  "n_runs": 1,
+  "initial_bias": 0
+}