INNER CODE UNIT · Python
__init__
NorskRegnesentral/skweak · skweak/aggregation.py:410
def __init__(self, aggregator:Type[AbstractAggregator], **kwargs):
self.models = {}
# If the underlying model is for sequence labelling, we create one
# separate model for detecting each entity
if isinstance(self, SequenceAggregatorMixin):
for label in self.out_labels_no_prefixes: #type: ignore
args = {"name":self.name + "_base", "labels":[label], "prefixes":self.prefixes, **kwargs}
self.models[label] = aggregator(**args)
self.models[label].observed_labels = list(self.observed_labels)
else:
for label in self.out_labels: #type: ignore
args = {"name":self.name + "_base", "labels":[f"NOT/{label}", label], **kwargs}
self.models[label] = aggregator(**args)
self.models[label].observed_labels = list(self.observed_labels)