spaCy/spacy/lang/sv/lex_attrs.py
Samuel Rodríguez Medina 8602daba85
Swedish like_num (#5371)
* Sign contributor agreement.

* Add like_num functionality to Swedish.

* Update spacy/tests/lang/sv/test_lex_attrs.py

Co-Authored-By: Sofie Van Landeghem <svlandeg@users.noreply.github.com>

* Update contributor agreement

Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
2020-04-29 21:25:22 +02:00

63 lines
1009 B
Python

# coding: utf8
from __future__ import unicode_literals
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}