v5: update config.json

config.json

@@ -1,23 +1,23 @@
 {
-  "input_dim": 70,
+  "input_dim": 77,
   "hidden_dims_a": [
-    64,
-    32
-  ],
-  "hidden_dims_b": [
     32,
     16
   ],
+  "hidden_dims_b": [
+    128,
+    64
+  ],
   "dropout_a": 0.2,
   "dropout_b": 0.0,
-  "threshold_a": 0.47421052631578947,
+  "threshold_a": 0.60,
   "threshold_b": 0.47421052631578947,
   "loss_a": "PID",
   "loss_b": "PID",
   "hp_a": {
     "hidden_dims": [
-      64,
-      32
+      32,
+      16
     ],
     "lr": 0.001,
     "beta_kl": 0.02,
@@ -27,8 +27,8 @@
   },
   "hp_b": {
     "hidden_dims": [
-      32,
-      16
+      128,
+      64
     ],
     "lr": 0.001,
     "beta_kl": 0.02,
@@ -43,11 +43,11 @@
   "n_val": 247,
   "n_test": 247,
   "test_results": {
-    "local_rate": 0.19433198380566802,
-    "sonnet_rate": 0.2145748987854251,
-    "opus_rate": 0.5910931174089069,
-    "utility": 0.6348987817764282,
+    "local_rate": 0.0,
+    "sonnet_rate": 0.2834008097165992,
+    "opus_rate": 0.7165991902834008,
+    "utility": 0.6205263137817383,
     "oracle_utility": 0.7178542613983154,
-    "regret": 0.08295547217130661
+    "regret": 0.09732794016599655
   }
 }