Skip to content

Add option in benchmark to hide scatter in snr #4558

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
alejoe91 merged 5 commits into from
May 5, 2026
Merged

Add option in benchmark to hide scatter in snr #4558

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
c291934
option in bechmark to hidde scatter in snr
samuelgarcia Apr 22, 2026
c78189d
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 22, 2026
4cebb39
Merge branch 'main' of github.com:SpikeInterface/spikeinterface into …
samuelgarcia Apr 29, 2026
5cc2773
Merge branch 'benchmark_plot_cosmetic2' of github.com:samuelgarcia/sp…
samuelgarcia Apr 29, 2026
4e817e2
legend
samuelgarcia Apr 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 16 additions & 2 deletions src/spikeinterface/benchmark/benchmark_plot_tools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -225,6 +225,7 @@ def plot_unit_counts(
with_rectangle=True,
revert_bad=True,
xticks_rotation=45.0,
show_legend=True,
figsize=None,
ax=None,
):
Expand Down Expand Up @@ -339,7 +340,9 @@ def plot_unit_counts(
xticklabels = labels_list
ax.set_xticks(np.arange(len(xticklabels)) + 1.5 - width)
ax.set_xticklabels(xticklabels, rotation=xticks_rotation)
ax.legend()

if show_legend:
ax.legend()

despine(ax)

Expand Down Expand Up @@ -413,6 +416,7 @@ def _plot_performances_vs_metric(
levels_to_group_by=None,
orientation="vertical",
show_legend=True,
show_scatter=True,
with_sigmoid_fit=False,
show_average_by_bin=True,
scatter_size=4,
Expand Down Expand Up @@ -521,7 +525,9 @@ def _plot_performances_vs_metric(
all_xs = np.concatenate(all_xs)
all_ys = np.concatenate(all_ys)

ax.scatter(all_xs, all_ys, marker=".", label=label, color=color, s=scatter_size, alpha=scatter_alpha)
if show_scatter:
ax.scatter(all_xs, all_ys, marker=".", label=label, color=color, s=scatter_size, alpha=scatter_alpha)

ax.set_ylabel(performance_name)

ax.set_ylim(-0.05, 1.05)
Expand All @@ -543,6 +549,7 @@ def plot_performances_vs_snr(
levels_to_group_by=None,
orientation="vertical",
show_legend=True,
show_scatter=True,
with_sigmoid_fit=False,
show_average_by_bin=True,
scatter_size=4,
Expand Down Expand Up @@ -571,6 +578,8 @@ def plot_performances_vs_snr(
The orientation of the plot.
show_legend : bool, default True
Show legend or not
show_scatter : bool, default True
Show scatter or not
show_sigmoid_fit : bool, default True
Show sigmoid that fit the performances.
show_average_by_bin : bool, default False
Expand Down Expand Up @@ -600,6 +609,7 @@ def plot_performances_vs_snr(
levels_to_group_by=levels_to_group_by,
orientation=orientation,
show_legend=show_legend,
show_scatter=show_scatter,
with_sigmoid_fit=with_sigmoid_fit,
show_average_by_bin=show_average_by_bin,
scatter_size=scatter_size,
Expand All @@ -618,6 +628,7 @@ def plot_performances_vs_firing_rate(
levels_to_group_by=None,
orientation="vertical",
show_legend=True,
show_scatter=True,
with_sigmoid_fit=False,
show_average_by_bin=True,
scatter_size=4,
Expand Down Expand Up @@ -646,6 +657,8 @@ def plot_performances_vs_firing_rate(
The orientation of the plot.
show_legend : bool, default True
Show legend or not
show_scatter : bool, default True
Show scatter or not
show_sigmoid_fit : bool, default True
Show sigmoid that fit the performances.
show_average_by_bin : bool, default False
Expand Down Expand Up @@ -675,6 +688,7 @@ def plot_performances_vs_firing_rate(
levels_to_group_by=levels_to_group_by,
orientation=orientation,
show_legend=show_legend,
show_scatter=show_scatter,
with_sigmoid_fit=with_sigmoid_fit,
show_average_by_bin=show_average_by_bin,
scatter_size=scatter_size,
Expand Down
Loading