Unverified Commit 2e5c6f59 authored by Serhii Hamotskyi's avatar Serhii Hamotskyi Committed by GitHub
Browse files

Fix error/typo in docstring of TokenClassificationPipeline (#19798)

parent cca51aa1
...@@ -173,7 +173,7 @@ class TokenClassificationPipeline(Pipeline): ...@@ -173,7 +173,7 @@ class TokenClassificationPipeline(Pipeline):
the following keys: the following keys:
- **word** (`str`) -- The token/word classified. This is obtained by decoding the selected tokens. If you - **word** (`str`) -- The token/word classified. This is obtained by decoding the selected tokens. If you
want to have the exact string in the original sentence, use `start` and `stop`. want to have the exact string in the original sentence, use `start` and `end`.
- **score** (`float`) -- The corresponding probability for `entity`. - **score** (`float`) -- The corresponding probability for `entity`.
- **entity** (`str`) -- The entity predicted for that token/word (it is named *entity_group* when - **entity** (`str`) -- The entity predicted for that token/word (it is named *entity_group* when
*aggregation_strategy* is not `"none"`. *aggregation_strategy* is not `"none"`.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment