-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpush_result.py
62 lines (56 loc) · 2.41 KB
/
push_result.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
import os
from pathlib import Path
from utils.log_utils import get_loss_path, get_object_path
from utils.alg_utils import init
if __name__ == '__main__':
config = init()
config['comm_radius'] = 100
config['ebrp_estimate_radius'] = 100
config['ver'] = '1.0'
config['actor_lr'] = 1e-5
config['critic_lr'] = 1e-4
config['batch_size'] = 32
config['ebrp_alpha'] = 0.1
config['ebrp_beta'] = 0.8
big_ver = '/v4'
small_ver = '/v4.11'
seedset = [1,2,3,4,5]
path = '../iot_reward_shaping' + big_ver
if not (Path(path)).is_dir():
os.mkdir(path)
path += small_ver
if (Path(path)).is_dir():
os.system(f'rm -rf {path}')
os.mkdir(path)
for seed in seedset:
config['seed'] = seed
object_path = get_object_path(config, False)
loss_path = get_loss_path(config, False)
seed_path = path + f'/seed{seed}'
os.mkdir(seed_path)
os.system(f"cp {object_path}/* {seed_path}")
os.system(f"cp {loss_path}/* {seed_path}")
with open(f'{path}/readme.md', 'w+') as f:
f.truncate()
# f.write(f'### 0 . seed1\n')
# f.write('|name|figure|\n')
# f.write('|:--:|:--:|\n')
# f.write('|lifetime_avg||\n')
idx = 1
for seed in seedset:
f.write(f'### {idx}. seed{seed}\n')
f.write('|name|figure|\n')
f.write('|:--:|:--:|\n')
f.write('|lifetime||\n')
f.write('|aloss||\n')
f.write('|closs||\n')
f.write('|ratio||\n')
f.write('|state value||\n')
f.write('\n')
idx += 1