-
Notifications
You must be signed in to change notification settings - Fork 4
/
grid_search.py
82 lines (74 loc) · 7.62 KB
/
grid_search.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#!/usr/bin/python3
import os
#ENVIRONMENT
networks=['NSFNet+GEANT2'] #['TopologyZoo/Bandcon','TopologyZoo/Ans','TopologyZoo/Aarnet','TopologyZoo/Spiralight','TopologyZoo/Cesnet2001','TopologyZoo/HostwayInternational','TopologyZoo/Rhnet','TopologyZoo/Integra','TopologyZoo/Marnet','TopologyZoo/Darkstrand','TopologyZoo/Packetexchange','TopologyZoo/GtsRomania','TopologyZoo/Noel','TopologyZoo/Restena','TopologyZoo/GtsHungary','TopologyZoo/Aconet','TopologyZoo/Amres','TopologyZoo/Arpanet19719','TopologyZoo/Arpanet19728','TopologyZoo/Garr199905','TopologyZoo/Psinet','TopologyZoo/Nsfnet','TopologyZoo/HurricaneElectric','TopologyZoo/HiberniaUs','TopologyZoo/Garr199904','TopologyZoo/Arn','TopologyZoo/Oxford','TopologyZoo/Abvt','TopologyZoo/Twaren','TopologyZoo/Renater1999','TopologyZoo/Xeex','TopologyZoo/Renater2004','TopologyZoo/Vinaren','TopologyZoo/Ilan','TopologyZoo/VisionNet','TopologyZoo/Sago','TopologyZoo/Ibm','TopologyZoo/Fatman','TopologyZoo/EliBackbone','TopologyZoo/Garr200404','TopologyZoo/KentmanJul2005','TopologyZoo/Shentel','TopologyZoo/WideJpn','TopologyZoo/Cynet','TopologyZoo/Nordu2010','TopologyZoo/Navigata','TopologyZoo/Claranet','TopologyZoo/Biznet','TopologyZoo/BtEurope','TopologyZoo/Arpanet19723','TopologyZoo/Quest','TopologyZoo/Gambia','TopologyZoo/Garr200112','TopologyZoo/Cesnet200304','TopologyZoo/Geant2001','TopologyZoo/KentmanAug2005','TopologyZoo/Peer1','TopologyZoo/Bbnplanet','TopologyZoo/Garr200109','TopologyZoo/Istar','TopologyZoo/Ernet','TopologyZoo/Jgn2Plus','TopologyZoo/Savvis','TopologyZoo/Janetbackbone','TopologyZoo/Agis','TopologyZoo/Uran','TopologyZoo/BtAsiaPac','TopologyZoo/HiberniaUk','TopologyZoo/Sprint','TopologyZoo/Grena','TopologyZoo/Compuserve','TopologyZoo/Atmnet','TopologyZoo/York','TopologyZoo/Goodnet','TopologyZoo/Renater2001']
traffics=['gravity']
routings=['ecmp']
#MULTI-AGENT
num_actions = [1,2,3,5,10]
horizons = [(25,50),(50,100),(75,150),(100,200)]
#PPOAGENT
batches = [25]
gae_lambdas = [0.95]
lrs = [0.0003]
epsilons = [0.01]
periods=[50]
gammas=[0.95]
clips=[0.2]
epochs=[3]
#ACTOR-CRITIC
link_state_sizes=[16]
message_iterations=[8]
aggregations=['min_max']
nn_sizes = [[64,128]]
dropouts = [0.15]
activations = ['selu']
for activation in activations:
for dropout in dropouts:
for nn_size in nn_sizes:
for aggregation in aggregations:
for message_iteration in message_iterations:
for link_state_size in link_state_sizes:
for network in networks:
for traffic in traffics:
for routing in routings:
for period in periods:
for gamma in gammas:
for clip in clips:
for epoch in epochs:
for batch in batches:
for gae_lambda in gae_lambdas:
for lr in lrs:
for epsilon in epsilons:
for num_action in num_actions:
for horizon in horizons:
cmd = "python ./run.py" \
" --gin_bindings='Environment.env_type = \""+network+"\"'" \
" --gin_bindings='Environment.traffic_profile = \""+traffic+"\"'" \
" --gin_bindings='Environment.routing = \""+routing+"\"'" \
" --gin_bindings='PPOAgent.max_simultaneous_actions = "+str(num_action)+"'" \
" --gin_bindings='PPOAgent.default_NSFNet_horizon = "+str(horizon[0])+"'" \
" --gin_bindings='PPOAgent.default_GEANT2_horizon = "+str(horizon[1])+"'" \
" --gin_bindings='PPOAgent.eval_period = "+str(period)+"'" \
" --gin_bindings='PPOAgent.gamma = "+str(gamma)+"'" \
" --gin_bindings='PPOAgent.clip_param = "+str(clip)+"'" \
" --gin_bindings='PPOAgent.epochs = "+str(epoch)+"'" \
" --gin_bindings='PPOAgent.batch_size = "+str(batch)+"'" \
" --gin_bindings='PPOAgent.gae_lambda = "+str(gae_lambda)+"'" \
" --gin_bindings='tf.keras.optimizers.Adam.learning_rate = "+str(lr)+"'" \
" --gin_bindings='tf.keras.optimizers.Adam.epsilon = "+str(epsilon)+"'" \
" --gin_bindings='Actor.link_state_size = "+str(link_state_size)+"'" \
" --gin_bindings='Actor.aggregation = \""+aggregation+"\"'" \
" --gin_bindings='Actor.first_hidden_layer_size = "+str(nn_size[1])+"'" \
" --gin_bindings='Actor.final_hidden_layer_size = "+str(nn_size[0])+"'" \
" --gin_bindings='Actor.dropout_rate = "+str(dropout)+"'" \
" --gin_bindings='Actor.message_iterations = "+str(message_iteration)+"'" \
" --gin_bindings='Actor.activation_fn = \""+activation+"\"'" \
" --gin_bindings='Critic.link_state_size = "+str(link_state_size)+"'" \
" --gin_bindings='Critic.aggregation = \""+aggregation+"\"'" \
" --gin_bindings='Critic.first_hidden_layer_size = "+str(nn_size[1])+"'" \
" --gin_bindings='Critic.final_hidden_layer_size = "+str(nn_size[0])+"'" \
" --gin_bindings='Critic.dropout_rate = "+str(dropout)+"'" \
" --gin_bindings='Critic.message_iterations = "+str(message_iteration)+"'" \
" --gin_bindings='Critic.activation_fn = \""+activation+"\"' &"
os.system(cmd)