Initial commit

1 files changed, 15 insertions, 0 deletions

diff --git a/search_space.json b/search_space.json
new file mode 100644
index 0000000..f75f66f
--- /dev/null
+++ b/search_space.json
@@ -0,0 +1,15 @@
+{
+  "num_layer": {"_type":"choice", "_value": [3, 5, 7, 10,12]},
+  "num_mlp_layers": {"_type":"choice", "_value": [2, 3, 4, 5,6,7,8]},
+
+  "hidden_dim":{"_type":"choice","_value":[64,128, 256, 512, 1024]},
+  "final_dropout": {"_type":"uniform", "_value": [0.01,0.75]},
+
+  "graph_pooling_type": {"_type":"choice", "_value": ["sum","average"]},
+  "neighbor_pooling_type": {"_type":"choice", "_value": ["sum", "average", "max"]},
+  "lr":{"_type":"uniform","_value":[0.0001, 0.1]},
+  "batch_size": {"_type":"choice", "_value": [16, 32, 64, 128,256,512,1024,2048]},
+  "iters_per_epoch": {"_type":"choice", "_value": [16, 32, 64, 128,256,512]},
+  "epochs": {"_type":"choice", "_value": [5, 10, 50, 100,200]},
+  "num_node": {"_type":"choice", "_value": [10, 20, 25, 30,35]}
+}