<jupyter_start><jupyter_text>Mean of Rewards over trials<jupyter_code>mean_rwrd = rwrds.mean(0)
import seaborn as sns
sns.set_style("white")
sns.set_palette("muted")
# plt.rcParams['legend.title_fontsize'] = 'x-large'

ax_fs = 18
leg_fs = 18
tick_fs = 16
title_fs = 24
lwd = 3

plt.figure(figsize=(10, 6))
for ind, mj in enumerate(max_jumps[:6]):
    plt.plot(iterations, mean_rwrd[ind, :],
            label=str(mj.item()), linewidth=lwd)

    
for ind, mj in enumerate(max_jumps[6:]):
    plt.plot(iterations, mean_rwrd[ind+6, :],
            label=str(mj.item()), linestyle=":", linewidth=lwd)

sns.despine()
plt.xlabel("Iteration", fontsize=ax_fs)
plt.ylabel("Mean Reward", fontsize=ax_fs)
plt.tick_params("both", labelsize=tick_fs)
plt.title("Effect of Maximum Jump Size", fontsize=title_fs)
plt.legend(loc='center right', bbox_to_anchor=(1.2, 0.5),
          fontsize=leg_fs, title="Max Jump Size",
          title_fontsize="xx-large")
plt.savefig("./jump_size.pdf", bbox_inches="tight")
plt.show()<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.1653771760154739, 'fraction_numerical': 0.019342359767891684, 'mean_word_length': 5.197604790419161, 'pattern_counts': {'":': 1, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '83070', 'license': 'Apache-2.0', 'n_tokens_mistral': 447, 'n_tokens_neox': 409, 'n_words': 71, 'path': 'experiments/max_jump_ablation/output_testing.ipynb', 'repo_name': 'g-benton/PCC-RL', 'size': '1034'}