diff --git a/panels/CompareMaxAccuracyOverTime/CompareMaxAccuracyOverTime.py b/panels/CompareMaxAccuracyOverTime/CompareMaxAccuracyOverTime.py index 728edf7..7fdeed6 100644 --- a/panels/CompareMaxAccuracyOverTime/CompareMaxAccuracyOverTime.py +++ b/panels/CompareMaxAccuracyOverTime/CompareMaxAccuracyOverTime.py @@ -12,12 +12,19 @@ # Use API to fetch the metric data for all experiments in the panel scope experiment_keys = api.get_panel_experiment_keys() + +# Use API to fetch experiment colors +colors = api.get_panel_experiment_colors() +if colors: + colors = {key: value["primary"] for key,value in colors.items()} + if experiment_keys and selected_metric: data = api.get_metrics_for_chart(experiment_keys, [selected_metric]) # Prepare data for the scatter plot and calculate averages x_data = [] y_data = [] hover_text = [] + exp_color = [] # To hold date and accuracy values for calculating the average per date date_accuracy_pairs = [] @@ -38,6 +45,7 @@ x_data.append(timestamp_dt) y_data.append(max_accuracy) hover_text.append(exp_data["experimentName"]) + exp_color.append(colors.get(exp_id, "#000000")) # Store date and accuracy for average calculation date_accuracy_pairs.append((timestamp_dt, max_accuracy)) @@ -54,7 +62,9 @@ x=x_data, y=y_data, mode='markers', - marker=dict(size=10), + marker=dict( + size=10, + color=exp_color), text=hover_text, # Experiment names for hover #hoverinfo='text', # Show only hover text name=selected_metric, @@ -79,4 +89,4 @@ ) ui.display(fig) else: - ui.display("No data to plot") + ui.display("No data to plot") \ No newline at end of file diff --git a/panels/CompareMaxAccuracyOverTime/compare-max-accuracy-over-time.png b/panels/CompareMaxAccuracyOverTime/compare-max-accuracy-over-time.png index 8e1d610..4038621 100644 Binary files a/panels/CompareMaxAccuracyOverTime/compare-max-accuracy-over-time.png and b/panels/CompareMaxAccuracyOverTime/compare-max-accuracy-over-time.png differ