kumoai.encoder.GloVe

class kumoai.encoder.GloVe

Bases: Encoder

A GloVe encoder uses embeddings from the GloVe project to embed text in a semantically meaningful manner.

model_name: Literal['glove.test', 'glove.6B', 'glove.42B', 'glove.840B', 'glove_twitter.27B'] = 'glove.6B'

Options for the GloVe model to be used.

embedding_dim: int = 50

The embedding dimension. Must correspond to the model_name.

__init__(model_name='glove.6B', embedding_dim=50, na_strategy=NAStrategy.ZERO, name='GloVe', _stats=<factory>, _target_='kumo.encoder.sequential.GloVe')