1
2
3
4
5
6
7
8
9
10
11
12
13
14
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]}
}
|
