| | |
|---|
| | | |
|---|
| | | def seg_tokenize(txt, seg_dict): |
|---|
| | | out_txt = "" |
|---|
| | | pattern = re.compile(r"([\u4E00-\u9FA5A-Za-z0-9])") |
|---|
| | | for word in txt: |
|---|
| | | if pattern.match(word): |
|---|
| | | if word in seg_dict: |
|---|
| | | out_txt += seg_dict[word] + " " |
|---|
| | | else: |
|---|
| | | out_txt += "<unk>" + " " |
|---|
| | | if word in seg_dict: |
|---|
| | | out_txt += seg_dict[word] + " " |
|---|
| | | else: |
|---|
| | | continue |
|---|
| | | out_txt += "<unk>" + " " |
|---|
| | | return out_txt.strip().split() |
|---|
| | | |
|---|
| | | def tokenize(data, |
|---|
| | | vocab=None, |
|---|
| | | seg_dict=None, |
|---|
| | | punc_dict=None): |
|---|
| | | punc_dict=None, |
|---|
| | | bpe_tokenizer=None): |
|---|
| | | assert "text" in data |
|---|
| | | assert isinstance(vocab, dict) |
|---|
| | | text = data["text"] |
|---|
| | | token = [] |
|---|
| | | vad = -2 |
|---|
| | | |
|---|
| | | if bpe_tokenizer is not None: |
|---|
| | | text = bpe_tokenizer.text2tokens("".join(text)) |
|---|
| | | |
|---|
| | | if seg_dict is not None: |
|---|
| | | assert isinstance(seg_dict, dict) |
|---|
| | | txt = forward_segment("".join(text).lower(), seg_dict) |
|---|
