<jupyter_start><jupyter_text>Train NLL  Softmax<jupyter_code>fig, ax = plt.subplots(figsize=(7,5.5))

sns.lineplot(ax=ax, data=results[results.likelihood == 'softmax'],
             x='temperature', y='train/ce_nll', hue='augment', ci='sd',
             palette=sns.color_palette([palette[2], palette[-3]]),
             marker='o', linewidth=6, markersize=14)

ax.set(xscale='log', yscale='log', xlabel=r'$T$', ylabel='BMA Train NLL')
handles, labels = ax.get_legend_handles_labels()
for h in handles:
    h.set(linewidth=6)
labels = ['No Aug.', 'Aug.']
ax.legend(handles=handles, labels=labels, loc='lower right')
# ax.legend().remove()

fig.tight_layout()
# fig.savefig('aug_noaug_train_lik_softmax.pdf', bbox_inches='tight')<jupyter_output><empty_output><jupyter_text>Dirichlet<jupyter_code>fig, ax = plt.subplots(figsize=(7,5))

sns.lineplot(ax=ax, data=results[results.likelihood == 'dirichlet'],
             x='noise', y='train/ce_nll', hue='augment', ci='sd',
             palette=sns.color_palette([palette[2], palette[-3]]),
             marker='o', linewidth=6, markersize=14)

ax.set(xscale='log', yscale='log', xlabel=r'$T$', ylabel='BMA Train NLL')
handles, labels = ax.get_legend_handles_labels()
for h in handles:
    h.set(linewidth=6)
labels = ['No Aug.', 'Aug.']
ax.legend(handles=handles, labels=labels)
# ax.legend().remove()

fig.tight_layout()
# fig.savefig('aug_noaug_train_lik_dirichlet.pdf', bbox_inches='tight')<jupyter_output><empty_output><jupyter_text>Combined<jupyter_code>fig, axes = plt.subplots(ncols=2, sharex=True, figsize=(13, 5))

sns.lineplot(ax=axes[0], data=results[results.likelihood == 'softmax'],
             x='temperature', y='train/ce_nll', hue='augment', ci='sd', legend=False,
             palette=sns.color_palette([palette[2], palette[-3]]),
             marker='o', linewidth=6, markersize=14)

sns.lineplot(ax=axes[1], data=results[results.likelihood == 'dirichlet'],
             x='noise', y='train/ce_nll', hue='augment', ci='sd',
             palette=sns.color_palette([palette[2], palette[-3]]),
             marker='o', linewidth=6, markersize=14)

axes[0].set(xscale='log', yscale='log', xlabel=r'$T$', ylabel='BMA Train NLL')

axes[1].set(xscale='log', yscale='log', xlabel=r'$\alpha_\epsilon$', ylabel='')
handles, labels = axes[1].get_legend_handles_labels()
for h in handles:
    h.set(linewidth=6)
labels = ['No Aug.', 'Aug.']
axes[1].legend(handles=handles, labels=labels, loc='lower right')

# sns.move_legend(axes[1], bbox_to_anchor=(-1, -.25, 1, 0),
#                 loc='lower center', ncol=2, borderaxespad=0., frameon=True, title='')

fig.tight_layout()
# fig.savefig('aug_noaug_train_lik.pdf', bbox_inches='tight')<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.19558823529411765, 'fraction_numerical': 0.016911764705882352, 'mean_word_length': 5.31322505800464, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 13, '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': '3', 'dir': 'jupyter-structured-clean-dedup', 'id': '129071', 'license': 'Apache-2.0', 'n_tokens_mistral': 1085, 'n_tokens_neox': 1029, 'n_words': 151, 'path': 'notebooks/viz_train_nll.ipynb', 'repo_name': 'activatedgeek/uncertainty-da-bayesian-classification', 'size': '2720'}