Skip to content

Commit

Permalink
Consistent plotting style
Browse files Browse the repository at this point in the history
  • Loading branch information
baskrahmer committed Mar 19, 2024
1 parent b0c316a commit 42f3b3b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torchmetrics/utilities/plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,7 @@ def plot_curve(
for i, (x_, y_) in enumerate(zip(x, y)):
label = f"{legend_name}_{i}" if legend_name is not None else str(i)
label += f" AUC={score[i].item():0.3f}" if score is not None else ""
ax.plot(x_.detach().cpu(), y_.detach().cpu(), label=label)
ax.plot(x_.detach().cpu(), y_.detach().cpu(), linestyle="-", linewidth=2, label=label)
ax.legend()
else:
raise ValueError(
Expand Down

0 comments on commit 42f3b3b

Please sign in to comment.