spaCy/spacy/tokens.pxd

48 lines
1.0 KiB
Cython
Raw Normal View History

2014-12-16 11:44:43 +00:00
from libc.stdint cimport uint32_t
2014-12-02 12:48:05 +00:00
import numpy as np
cimport numpy as np
from cymem.cymem cimport Pool
2014-12-21 20:25:43 +00:00
from thinc.typedefs cimport atom_t
2014-12-24 06:42:00 +00:00
from .typedefs cimport flags_t, attr_id_t, attr_t
2014-12-21 20:25:43 +00:00
from .structs cimport Morphology, TokenC, Lexeme
from .vocab cimport Vocab
2014-12-19 20:03:26 +00:00
from .strings cimport StringStore
ctypedef const Lexeme* const_Lexeme_ptr
ctypedef TokenC* TokenC_ptr
ctypedef fused LexemeOrToken:
const_Lexeme_ptr
TokenC_ptr
2014-12-24 06:42:00 +00:00
cdef attr_t get_lex_attr(const Lexeme* lex, attr_id_t feat_name) nogil
cdef attr_t get_token_attr(const TokenC* lex, attr_id_t feat_name) nogil
cdef inline bint check_flag(const Lexeme* lexeme, attr_id_t flag_id) nogil:
return lexeme.flags & (1 << flag_id)
cdef class Tokens:
cdef Pool mem
2014-12-21 20:25:43 +00:00
cdef Vocab vocab
cdef list tag_names
cdef TokenC* data
cdef int length
cdef int max_length
cdef int push_back(self, int i, LexemeOrToken lex_or_tok) except -1
2014-12-23 04:18:48 +00:00
cpdef np.ndarray[long, ndim=2] to_array(self, object features)
2014-12-02 12:48:05 +00:00
cdef class Token:
2014-12-24 06:42:00 +00:00
cdef Tokens _seq
cdef readonly int i