Update website/docs/api/spancategorizer.mdx

Co-authored-by: Adriane Boyd <adrianeboyd@gmail.com>
This commit is contained in:
kadarakos 2023-03-03 15:54:11 +01:00 committed by GitHub
parent 6182213fef
commit 61576b50c7

View File

@ -18,7 +18,7 @@ need to perform multi-label classification on your spans, use `spancat`. The
`spancat` component uses a `Logistic` layer where the output class probabilities
are independent for each class. However, if you need to predict at most one true
class for a span, then use `spancat_singlelabel`. It uses a `Softmax` layer and
treats the entities as a multi-class problem.
treats the spans as a multi-class problem.
Predicted spans will be saved in a [`SpanGroup`](/api/spangroup) on the doc.
Individual span scores can be found in `spangroup.attrs["scores"]`.