AuroraMiddleware/bullet3
1
0
Fork 0
mirror of https://github.com/bulletphysics/bullet3 synced 2024-12-14 05:40:05 +00:00
Code Issues Releases Wiki Activity

add plotting utility

Browse Source
This commit is contained in:
tfederico 2020-09-23 16:18:51 +01:00
parent 61df6d2ff4
commit 91cb40c5ba
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
examples/pybullet/gym/pybullet_envs/deep_mimic/plot_return.py Normal file
View File

@ -0,0 +1,30 @@
import pandas as pd
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
content = []
with open("output/agent0_log.txt", "r") as f:
content = f.readlines()
for i in range(len(content)):
content[i] = ' '.join(content[i].split())
content[i] = content[i].split(" ")
header = content[0]
df = pd.DataFrame(content[1:], columns=header, dtype=float)
tr_vals = df["Train_Return"].values
te_vals = df["Test_Return"].values
x = range(0, len(tr_vals)*400, 400)
fig, ax = plt.subplots()
ax.plot(x, tr_vals, label="Train")
ax.plot(x, te_vals, label="Test")
ax.set_xlim(0, (len(tr_vals)+1)*400)
ax.set_xlabel('Iterations')
ax.set_ylim(0, 100, 10)
ax.set_ylabel('Return')
ax.legend()
plt.savefig("output/rewards_log.png")