mirror of
https://github.com/vale981/ray
synced 2025-03-10 13:26:39 -04:00
18 lines
494 B
Python
18 lines
494 B
Python
|
import matplotlib
|
||
|
|
from matplotlib import pyplot as plt
|
||
|
|
import pickle
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
matplotlib.use("Agg")
|
||
|
|
|
||
|
|
logs = pickle.load(open("logs_rl_original.p", "rb"))
|
||
|
|
times_og = range(1, (len(logs) + 1))
|
||
|
|
reward_og = map(lambda x:x[2], logs)
|
||
|
|
plt.plot(times_og, reward_og)
|
||
|
|
plt.savefig("original_batchnum_graph")
|
||
|
|
logs = pickle.load(open("logs_rl_ray.p", "rb"))
|
||
|
|
times_ray = range(1, (len(logs) + 1))
|
||
|
|
reward_ray = map(lambda x: x[2], logs)
|
||
|
|
plt.plot(times_ray, reward_ray)
|
||
|
|
plt.savefig("rl_pong_graph")
|