data-driven-init/plot_training.ipynb

In [1]:

from pandas import read_csv
from seaborn import relplot, set_theme

data = read_csv("results.csv")
set_theme(font_scale=1.5, style="whitegrid")
plot = relplot(data, kind="line", x="Step", y="Validation metric", hue="Method", col="Dataset", col_wrap=2, legend=False, aspect=1.2, facet_kws=dict(sharex=False, sharey=False))