(self, tokenizer, type_schema, *args, **kwargs)
| 371 | # ET + Src -> ((Tri)(Tri)), ETText2Tri 使用 |
| 372 | class TriConstraintDecoder(ConstraintDecoder): |
| 373 | def __init__(self, tokenizer, type_schema, *args, **kwargs): |
| 374 | super().__init__(tokenizer, *args, **kwargs) |
| 375 | self.tree_end = '<tree-end>' |
| 376 | self.type_schema = type_schema |
| 377 | self.type_tree = get_label_name_tree(type_schema.role_list, |
| 378 | tokenizer=self.tokenizer, |
| 379 | end_symbol=self.tree_end) |
| 380 | self.type_start = self.tokenizer.convert_tokens_to_ids([type_start])[0] |
| 381 | self.type_end = self.tokenizer.convert_tokens_to_ids([type_end])[0] |
| 382 | |
| 383 | def check_state(self, tgt_generated): |
| 384 | if tgt_generated[-1] == self.tokenizer.pad_token_id: # t5-base |
nothing calls this directly
no test coverage detected