mirror of https://github.com/explosion/spaCy.git
59 lines
1.3 KiB
Cython
59 lines
1.3 KiB
Cython
import numpy as np
|
|
cimport numpy as np
|
|
|
|
from cymem.cymem cimport Pool
|
|
|
|
from .lexeme cimport Lexeme
|
|
from .typedefs cimport flags_t
|
|
from .utf8string cimport StringStore
|
|
|
|
from thinc.typedefs cimport atom_t
|
|
|
|
|
|
cdef class Tokens:
|
|
cdef Pool mem
|
|
cdef StringStore _string_store
|
|
|
|
cdef const Lexeme** _lex_ptr
|
|
cdef int* _idx_ptr
|
|
cdef int* _pos_ptr
|
|
cdef int* _ner_ptr
|
|
cdef const Lexeme** lex
|
|
cdef int* idx
|
|
cdef int* pos
|
|
cdef int* ner
|
|
|
|
cdef int length
|
|
cdef int max_length
|
|
|
|
cdef int extend(self, int i, const Lexeme* const* lexemes, int n) except -1
|
|
cdef int push_back(self, int i, const Lexeme* lexeme) except -1
|
|
cpdef int set_tag(self, int i, int tag_type, int tag) except -1
|
|
|
|
cpdef np.ndarray[atom_t, ndim=2] get_array(self, list features)
|
|
|
|
|
|
cdef class Token:
|
|
cdef StringStore _string_store
|
|
cdef public int i
|
|
cdef public int idx
|
|
cdef public int pos
|
|
cdef public int ner
|
|
|
|
cdef public atom_t id
|
|
cdef public atom_t cluster
|
|
cdef public atom_t length
|
|
cdef public atom_t postype
|
|
cdef public atom_t sensetype
|
|
|
|
cdef public atom_t sic
|
|
cdef public atom_t norm
|
|
cdef public atom_t shape
|
|
cdef public atom_t asciied
|
|
cdef public atom_t prefix
|
|
cdef public atom_t suffix
|
|
|
|
cdef public float prob
|
|
|
|
cdef public flags_t flags
|