Skip to content

Fix evaluation bugs: micro_auroc NaN, softmax dim, W&B entity#193

Open
ksimmu1704-lgtm wants to merge 2 commits intoML4SCI:mainfrom
ksimmu1704-lgtm:my-first-fix
Open

Fix evaluation bugs: micro_auroc NaN, softmax dim, W&B entity#193
ksimmu1704-lgtm wants to merge 2 commits intoML4SCI:mainfrom
ksimmu1704-lgtm:my-first-fix

Conversation

@ksimmu1704-lgtm
Copy link
Copy Markdown

This PR fixes multiple issues in the evaluation pipeline:

  1. Fixed micro_auroc returning NaN by appending values during evaluation.
  2. Added dim=-1 in softmax to resolve PyTorch warning.
  3. Replaced hardcoded W&B entity with configurable environment variable.

These changes improve correctness and usability for contributors.

@ksimmu1704-lgtm
Copy link
Copy Markdown
Author

I have also fixed the W&B entity configuration in train.py and corrected a syntax issue to ensure proper initialization.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant