https://github.com/VincentYu68/SymmetryCurriculumLocomotion
Tip revision: b50478f8eca673730e3ce1a5441b1948b31a5187 authored by Wenhao Yu on 31 January 2019, 22:41:28 UTC
update readme
update readme
Tip revision: b50478f
plot_actions.py
__author__ = 'yuwenhao'
import gym
import sys, os, time
import joblib
import numpy as np
import matplotlib.pyplot as plt
import json
np.random.seed(1)
if __name__ == '__main__':
ht_mean = np.loadtxt('data/force_data/action_mean_ht.txt')
ht_std = np.loadtxt('data/force_data/action_std_ht.txt')
our_mean = np.loadtxt('data/force_data/action_mean_our.txt')
our_std = np.loadtxt('data/force_data/action_std_our.txt')
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
ax.plot(our_mean, 'g', label = 'ECL + MSL')
ax.plot(ht_mean, 'y', label = 'PPO high torque')
plt.legend()
plt.title('Comparison on average action', fontsize=14)
plt.xlabel("Timestep", fontsize=14)
plt.ylabel("Average action", fontsize=14)
for tick in ax.xaxis.get_major_ticks():
tick.label.set_fontsize(13)
for tick in ax.yaxis.get_major_ticks():
tick.label.set_fontsize(13)
plt.show()