neuralset.extractors.text.SpacyEmbedding¶

class neuralset.extractors.text.SpacyEmbedding(*, event_types: Literal['Word'] = 'Word', aggregation: Literal['single', 'sum', 'mean', 'first', 'middle', 'last', 'cat', 'stack', 'trigger'] = 'single', allow_missing: bool = False, frequency: float = 0.0, language: str = '', infra: MapInfra = MapInfra(folder=None, cluster=None, logs='{folder}/logs/{user}/%j', job_name=None, timeout_min=None, nodes=1, tasks_per_node=1, cpus_per_task=None, gpus_per_node=None, mem_gb=None, max_pickle_size_gb=None, slurm_constraint=None, slurm_partition=None, slurm_account=None, slurm_qos=None, slurm_use_srun=False, slurm_additional_parameters=None, conda_env=None, workdir=None, permissions=511, version='0', keep_in_ram=True, max_jobs=128, min_samples_per_job=1, forbid_single_item_computation=False, mode='cached'), model_name: str = '')[source][source]¶

Get word embedding from spacy.

Parameters:

model_name (str) – Explicit spacy model name (e.g. "en_core_web_sm"). Mutually exclusive with language.
language (str) – Language name or ISO code (e.g. "english", "fr"). Resolved to a default spacy model via get_spacy_model(). Mutually exclusive with model_name. If neither is set, language is read from events.