RL-RVOp/plotlog.py at master · XinyuWuu/RL-RVOp · GitHub

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
import re
from numpy import array
from matplotlib import pyplot as plt
import numpy as np
step_now = 0
eps_rets = []
eps_rets = []
pi_loss = []
q_loss = []
lrs = []
fold_dir = "module_saves/data300g/module_saves/nornn33/"
# fold_dir = "module_saves/nornn29/"
log_file_url = fold_dir + "0000aaaa.txt"

with open(log_file_url, "r") as fp:
    for line in fp.readlines():
        m = re.search(r"t: ([0-9]+).*?ep_ret: [-0-9.]*?:([-0-9.]*)", line)
        if m != None:
            eps_rets.append([int(m.groups()[0]), float(m.groups()[1])])
        m = re.search(r"step ([-0-9.]*);", line)
        if m != None:
            step_now = int(m.groups()[0])
        m = re.search(r"mean losspi: ([-0-9.]*)", line)
        if m != None:
            pi_loss.append([step_now, float(m.groups()[0])])
        m = re.search(r"mean lossq: ([-0-9.]*)", line)
        if m != None:
            q_loss.append([step_now, float(m.groups()[0])])
        m = re.search(r"lr:([-0-9.E]*),([-0-9.E]*)", line)
        if m != None:
            lrs.append([step_now, float(m.groups()[0])])

np.save(fold_dir + "0000aaaa_" + "eps_rets.npy", array(eps_rets))
np.save(fold_dir + "0000aaaa_" + "pi_loss.npy", array(pi_loss))
np.save(fold_dir + "0000aaaa_" + "q_loss.npy", array(q_loss))
np.save(fold_dir + "0000aaaa_" + "lrs.npy", array(lrs))

plt.plot(array(eps_rets)[:, 0], array(eps_rets)[:, 1] / 2, label="ave ret")
plt.plot(array(pi_loss)[:, 0], array(pi_loss)[:, 1], label="pi loss")
plt.plot(array(q_loss)[:, 0], array(q_loss)[:, 1], label="q loss")
plt.plot(array(lrs)[:, 0], array(lrs)[:, 1] * 5e5, label="lr")
plt.legend()
# plt.show()
plt.ylim((-200, 255))
plt.savefig("assets/logplot.png")