mirror of https://github.com/doccano/doccano.git
pythondatasetnatural-language-processingdata-labelingmachine-learningannotation-tooldatasetsactive-learningtext-annotation
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
464 B
18 lines
464 B
from typing import Dict, List, Optional
|
|
|
|
from .label import Label
|
|
|
|
|
|
class Labels:
|
|
|
|
def __init__(self, labels: List[Label]):
|
|
self.labels = labels
|
|
|
|
def replace_label(self, mapping: Optional[Dict[str, int]] = None):
|
|
if not mapping:
|
|
return self
|
|
labels = [label.replace(mapping) for label in self.labels]
|
|
return Labels(labels)
|
|
|
|
def dict(self) -> List[dict]:
|
|
return [label.dict() for label in self.labels]
|