spaCy/spacy/pipeline/trainable_pipe.pxd
Daniël de Kok c3da32b46b Change type of store_activations to Union[bool, List[str]]
When the value is:

- A bool: all activations are stored when set to `True`.
- A List[str]: the activations named in the list are stored
2022-06-22 14:28:03 +02:00

10 lines
233 B
Cython

from .pipe cimport Pipe
from ..vocab cimport Vocab
cdef class TrainablePipe(Pipe):
cdef public Vocab vocab
cdef public object model
cdef public object cfg
cdef public object scorer
cdef object _store_activations