-
Notifications
You must be signed in to change notification settings - Fork 3
/
generate_runs.py
52 lines (40 loc) · 1.16 KB
/
generate_runs.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
import numpy as np
import random
FIXED = {
"batch_size": 20,
"max_norm": 40,
"n_epochs": 20,
"num_hops": 3,
"adj_weight_typing": True,
"shuffle_batch": True
}
VAR = {
"embedding_size": ("LIN", 10, 100),
"lr": ("EXP", 0.0001, 0.01),
"l2_reg": ("EXP", 0.001, 0.1)
}
def generate_run_cmd():
cmd_str = "python memn2n "
# Add fixed params
for name, value in FIXED.iteritems():
cmd_str += " --" + name + " " + str(value)
# Add var params
for name, value in VAR.iteritems():
if type(value) == tuple:
sample = value[0]
start = float(value[1])
end = float(value[2])
r = random.uniform(0,1)
if sample == "LIN":
diff = end - start
new_value = int(start + r * diff)
elif sample == "EXP":
log_end = np.log(end)
log_start = np.log(start)
new_value = np.exp(log_start + (log_end - log_start) * r)
cmd_str += " --" + name + " " + str(new_value)
print cmd_str
NUM_RUNS = 5
# Generate a number of strings for runs
for _ in range(NUM_RUNS):
generate_run_cmd()