explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 18:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Integrate sense tagger module

Browse Source
This commit is contained in:
Matthew Honnibal 2015-07-02 00:54:46 +02:00
parent 9c74f82d20
commit 2256ba7590
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/en/__init__.py
View File

@ -13,6 +13,9 @@ from ..multi_words import RegexMerger
from .pos import EnPosTagger from .pos import EnPosTagger
from .pos import POS_TAGS from .pos import POS_TAGS
from ..sense_tagger import SenseTagger
from .attrs import get_flags from .attrs import get_flags
from . import regexes from . import regexes
@ -80,6 +83,7 @@ class English(object):
self.has_parser_model = False self.has_parser_model = False
self.has_tagger_model = False self.has_tagger_model = False
self.has_entity_model = False self.has_entity_model = False
self.has_senser_model = False
else: else:
tok_data_dir = path.join(data_dir, 'tokenizer') tok_data_dir = path.join(data_dir, 'tokenizer')
tok_rules, prefix_re, suffix_re, infix_re = read_lang_data(tok_data_dir) tok_rules, prefix_re, suffix_re, infix_re = read_lang_data(tok_data_dir)
@ -89,6 +93,7 @@ class English(object):
self.has_parser_model = path.exists(path.join(self._data_dir, 'deps')) self.has_parser_model = path.exists(path.join(self._data_dir, 'deps'))
self.has_tagger_model = path.exists(path.join(self._data_dir, 'pos')) self.has_tagger_model = path.exists(path.join(self._data_dir, 'pos'))
self.has_entity_model = path.exists(path.join(self._data_dir, 'ner')) self.has_entity_model = path.exists(path.join(self._data_dir, 'ner'))
self.has_senser_model = path.exists(path.join(self._data_dir, 'wsd'))
self.tokenizer = Tokenizer(self.vocab, tok_rules, prefix_re, self.tokenizer = Tokenizer(self.vocab, tok_rules, prefix_re,
suffix_re, infix_re, suffix_re, infix_re,
@ -102,6 +107,7 @@ class English(object):
self._tagger = None self._tagger = None
self._parser = None self._parser = None
self._entity = None self._entity = None
self._senser = None
@property @property
def tagger(self): def tagger(self):
@ -109,6 +115,12 @@ class English(object):
self._tagger = EnPosTagger(self.vocab.strings, self._data_dir) self._tagger = EnPosTagger(self.vocab.strings, self._data_dir)
return self._tagger return self._tagger
@property
def senser(self):
if self._senser is None:
self._senser = SenseTagger(self.vocab.strings, self._data_dir)
return self._senser
@property @property
def parser(self): def parser(self):
if self._parser is None: if self._parser is None: