(x, color=None, label=None)
| 82 | return rewards |
| 83 | |
| 84 | def range_plot(x, color=None, label=None): |
| 85 | print(f'{len(x)} experiments, (num_generation, num_population)={x[0].shape}') |
| 86 | x = prepare_reward(x) |
| 87 | center = x.quantile(0.5, dim=-1) |
| 88 | lower = x.quantile(0.25, dim=-1) |
| 89 | upper = x.quantile(0.75, dim=-1) |
| 90 | X = np.arange(len(center)) + 1 |
| 91 | plt.plot(X, center, color=color, label=label) |
| 92 | plt.fill_between(X, lower, upper, alpha=0.25, color=color, edgecolor='none') |
| 93 | |
| 94 | |
| 95 | def reward_compare_plot(*rewards, labels=None, colors=None, ax=None): |
no test coverage detected