brainscore_language.model_helpers.embedding

Functions

mean_over_words(sentence_features)

remove_punctuation(word)

Remove dots, question marks, exclamation marks, and commas (.?!,) from the word

Classes

EmbeddingSubject(identifier, lookup, layer_name)

Lookup-table for word inputs.

GensimKeyedVectorsSubject(identifier, ...[, ...])

Lookup-table models in the gensim library using KeyedVectors.

_GensimLookup(model, vector_size)