| 44 | |
| 45 | |
| 46 | class PrefixTree: |
| 47 | def __init__(self, label_name_list, tokenizer, end_symbol='<end>'): |
| 48 | self.label_name_list = label_name_list |
| 49 | self._tokenizer = tokenizer |
| 50 | self.label_name_tree = get_label_name_tree( |
| 51 | label_name_list, tokenizer, end_symbol) |
| 52 | self._end_symbol = end_symbol |
| 53 | |
| 54 | def is_end_of_tree(self, tree: Dict): |
| 55 | return len(tree) == 1 and self._end_symbol in tree |
| 56 | |
| 57 | |
| 58 | if __name__ == "__main__": |
nothing calls this directly
no outgoing calls
no test coverage detected