()
| 88 | |
| 89 | |
| 90 | def main(): |
| 91 | from transformers import AutoTokenizer |
| 92 | |
| 93 | tokenizer = AutoTokenizer.from_pretrained("hfl/chinese-macbert-base") |
| 94 | dataset = build_dataset(data_dir="data/seg", task_name="seg", tokenizer=tokenizer, mode="bmes") |
| 95 | print(dataset) |
| 96 | |
| 97 | |
| 98 | if __name__ == "__main__": |
no test coverage detected