Update website/docs/api/spancategorizer.mdx

Co-authored-by: Adriane Boyd <adrianeboyd@gmail.com>
This commit is contained in:
kadarakos 2023-03-07 17:13:53 +01:00 committed by GitHub
parent 75b9819553
commit 086bb7f544
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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`. class for a span, then use `spancat_singlelabel`.
It uses a `Softmax` layer and treats the spans as a multi-class problem. It uses a `Softmax` layer and treats the task 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"]`.