Skip to content
Discussion options

You must be logged in to vote

No, you can only assign weights to the top-level (float) entries in the scores dict. Even if you could do this, it would only affect the eval step (primarily used for early stopping) and wouldn't influence the actual model training at all.

If you have a rare class, you can try other options like augmenting your training data or experimenting with having the rare class in its own spancat component with a lower threshold (obviously the whole pipeline is slower overall, but if recall is important the tradeoff might be worth it for your task).

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@phlobo
Comment options

Answer selected by phlobo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feat / scorer Feature: Scorer feat / spancat Feature: Span Categorizer
2 participants