spaCy/spacy/lang/id/lex_attrs.py

33 lines
915 B
Python
Raw Normal View History

2017-07-23 18:55:22 +03:00
# coding: utf8
from __future__ import unicode_literals
2017-07-29 13:22:21 +03:00
from ...attrs import LIKE_NUM
_num_words = ['nol', 'satu', 'dua', 'tiga', 'empat', 'lima', 'enam', 'tujuh',
'delapan', 'sembilan', 'sepuluh', 'sebelas', 'belas', 'puluh',
'ratus', 'ribu', 'juta', 'miliar', 'biliun', 'triliun', 'kuadriliun',
'kuintiliun', 'sekstiliun', 'septiliun', 'oktiliun', 'noniliun', 'desiliun']
2017-07-29 13:22:21 +03:00
def like_num(text):
text = text.replace(',', '').replace('.', '')
if text.isdigit():
return True
if text.count('/') == 1:
num, denom = text.split('/')
if num.isdigit() and denom.isdigit():
return True
2018-01-08 05:25:08 +03:00
if text.lower() in _num_words:
2017-07-29 13:22:21 +03:00
return True
if text.count('-') == 1:
2017-07-29 13:44:46 +03:00
_, num = text.split('-')
if num.isdigit() or num in _num_words:
2017-07-29 13:22:21 +03:00
return True
return False
LEX_ATTRS = {
LIKE_NUM: like_num
}