brainscore_language.model_helpers.localize

Functions

_get_layer(module, layer_name)
_register_hook(layer, key, target_dict)
extract_batch(model, input_ids, ...)
extract_representations(model, tokenizer, ...)
localize_fedorenko2010(model_id, model, ...)	Localize the model by selecting the top top_k units.
setup_hooks(model, layer_names)	set up the hooks for recording internal neural activity from the model (aka layer activations)

Classes

Fed10_langlocDataset()