(self, config=None, tokenizer=None)
| 45 | srl_vocab: List[str] |
| 46 | |
| 47 | def __init__(self, config=None, tokenizer=None): |
| 48 | super().__init__() |
| 49 | self.model = instantiate(config["model"]) |
| 50 | self.tokenizer = tokenizer |
| 51 | self.hook = Hook() |
| 52 | |
| 53 | self.cws_vocab = config["vocabs"].get("cws", [WORD_MIDDLE, WORD_START]) |
| 54 | self.pos_vocab = config["vocabs"].get("pos", []) |
| 55 | self.ner_vocab = config["vocabs"].get("ner", []) |
| 56 | self.srl_vocab = config["vocabs"].get("srl", []) |
| 57 | self.dep_vocab = config["vocabs"].get("dep", []) |
| 58 | self.sdp_vocab = config["vocabs"].get("sdp", []) |
| 59 | |
| 60 | self.supported_tasks = set() |
| 61 | self.post = {} |
| 62 | self._check() |
| 63 | |
| 64 | def add_word(self, word: str, freq: int = 1): |
| 65 | self.hook.add_word(word, freq) |
nothing calls this directly
no test coverage detected