(self, tokenizer, type_schema, *args, **kwargs)
| 110 | # ET + RT + Src -> ((Role)(Role)), ETRTText2Role 使用 |
| 111 | class RoleConstraintDecoder(ConstraintDecoder): |
| 112 | def __init__(self, tokenizer, type_schema, *args, **kwargs): |
| 113 | super().__init__(tokenizer, *args, **kwargs) |
| 114 | self.tree_end = '<tree-end>' |
| 115 | self.type_schema = type_schema |
| 116 | self.type_tree = get_label_name_tree(type_schema.role_list, |
| 117 | tokenizer=self.tokenizer, |
| 118 | end_symbol=self.tree_end) |
| 119 | self.type_start = self.tokenizer.convert_tokens_to_ids([type_start])[0] |
| 120 | self.type_end = self.tokenizer.convert_tokens_to_ids([type_end])[0] |
| 121 | |
| 122 | def check_state(self, tgt_generated): |
| 123 | if tgt_generated[-1] == self.tokenizer.pad_token_id: # t5-base |
nothing calls this directly
no test coverage detected