spaCy/spacy/lang/sv/lex_attrs.py
2020-06-21 22:34:10 +02:00

60 lines
954 B
Python

from ...attrs import LIKE_NUM
_num_words = [
"noll",
"en",
"ett",
"två",
"tre",
"fyra",
"fem",
"sex",
"sju",
"åtta",
"nio",
"tio",
"elva",
"tolv",
"tretton",
"fjorton",
"femton",
"sexton",
"sjutton",
"arton",
"nitton",
"tjugo",
"trettio",
"fyrtio",
"femtio",
"sextio",
"sjuttio",
"åttio",
"nittio",
"hundra",
"tusen",
"miljon",
"miljard",
"biljon",
"biljard",
"kvadriljon",
]
def like_num(text):
if text.startswith(("+", "-", "±", "~")):
text = text[1:]
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
if text.lower() in _num_words:
return True
return False
LEX_ATTRS = {LIKE_NUM: like_num}