Add labels to the optimizer trends plot (#634)

Co-authored-by: Sergiy Matusevych <sergiym@microsoft.com>
This commit is contained in:
Brian Kroth 2024-01-16 15:18:30 -06:00 коммит произвёл GitHub
Родитель 52fdc3c2e9
Коммит 769e43e96f
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 6 добавлений и 2 удалений

Просмотреть файл

@ -42,8 +42,12 @@ def _plot_optimizer_trends(exp_data: ExperimentData) -> None:
results_df = exp_data.results
plt.rcParams["figure.figsize"] = (10, 4)
sns.scatterplot(x=results_df.trial_id, y=results_df[objective_column], alpha=0.7) # Result of each trial
sns.lineplot(x=results_df.trial_id, y=results_df[objective_column].cummin()) # the best result so far (cummin)
sns.scatterplot(
x=results_df.trial_id, y=results_df[objective_column],
alpha=0.7, label="Trial") # Result of each trial
sns.lineplot(
x=results_df.trial_id, y=results_df[objective_column].cummin(),
label="Incumbent") # the best result so far (cummin)
plt.yscale('log')