Change default activation and learning rate step

mlexchange · Mar 7, 2024 · 02732a6 · 02732a6
1 parent 8e6e488
commit 02732a6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/assets/models.json b/assets/models.json
@@ -237,14 +237,18 @@
                     "title": "Learning Rate",
                     "param_key": "learning_rate",
                     "value": 0.001,
+                    "min": 0,
+                    "max": 0.1,
+                    "step": 0.0001,
+                    "precision": 4,
                     "comp_group": "train_model"
                 },
                 {
                     "type": "dropdown",
                     "name": "activation",
                     "title": "Activation",
                     "param_key": "activation",
-                    "value": "ReLU",
+                    "value": "Sigmoid",
                     "data": [
                         {
                             "label": "ReLU",
@@ -621,14 +625,18 @@
                     "title": "Learning Rate",
                     "param_key": "learning_rate",
                     "value": 0.001,
+                    "min": 0,
+                    "max": 0.1,
+                    "step": 0.0001,
+                    "precision": 4,
                     "comp_group": "train_model"
                 },
                 {
                     "type": "dropdown",
                     "name": "activation",
                     "title": "Activation",
                     "param_key": "activation",
-                    "value": "ReLU",
+                    "value": "Sigmoid",
                     "data": [
                         {
                             "label": "ReLU",
@@ -1009,14 +1017,18 @@
                     "title": "Learning Rate",
                     "param_key": "learning_rate",
                     "value": 0.001,
+                    "min": 0,
+                    "max": 0.1,
+                    "step": 0.0001,
+                    "precision": 4,
                     "comp_group": "train_model"
                 },
                 {
                     "type": "dropdown",
                     "name": "activation",
                     "title": "Activation",
                     "param_key": "activation",
-                    "value": "ReLU",
+                    "value": "Sigmoid",
                     "data": [
                         {
                             "label": "ReLU",

diff --git a/components/control_bar.py b/components/control_bar.py
@@ -603,7 +603,7 @@ def layout():
                                         placeholder="Select a model...",
                                     ),
                                 ),
-                                dmc.Space(h=25),
+                                dmc.Space(h=15),
                                 html.Div(id="model-parameters"),
                                 dcc.Store(id="model-parameter-values", data={}),
                                 dmc.Space(h=25),