sns.set_style("whitegrid")
if title is not None:
plt.title(title)
if num_algorithms == 1:
colors = plt.get_cmap("tab10").colors
else: // num_algorithms > 1
After Change
ax.set_xlabel("epochs")
ax.set_ylabel(metric.replace("_", " "))
xs = list(range(1, max_len + 1))
for i in range(num_algorithms):
name_prefix = algorithm_names[
i] + " " if algorithm_names is not None and i < len(