Skip to content

Commit

Permalink
Simplify plotting labels by removing CI from legend.
Browse files Browse the repository at this point in the history
  • Loading branch information
internetcoffeephone committed Sep 24, 2020
1 parent 45f0aae commit 9c53fc3
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions visualization/plot_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,12 +116,10 @@ def plot_single_category_result(

plt.plot(interpolated_time, means, color=color, label=legend_name)
fill_color = change_color_luminosity(color, 0.2)
fill_label_name = str(1 - significance_level) + "% confidence interval"
plt.fill_between(
interpolated_time,
lower_confidence_bound,
upper_confidence_bound,
label=fill_label_name,
color=fill_color,
alpha=0.5,
)
Expand Down Expand Up @@ -311,11 +309,7 @@ def get_experiment_rewards(paths):
interp_y = np.interp(interp_x, x, y, left=np.nan, right=np.nan)
interpolated.append(interp_y)
reward_plotdata = PlotData(
[interp_x] * 5,
interpolated,
"Mean collective reward",
model_name + " experiment mean",
color,
[interp_x] * 5, interpolated, "Mean collective reward", model_name, color,
)
return reward_plotdata, env

Expand Down

0 comments on commit 9c53fc3

Please sign in to comment.