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 `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 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 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. Predicted spans will be saved in a [`SpanGroup`](/api/spangroup) on the doc.
Individual span scores can be found in `spangroup.attrs["scores"]`. Individual span scores can be found in `spangroup.attrs["scores"]`.