mirror of https://github.com/explosion/spaCy.git
87 lines
2.6 KiB
Cython
87 lines
2.6 KiB
Cython
# cython: embedsignature=True
|
|
from cpython.ref cimport Py_INCREF
|
|
from cymem.cymem cimport Pool
|
|
from murmurhash.mrmr cimport hash64
|
|
|
|
from libc.string cimport memset
|
|
|
|
from .orth cimport word_shape
|
|
from .typedefs cimport attr_t, flags_t
|
|
import numpy
|
|
|
|
from .attrs cimport IS_ALPHA, IS_ASCII, IS_DIGIT, IS_LOWER, IS_PUNCT, IS_SPACE
|
|
from .attrs cimport IS_TITLE, IS_UPPER, LIKE_URL, LIKE_NUM, LIKE_EMAIL, IS_STOP
|
|
from .attrs cimport IS_OOV
|
|
|
|
|
|
memset(&EMPTY_LEXEME, 0, sizeof(LexemeC))
|
|
|
|
|
|
cdef int set_lex_struct_props(LexemeC* lex, dict props, StringStore string_store,
|
|
const float* empty_vec) except -1:
|
|
lex.length = props['length']
|
|
lex.orth = string_store[props['orth']]
|
|
lex.lower = string_store[props['lower']]
|
|
lex.norm = string_store[props['norm']]
|
|
lex.shape = string_store[props['shape']]
|
|
lex.prefix = string_store[props['prefix']]
|
|
lex.suffix = string_store[props['suffix']]
|
|
|
|
lex.cluster = props['cluster']
|
|
lex.prob = props['prob']
|
|
lex.sentiment = props['sentiment']
|
|
|
|
lex.flags = props['flags']
|
|
lex.repvec = empty_vec
|
|
|
|
|
|
cdef class Lexeme:
|
|
"""An entry in the vocabulary. A Lexeme has no string context --- it's a
|
|
word-type, as opposed to a word token. It therefore has no part-of-speech
|
|
tag, dependency parse, or lemma (lemmatization depends on the part-of-speech
|
|
tag).
|
|
"""
|
|
def __cinit__(self, int vec_size):
|
|
self.repvec = numpy.ndarray(shape=(vec_size,), dtype=numpy.float32)
|
|
|
|
@property
|
|
def has_repvec(self):
|
|
return self.l2_norm != 0
|
|
|
|
cpdef bint check_flag(self, attr_id_t flag_id) except -1:
|
|
cdef flags_t one = 1
|
|
return self.flags & (one << flag_id)
|
|
|
|
property is_oov:
|
|
def __get__(self): return self.check_flag(IS_OOV)
|
|
|
|
property is_alpha:
|
|
def __get__(self): return self.check_flag(IS_ALPHA)
|
|
|
|
property is_ascii:
|
|
def __get__(self): return self.check_flag(IS_ASCII)
|
|
|
|
property is_digit:
|
|
def __get__(self): return self.check_flag(IS_DIGIT)
|
|
|
|
property is_lower:
|
|
def __get__(self): return self.check_flag(IS_LOWER)
|
|
|
|
property is_title:
|
|
def __get__(self): return self.check_flag(IS_TITLE)
|
|
|
|
property is_punct:
|
|
def __get__(self): return self.check_flag(IS_PUNCT)
|
|
|
|
property is_space:
|
|
def __get__(self): return self.check_flag(IS_SPACE)
|
|
|
|
property like_url:
|
|
def __get__(self): return self.check_flag(LIKE_URL)
|
|
|
|
property like_num:
|
|
def __get__(self): return self.check_flag(LIKE_NUM)
|
|
|
|
property like_email:
|
|
def __get__(self): return self.check_flag(LIKE_EMAIL)
|