ner package

Subpackages

• ner.data_processing package
  • Submodules
    • ner.data_processing.constants module
      • NER_DECODING_MAP
      • NER_ENCODING_MAP
      • PAD_NER_TAG
      • PAD_TOKEN
      • UNK_TOKEN
    • ner.data_processing.data_collator module
      • DataCollator
        • DataCollator.__call__()
        • DataCollator._get_max_length()
        • DataCollator._process_labels()
    • ner.data_processing.tokenizer module
      • Tokenizer
        • Tokenizer.__str__()
        • Tokenizer.decode()
        • Tokenizer.extend()
        • Tokenizer.from_dict()
        • Tokenizer.from_file()
        • Tokenizer.id2token
        • Tokenizer.reset()
        • Tokenizer.save()
        • Tokenizer.tokenize()
        • Tokenizer.train()
        • Tokenizer.vocab
        • Tokenizer.vocab_size
• ner.models package
  • Submodules
    • ner.models.ner_predictor module
      • NERPredictor
        • NERPredictor.forward()
• ner.nn package
  • Subpackages
    • ner.nn.embeddings package
      • Submodules
        • ner.nn.embeddings.embedding module
          • TokenEmbedding
            • TokenEmbedding.forward()
    • ner.nn.models package
      • Submodules
        • ner.nn.models.ffnn module
          • FFNN
            • FFNN.forward()
        • ner.nn.models.rnn module
          • RNN
            • RNN._initial_hidden_states()
            • RNN.forward()
  • Submodules
    • ner.nn.module module
      • Module
        • Module.attach_hook()
        • Module.detach_all_hooks()
        • Module.detach_hook()
        • Module.forward()
        • Module.from_pretrained()
        • Module.get_trainable_params()
        • Module.init_weights()
        • Module.print_params()
        • Module.save_pretrained()
        • Module.summary()
• ner.trainers package
  • Submodules
    • ner.trainers.trainer module
      • Trainer
        • Trainer._compute_class_weights()
        • Trainer._eval_epoch()
        • Trainer._train_epoch()
        • Trainer.from_checkpoint()
        • Trainer.save_checkpoint()
        • Trainer.test()
        • Trainer.train_and_eval()
• ner.utils package
  • Submodules
    • ner.utils.metrics module
      • compute_entity_f1()
      • compute_loss()
      • compute_metrics()
    • ner.utils.styling module
      • ATTRS_MAP
      • COLOR_MAP
    • ner.utils.tracker module
      • Tracker
        • Tracker.log_metrics()
        • Tracker.save_checkpoint()
        • Tracker.save_model()
    • ner.utils.utils module
      • colored()
      • get_named_entity_spans()
      • load_json()
      • set_seed()
      • success()
      • warn_once()
    • ner.utils.visualize module
      • _get_pred_tags_and_unk_tokens_from_text()
      • _print_convention()
      • inspect_preds()
      • visualize_activations()