diagNNose
latest
Documentation overview:
activations
Activation Index
Activation Reader
Activation Writer
Selection Funcs
attribute
Decomposer
Explainer
GCD Tensor
Shapley Tensor
Utils
config
Arg Descriptions
Config Dict
corpus
Corpus
Create Iterator
Create Labels
extract
Extractor
Simple Extract
models
Import Model
Init States
Language Model
Recurrent LM
Transformer LM
models.wrappers
AWD LSTM
Forward LSTM
Google LM
probe
Data Loader
DC Trainer
Logreg
syntax
Evaluator
Task
syntax.tasks
Lakretz
Linzen
Marvin
Warstadt
Warstadt Preproc
Winobias
tokenizer
C2I
Create
W2I
utils
Misc
Pickle
Custom type definitions
Activations
Probing
diagNNose
Docs
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
_
__getitem__() (diagnnose.activations.activation_reader.ActivationReader method)
__iter__() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
__len__() (diagnnose.activations.activation_reader.ActivationReader method)
A
activation_index_len() (in module diagnnose.activations.activation_index)
activation_index_to_iterable() (in module diagnnose.activations.activation_index)
activation_names() (diagnnose.models.language_model.LanguageModel method)
(diagnnose.models.recurrent_lm.RecurrentLM method)
(diagnnose.models.transformer_lm.TransformerLM static method)
activation_ranges (diagnnose.activations.activation_reader.ActivationReader property)
ActivationReader (class in diagnnose.activations.activation_reader)
activations
activations() (diagnnose.activations.activation_reader.ActivationReader method)
ActivationWriter (class in diagnnose.activations.activation_writer)
add_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
all_nouns (diagnnose.syntax.tasks.linzen.RawItem property)
attach_tokenizer() (in module diagnnose.corpus.corpus)
attribute
AWDLSTM (class in diagnnose.models.wrappers.awd_lstm)
B
base_model() (diagnnose.models.transformer_lm.TransformerLM method)
bias (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
C
C2I (class in diagnnose.tokenizer.c2i)
calc_exact_shapley_values() (in module diagnnose.attribute.utils)
calc_sample_shapley_values() (in module diagnnose.attribute.utils)
calc_shapley_factors() (in module diagnnose.attribute.utils)
cat_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
CharCNN (class in diagnnose.models.wrappers.google_lm)
classifier (diagnnose.probe.dc_trainer.DCTrainer attribute)
concat_pickle_dumps() (diagnnose.activations.activation_writer.ActivationWriter method)
config
ContextualDecomposer (class in diagnnose.attribute.decomposer)
corpus
Corpus (class in diagnnose.corpus.corpus)
create() (diagnnose.corpus.corpus.Corpus class method)
create_attention_mask() (diagnnose.models.transformer_lm.TransformerLM method)
create_char_vocab() (in module diagnnose.tokenizer.create)
create_config_dict() (in module diagnnose.config.config_dict)
create_downstream_corpus() (in module diagnnose.syntax.tasks.warstadt_preproc)
create_examples() (diagnnose.corpus.corpus.Corpus static method)
create_fields() (diagnnose.corpus.corpus.Corpus static method)
create_header() (diagnnose.corpus.corpus.Corpus static method)
create_inputs_embeds() (diagnnose.models.language_model.LanguageModel method)
(diagnnose.models.recurrent_lm.RecurrentLM method)
(diagnnose.models.transformer_lm.TransformerLM method)
(diagnnose.models.wrappers.google_lm.GoogleLM method)
create_labels_from_corpus() (in module diagnnose.corpus.create_labels)
create_output_files() (diagnnose.activations.activation_writer.ActivationWriter method)
create_raw_corpus() (diagnnose.corpus.corpus.Corpus static method)
create_tokenizer() (in module diagnnose.tokenizer.create)
D
DataLoader (class in diagnnose.probe.data_loader)
DCTrainer (class in diagnnose.probe.dc_trainer)
decode() (diagnnose.models.recurrent_lm.RecurrentLM method)
(diagnnose.models.wrappers.google_lm.GoogleLM method)
decoder (diagnnose.models.transformer_lm.TransformerLM property)
decoder_b (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
decoder_w (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
decompose() (diagnnose.attribute.decomposer.ContextualDecomposer method)
(diagnnose.attribute.decomposer.Decomposer method)
(diagnnose.attribute.decomposer.ShapleyDecomposer method)
Decomposer (class in diagnnose.attribute.decomposer)
descriptions (diagnnose.syntax.tasks.lakretz.LakretzTask attribute)
diagnnose.activations
module
diagnnose.activations.activation_index
module
diagnnose.activations.activation_reader
module
diagnnose.activations.activation_writer
module
diagnnose.activations.selection_funcs
module
diagnnose.attribute
module
diagnnose.attribute.decomposer
module
diagnnose.attribute.explainer
module
diagnnose.attribute.gcd_tensor
module
diagnnose.attribute.shapley_tensor
module
diagnnose.attribute.utils
module
diagnnose.config
module
diagnnose.config.arg_descriptions
module
diagnnose.config.config_dict
module
diagnnose.corpus
module
diagnnose.corpus.corpus
module
diagnnose.corpus.create_iterator
module
diagnnose.corpus.create_labels
module
diagnnose.extract
module
diagnnose.extract.extractor
module
diagnnose.extract.simple_extract
module
diagnnose.models
module
diagnnose.models.import_model
module
diagnnose.models.language_model
module
diagnnose.models.recurrent_lm
module
diagnnose.models.transformer_lm
module
diagnnose.models.wrappers
module
diagnnose.models.wrappers.awd_lstm
module
diagnnose.models.wrappers.forward_lstm
module
diagnnose.models.wrappers.google_lm
module
diagnnose.probe
module
diagnnose.probe.data_loader
module
diagnnose.probe.dc_trainer
module
diagnnose.probe.logreg
module
diagnnose.syntax
module
diagnnose.syntax.evaluator
module
diagnnose.syntax.task
module
diagnnose.syntax.tasks
module
diagnnose.syntax.tasks.lakretz
module
diagnnose.syntax.tasks.linzen
module
diagnnose.syntax.tasks.marvin
module
diagnnose.syntax.tasks.warstadt
module
diagnnose.syntax.tasks.warstadt_preproc
module
diagnnose.syntax.tasks.winobias
module
diagnnose.tokenizer
module
diagnnose.tokenizer.c2i
module
diagnnose.tokenizer.create
module
diagnnose.tokenizer.w2i
module
diagnnose.utils
module
diagnnose.utils.misc
module
diagnnose.utils.pickle
module
dim() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
distance (diagnnose.syntax.tasks.linzen.RawItem property)
dropout2d_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
dropout3d_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
dropout_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
dump_activations() (diagnnose.activations.activation_writer.ActivationWriter method)
dump_meta_info() (diagnnose.activations.activation_writer.ActivationWriter method)
dump_pickle() (in module diagnnose.utils.pickle)
E
embeddings (diagnnose.models.transformer_lm.TransformerLM property)
explain() (diagnnose.attribute.explainer.Explainer method)
Explainer (class in diagnnose.attribute.explainer)
extract
extract() (diagnnose.extract.extractor.Extractor method)
Extractor (class in diagnnose.extract.extractor)
F
final_hidden() (diagnnose.models.recurrent_lm.RecurrentLM method)
final_sen_token() (in module diagnnose.activations.selection_funcs)
final_token() (in module diagnnose.activations.selection_funcs)
first_n() (in module diagnnose.activations.selection_funcs)
forget_offset (diagnnose.models.recurrent_lm.RecurrentLM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
forward() (diagnnose.models.language_model.LanguageModel method)
(diagnnose.models.recurrent_lm.RecurrentLM method)
(diagnnose.models.transformer_lm.TransformerLM method)
(diagnnose.models.wrappers.google_lm.CharCNN method)
(diagnnose.probe.logreg.LogRegModule method)
forward_cell() (diagnnose.models.recurrent_lm.RecurrentLM method)
forward_step() (diagnnose.models.recurrent_lm.RecurrentLM method)
ForwardLSTM (class in diagnnose.models.wrappers.forward_lstm)
G
GCDTensor (class in diagnnose.attribute.gcd_tensor)
get_item_generator() (diagnnose.activations.activation_reader.ActivationReader static method)
get_loss() (diagnnose.probe.logreg.L1NeuralNetClassifier method)
GoogleLM (class in diagnnose.models.wrappers.google_lm)
H
has_nsubj (diagnnose.syntax.tasks.linzen.RawItem property)
has_rel (diagnnose.syntax.tasks.linzen.RawItem property)
I
ih_concat_order (diagnnose.models.recurrent_lm.RecurrentLM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
import_model() (in module diagnnose.models.import_model)
in_sen_ids() (in module diagnnose.activations.selection_funcs)
init_contributions() (in module diagnnose.attribute.utils)
init_hidden() (diagnnose.models.recurrent_lm.RecurrentLM method)
init_states (diagnnose.models.recurrent_lm.RecurrentLM attribute)
initialize() (diagnnose.syntax.task.SyntaxEvalTask method)
(diagnnose.syntax.tasks.lakretz.LakretzTask method)
(diagnnose.syntax.tasks.linzen.LinzenTask method)
(diagnnose.syntax.tasks.marvin.MarvinTask method)
(diagnnose.syntax.tasks.warstadt.WarstadtTask method)
(diagnnose.syntax.tasks.winobias.WinobiasTask method)
intersection() (in module diagnnose.activations.selection_funcs)
is_causal (diagnnose.models.language_model.LanguageModel attribute)
(diagnnose.models.recurrent_lm.RecurrentLM attribute)
L
L1NeuralNetClassifier (class in diagnnose.probe.logreg)
LakretzTask (class in diagnnose.syntax.tasks.lakretz)
LanguageModel (class in diagnnose.models.language_model)
last_intervening (diagnnose.syntax.tasks.linzen.RawItem property)
linear_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
LinzenTask (class in diagnnose.syntax.tasks.linzen)
load() (diagnnose.probe.data_loader.DataLoader method)
load_model() (diagnnose.models.transformer_lm.TransformerLM method)
load_pickle() (in module diagnnose.utils.pickle)
LogRegModule (class in diagnnose.probe.logreg)
M
MarvinTask (class in diagnnose.syntax.tasks.marvin)
matmul_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
max_depth (diagnnose.syntax.tasks.linzen.RawItem property)
max_word_length (diagnnose.tokenizer.c2i.C2I property)
merge_dicts() (in module diagnnose.utils.misc)
models
module
diagnnose.activations
diagnnose.activations.activation_index
diagnnose.activations.activation_reader
diagnnose.activations.activation_writer
diagnnose.activations.selection_funcs
diagnnose.attribute
diagnnose.attribute.decomposer
diagnnose.attribute.explainer
diagnnose.attribute.gcd_tensor
diagnnose.attribute.shapley_tensor
diagnnose.attribute.utils
diagnnose.config
diagnnose.config.arg_descriptions
diagnnose.config.config_dict
diagnnose.corpus
diagnnose.corpus.corpus
diagnnose.corpus.create_iterator
diagnnose.corpus.create_labels
diagnnose.extract
diagnnose.extract.extractor
diagnnose.extract.simple_extract
diagnnose.models
diagnnose.models.import_model
diagnnose.models.language_model
diagnnose.models.recurrent_lm
diagnnose.models.transformer_lm
diagnnose.models.wrappers
diagnnose.models.wrappers.awd_lstm
diagnnose.models.wrappers.forward_lstm
diagnnose.models.wrappers.google_lm
diagnnose.probe
diagnnose.probe.data_loader
diagnnose.probe.dc_trainer
diagnnose.probe.logreg
diagnnose.syntax
diagnnose.syntax.evaluator
diagnnose.syntax.task
diagnnose.syntax.tasks
diagnnose.syntax.tasks.lakretz
diagnnose.syntax.tasks.linzen
diagnnose.syntax.tasks.marvin
diagnnose.syntax.tasks.warstadt
diagnnose.syntax.tasks.warstadt_preproc
diagnnose.syntax.tasks.winobias
diagnnose.tokenizer
diagnnose.tokenizer.c2i
diagnnose.tokenizer.create
diagnnose.tokenizer.w2i
diagnnose.utils
diagnnose.utils.misc
diagnnose.utils.pickle
monkey_patch() (in module diagnnose.attribute.utils)
mul_contributions() (diagnnose.attribute.gcd_tensor.GCDTensor method)
(diagnnose.attribute.shapley_tensor.ShapleyTensor method)
N
n_diff_intervening (diagnnose.syntax.tasks.linzen.RawItem property)
n_intervening (diagnnose.syntax.tasks.linzen.RawItem property)
negate() (in module diagnnose.activations.selection_funcs)
nhid() (diagnnose.models.language_model.LanguageModel method)
(diagnnose.models.recurrent_lm.RecurrentLM method)
(diagnnose.models.transformer_lm.TransformerLM method)
no_special_tokens() (in module diagnnose.activations.selection_funcs)
normalize_contributions() (in module diagnnose.attribute.utils)
nouns_up_to_verb (diagnnose.syntax.tasks.linzen.RawItem property)
nth_token() (in module diagnnose.activations.selection_funcs)
num_features (diagnnose.attribute.shapley_tensor.ShapleyTensor property)
num_layers (diagnnose.models.language_model.LanguageModel property)
(diagnnose.models.recurrent_lm.RecurrentLM property)
(diagnnose.models.transformer_lm.TransformerLM property)
O
only_mask_token() (in module diagnnose.activations.selection_funcs)
orig_sentence (diagnnose.syntax.tasks.linzen.RawItem property)
output_size (diagnnose.models.recurrent_lm.RecurrentLM property)
P
param_names() (diagnnose.models.wrappers.awd_lstm.AWDLSTM static method)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM static method)
peepholes (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
perm_generator() (in module diagnnose.attribute.utils)
pos_sentence (diagnnose.syntax.tasks.linzen.RawItem property)
preproc_warstadt() (in module diagnnose.syntax.tasks.warstadt_preproc)
print_attributions() (diagnnose.attribute.explainer.Explainer method)
probe
profile() (in module diagnnose.utils.misc)
R
RawItem (class in diagnnose.syntax.tasks.linzen)
RecurrentLM (class in diagnnose.models.recurrent_lm)
return_all() (in module diagnnose.activations.selection_funcs)
run() (diagnnose.syntax.evaluator.SyntacticEvaluator method)
(diagnnose.syntax.task.SyntaxEvalTask method)
S
selection_func (diagnnose.activations.activation_reader.ActivationReader property)
sentence (diagnnose.syntax.tasks.linzen.RawItem property)
set_activation_ranges() (diagnnose.extract.extractor.Extractor method)
set_init_states() (diagnnose.models.recurrent_lm.RecurrentLM method)
ShapleyDecomposer (class in diagnnose.attribute.decomposer)
ShapleyTensor (class in diagnnose.attribute.shapley_tensor)
simple_extract() (in module diagnnose.extract.simple_extract)
size() (diagnnose.attribute.shapley_tensor.ShapleyTensor method)
sizes (diagnnose.models.language_model.LanguageModel attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
slice() (diagnnose.corpus.corpus.Corpus method)
SoftMax (class in diagnnose.models.wrappers.google_lm)
split_contributions() (diagnnose.attribute.shapley_tensor.ShapleyTensor static method)
split_order (diagnnose.models.recurrent_lm.RecurrentLM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
subj (diagnnose.syntax.tasks.linzen.RawItem property)
subj_index (diagnnose.syntax.tasks.linzen.RawItem property)
subj_pos (diagnnose.syntax.tasks.linzen.RawItem property)
sum_contributions() (in module diagnnose.attribute.utils)
suppress_print() (in module diagnnose.utils.misc)
SyntacticEvaluator (class in diagnnose.syntax.evaluator)
syntax
SyntaxEvalTask (class in diagnnose.syntax.task)
T
to() (diagnnose.activations.activation_reader.ActivationReader method)
token_to_char_ids() (diagnnose.tokenizer.c2i.C2I method)
token_to_index() (in module diagnnose.tokenizer.create)
top_layer (diagnnose.models.language_model.LanguageModel property)
(diagnnose.models.recurrent_lm.RecurrentLM property)
(diagnnose.models.transformer_lm.TransformerLM property)
train() (diagnnose.probe.dc_trainer.DCTrainer method)
training (diagnnose.models.transformer_lm.TransformerLM attribute)
(diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.CharCNN attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
(diagnnose.probe.logreg.LogRegModule attribute)
TransformerLM (class in diagnnose.models.transformer_lm)
U
union() (in module diagnnose.activations.selection_funcs)
unwrap() (in module diagnnose.attribute.utils)
update_contributions() (in module diagnnose.attribute.utils)
use_char_embs (diagnnose.models.recurrent_lm.RecurrentLM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
use_peepholes (diagnnose.models.recurrent_lm.RecurrentLM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
V
verb (diagnnose.syntax.tasks.linzen.RawItem property)
verb_index (diagnnose.syntax.tasks.linzen.RawItem property)
verb_pos (diagnnose.syntax.tasks.linzen.RawItem property)
W
W2I (class in diagnnose.tokenizer.w2i)
w2i (diagnnose.tokenizer.w2i.W2I property)
WarstadtTask (class in diagnnose.syntax.tasks.warstadt)
weight (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
weight_P (diagnnose.models.wrappers.awd_lstm.AWDLSTM attribute)
(diagnnose.models.wrappers.forward_lstm.ForwardLSTM attribute)
(diagnnose.models.wrappers.google_lm.GoogleLM attribute)
WinobiasTask (class in diagnnose.syntax.tasks.winobias)
wrap_inputs_embeds() (diagnnose.attribute.decomposer.ContextualDecomposer method)
(diagnnose.attribute.decomposer.Decomposer method)
(diagnnose.attribute.decomposer.ShapleyDecomposer method)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.