| from pathlib import Path | |
| from typing import Iterable | |
| from typing import List | |
| from typing import Union | |
| import warnings | |
| from typeguard import check_argument_types | |
| from espnet2.text.abs_tokenizer import AbsTokenizer | |
| class WordTokenizer(AbsTokenizer): | |
| def __init__( | |
| self, | |
| delimiter: str = None, | |
| non_linguistic_symbols: Union[Path, str, Iterable[str]] = None, | |
| remove_non_linguistic_symbols: bool = False, | |
| ): | |
| assert check_argument_types() | |
| self.delimiter = delimiter | |
| if not remove_non_linguistic_symbols and non_linguistic_symbols is not None: | |
| warnings.warn( | |
| "non_linguistic_symbols is only used " | |
| "when remove_non_linguistic_symbols = True" | |
| ) | |
| if non_linguistic_symbols is None: | |
| self.non_linguistic_symbols = set() | |
| elif isinstance(non_linguistic_symbols, (Path, str)): | |
| non_linguistic_symbols = Path(non_linguistic_symbols) | |
| with non_linguistic_symbols.open("r", encoding="utf-8") as f: | |
| self.non_linguistic_symbols = set(line.rstrip() for line in f) | |
| else: | |
| self.non_linguistic_symbols = set(non_linguistic_symbols) | |
| self.remove_non_linguistic_symbols = remove_non_linguistic_symbols | |
| def __repr__(self): | |
| return f'{self.__class__.__name__}(delimiter="{self.delimiter}")' | |
| def text2tokens(self, line: str) -> List[str]: | |
| tokens = [] | |
| for t in line.split(self.delimiter): | |
| if self.remove_non_linguistic_symbols and t in self.non_linguistic_symbols: | |
| continue | |
| tokens.append(t) | |
| return tokens | |
| def tokens2text(self, tokens: Iterable[str]) -> str: | |
| if self.delimiter is None: | |
| delimiter = " " | |
| else: | |
| delimiter = self.delimiter | |
| return delimiter.join(tokens) | |