File size: 3,472 Bytes
14b1d78 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
#
# Copyright (c) 2013-present, Anoop Kunchukuttan
# All rights reserved.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
#
#Program for tokenizing Indian language input
#
# @author Anoop Kunchukuttan
#
"""
Tokenizer for Indian languages. Currently, simple punctuation-based tokenizers
are supported (see `trivial_tokenize`). Major Indian language punctuations are
handled.
"""
import string, re, sys
from indicnlp.common import IndicNlpException
### tokenizer patterns
triv_tokenizer_indic_pat=re.compile(r'(['+string.punctuation+r'\u0964\u0965'+r'])')
triv_tokenizer_urdu_pat=re.compile(r'(['+string.punctuation+r'\u0609\u060A\u060C\u061E\u066A\u066B\u066C\u066D\u06D4'+r'])')
## date, numbers, section/article numbering
pat_num_seq=re.compile(r'([0-9]+ [,.:/] )+[0-9]+')
def trivial_tokenize_indic(text):
"""tokenize string for Indian language scripts using Brahmi-derived scripts
A trivial tokenizer which just tokenizes on the punctuation boundaries.
This also includes punctuations for the Indian language scripts (the
purna virama and the deergha virama). This is a language independent
tokenizer
Args:
text (str): text to tokenize
Returns:
list: list of tokens
"""
tok_str=triv_tokenizer_indic_pat.sub(r' \1 ',text.replace('\t',' '))
# return re.sub(r'[ ]+',' ',tok_str).strip(' ').split(' ')
s=re.sub(r'[ ]+',' ',tok_str).strip(' ')
# do not tokenize numbers and dates
new_s=''
prev=0
for m in pat_num_seq.finditer(s):
start=m.start()
end=m.end()
if start>prev:
new_s=new_s+s[prev:start]
new_s=new_s+s[start:end].replace(' ','')
prev=end
new_s=new_s+s[prev:]
s=new_s
return s.split(' ')
def trivial_tokenize_urdu(text):
"""tokenize Urdu string
A trivial tokenizer which just tokenizes on the punctuation boundaries.
This also includes punctuations for the Urdu script.
These punctuations characters were identified from the Unicode database
for Arabic script by looking for punctuation symbols.
Args:
text (str): text to tokenize
Returns:
list: list of tokens
"""
tok_str=triv_tokenizer_urdu_pat.sub(r' \1 ',text.replace('\t',' '))
return re.sub(r'[ ]+',' ',tok_str).strip(' ').split(' ')
def trivial_tokenize(text,lang='hi'):
"""trivial tokenizer for Indian languages using Brahmi for Arabic scripts
A trivial tokenizer which just tokenizes on the punctuation boundaries.
Major punctuations specific to Indian langauges are handled.
These punctuations characters were identified from the Unicode database.
Args:
text (str): text to tokenize
lang (str): ISO 639-2 language code
Returns:
list: list of tokens
"""
if lang=='ur':
return trivial_tokenize_urdu(text)
else:
return trivial_tokenize_indic(text)
# if __name__ == '__main__':
# if len(sys.argv)<4:
# print("Usage: python indic_tokenize.py <infile> <outfile> <language>")
# sys.exit(1)
# with open(sys.argv[1],'r', encoding='utf-8') as ifile:
# with open(sys.argv[2],'w', encoding='utf-8') as ofile:
# for line in ifile:
# tokenized_line=' '.join(trivial_tokenize(line,sys.argv[3]))
# ofile.write(tokenized_line)
|