Skip to content
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.

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

Add label names to roc curve plots #2511

Merged

Conversation

randombenj
Copy link
Contributor

@randombenj randombenj commented Apr 21, 2024

What does this PR do?

Adds the ability to display which class the plotted roc curves belong to.

Before:

image

After:

image

Before submitting
  • Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
  • Did you read the contributor guideline, Pull Request section?
  • Did you make sure to update the docs?
  • Did you write any new necessary tests?
PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.


📚 Documentation preview 📚: https://torchmetrics--2511.org.readthedocs.build/en/2511/

Copy link

codecov bot commented Apr 21, 2024

Codecov Report

Merging #2511 (bd3baae) into master (cd7ccfc) will decrease coverage by 0%.
The diff coverage is 0%.

Additional details and impacted files
@@          Coverage Diff           @@
##           master   #2511   +/-   ##
======================================
- Coverage      69%     69%   -0%     
======================================
  Files         307     307           
  Lines       17424   17427    +3     
======================================
  Hits        11998   11998           
- Misses       5426    5429    +3     

src/torchmetrics/utilities/plot.py Outdated Show resolved Hide resolved
@mergify mergify bot added the ready label Apr 22, 2024
@SkafteNicki SkafteNicki merged commit 0b33bd0 into Lightning-AI:master Apr 22, 2024
62 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants