Index

_

__call__() (ner.data_processing.data_collator.DataCollator method) __str__() (ner.data_processing.tokenizer.Tokenizer method) _compute_class_weights() (ner.trainers.trainer.Trainer static method) _eval_epoch() (ner.trainers.trainer.Trainer method) _get_max_length() (ner.data_processing.data_collator.DataCollator method)

_get_pred_tags_and_unk_tokens_from_text() (in module ner.utils.visualize) _initial_hidden_states() (ner.nn.models.rnn.RNN method) _print_convention() (in module ner.utils.visualize) _process_labels() (ner.data_processing.data_collator.DataCollator static method) _train_epoch() (ner.trainers.trainer.Trainer method)

A

attach_hook() (ner.nn.module.Module method)

ATTRS_MAP (in module ner.utils.styling)

C

COLOR_MAP (in module ner.utils.styling) colored() (in module ner.utils.utils)

compute_entity_f1() (in module ner.utils.metrics) compute_loss() (in module ner.utils.metrics) compute_metrics() (in module ner.utils.metrics)

D

DataCollator (class in ner.data_processing.data_collator) decode() (ner.data_processing.tokenizer.Tokenizer method)

detach_all_hooks() (ner.nn.module.Module method) detach_hook() (ner.nn.module.Module method)

E

extend() (ner.data_processing.tokenizer.Tokenizer method)

F

FFNN (class in ner.nn.models.ffnn) forward() (ner.models.ner_predictor.NERPredictor method) (ner.nn.embeddings.embedding.TokenEmbedding method) (ner.nn.models.ffnn.FFNN method) (ner.nn.models.rnn.RNN method) (ner.nn.module.Module method)

from_checkpoint() (ner.trainers.trainer.Trainer method) from_dict() (ner.data_processing.tokenizer.Tokenizer method) from_file() (ner.data_processing.tokenizer.Tokenizer method) from_pretrained() (ner.nn.module.Module method)

G

get_named_entity_spans() (in module ner.utils.utils)

get_trainable_params() (ner.nn.module.Module method)

I

id2token (ner.data_processing.tokenizer.Tokenizer property)

init_weights() (ner.nn.module.Module static method) inspect_preds() (in module ner.utils.visualize)

L

load_json() (in module ner.utils.utils)

log_metrics() (ner.utils.tracker.Tracker method)

M

module ner.data_processing.constants ner.data_processing.data_collator ner.data_processing.tokenizer ner.models.ner_predictor ner.nn.embeddings.embedding ner.nn.models.ffnn ner.nn.models.rnn ner.nn.module ner.trainers.trainer ner.utils.metrics ner.utils.styling ner.utils.tracker ner.utils.utils ner.utils.visualize

Module (class in ner.nn.module)

N

ner.data_processing.constants module ner.data_processing.data_collator module ner.data_processing.tokenizer module ner.models.ner_predictor module ner.nn.embeddings.embedding module ner.nn.models.ffnn module ner.nn.models.rnn module ner.nn.module module

ner.trainers.trainer module ner.utils.metrics module ner.utils.styling module ner.utils.tracker module ner.utils.utils module ner.utils.visualize module NER_DECODING_MAP (in module ner.data_processing.constants) NER_ENCODING_MAP (in module ner.data_processing.constants) NERPredictor (class in ner.models.ner_predictor)

P

PAD_NER_TAG (in module ner.data_processing.constants)

PAD_TOKEN (in module ner.data_processing.constants) print_params() (ner.nn.module.Module method)

R

reset() (ner.data_processing.tokenizer.Tokenizer method)

RNN (class in ner.nn.models.rnn)

S

save() (ner.data_processing.tokenizer.Tokenizer method) save_checkpoint() (ner.trainers.trainer.Trainer method) (ner.utils.tracker.Tracker method) save_model() (ner.utils.tracker.Tracker method)

save_pretrained() (ner.nn.module.Module method) set_seed() (in module ner.utils.utils) success() (in module ner.utils.utils) summary() (ner.nn.module.Module method)

T

test() (ner.trainers.trainer.Trainer static method) TokenEmbedding (class in ner.nn.embeddings.embedding) tokenize() (ner.data_processing.tokenizer.Tokenizer method) Tokenizer (class in ner.data_processing.tokenizer)

Tracker (class in ner.utils.tracker) train() (ner.data_processing.tokenizer.Tokenizer method) train_and_eval() (ner.trainers.trainer.Trainer method) Trainer (class in ner.trainers.trainer)

U

UNK_TOKEN (in module ner.data_processing.constants)

V

visualize_activations() (in module ner.utils.visualize)

vocab (ner.data_processing.tokenizer.Tokenizer property) vocab_size (ner.data_processing.tokenizer.Tokenizer property)

W

warn_once() (in module ner.utils.utils)