Sebayhi_new / camel_parser /src /utils /text_cleaner.py
MM58-crypto
wooo another commit
5ac8480
Raw
History Blame Contribute Delete
1.07 kB
from camel_tools.tokenizers.word import simple_word_tokenize
from camel_tools.utils.dediac import dediac_ar
from camel_tools.utils.normalize import normalize_unicode
from camel_tools.utils.charmap import CharMapper
def clean_line(line, arclean):
return simple_word_tokenize(arclean(dediac_ar(normalize_unicode(line.strip()))))
def split_lines_words(lines):
return [line.strip().split() for line in lines]
def clean_mad(lines):
new_lines = []
for line in lines:
new_line = []
for word in line:
# replacing underscore with mad character
if word == 'ـ':
new_line.append('_')
else:
new_line.append(word)
new_lines.append(new_line)
return new_lines
def clean_lines(lines, arclean):
return [clean_line(line, arclean) for line in lines]
if __name__ == '__main__':
lines = []
with open('data/sample_text.txt', 'r') as f:
lines = f.readlines()
arclean = CharMapper.builtin_mapper("arclean")
new_lines = clean_lines(lines, arclean)