mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-21 11:29:13 +03:00
ea20b72c08
* Only split + prefix if not numbers * Make like_num work for prefixed numbers * Add test for like_num
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
# coding: utf8
|
|
from __future__ import unicode_literals
|
|
|
|
from ...attrs import LIKE_NUM
|
|
|
|
|
|
_num_words = set("""
|
|
nul een één twee drie vier vijf zes zeven acht negen tien elf twaalf dertien
|
|
veertien twintig dertig veertig vijftig zestig zeventig tachtig negentig honderd
|
|
duizend miljoen miljard biljoen biljard triljoen triljard
|
|
""".split())
|
|
|
|
_ordinal_words = set("""
|
|
eerste tweede derde vierde vijfde zesde zevende achtste negende tiende elfde
|
|
twaalfde dertiende veertiende twintigste dertigste veertigste vijftigste
|
|
zestigste zeventigste tachtigste negentigste honderdste duizendste miljoenste
|
|
miljardste biljoenste biljardste triljoenste triljardste
|
|
""".split())
|
|
|
|
|
|
def like_num(text):
|
|
# This only does the most basic check for whether a token is a digit
|
|
# or matches one of the number words. In order to handle numbers like
|
|
# "drieëntwintig", more work is required.
|
|
# See this discussion: https://github.com/explosion/spaCy/pull/1177
|
|
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
|
|
if text.lower() in _ordinal_words:
|
|
return True
|
|
return False
|
|
|
|
|
|
LEX_ATTRS = {
|
|
LIKE_NUM: like_num
|
|
}
|