diff options
Diffstat (limited to 'search_space.json')
| -rw-r--r-- | search_space.json | 15 |
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]} +} |