def __init__( self, n_min: int = 1, n_max: int = 5, split_tokens: Collection[str] = [], types: Optional[str] = None ) -> None: """Initialize MentionNgrams.""" MentionNgrams.__init__(self, n_min=n_min, n_max=n_max, split_tokens=split_tokens) if types is not None: self.types = [t.strip().lower() for t in types] else: self.types = None
def __init__(self, parts_by_doc=None, n_max=3, expand=True, split_tokens=None): """MentionNgrams specifically for transistor parts. :param parts_by_doc: a dictionary d where d[document_name.upper()] = [partA, partB, ...] """ MentionNgrams.__init__(self, n_max=n_max, split_tokens=None) self.parts_by_doc = parts_by_doc self.expander = expand_part_range if expand else (lambda x: [x])