17 lines
434 B
Python
17 lines
434 B
Python
from typing import Any, List, Optional
|
|
|
|
from typing_extensions import Literal # <= Python 3.7
|
|
|
|
from ..hashable_base_model import HashableBaseModel
|
|
|
|
|
|
class LabeledToken(HashableBaseModel):
|
|
token: str
|
|
tag: Literal["B", "I", "O", "E", "S"]
|
|
confidence: float
|
|
explanation: Optional[Any] = None
|
|
|
|
|
|
class SequenceLabelingOutput(HashableBaseModel):
|
|
labeled_tokens: List[LabeledToken]
|
|
explanation: Optional[Any] = None
|