belt_nlp

• belt_nlp package
  • Submodules
  • belt_nlp.bert module
    • BertClassifier
      • BertClassifier.fit()
      • BertClassifier.load()
      • BertClassifier.predict()
      • BertClassifier.predict_classes()
      • BertClassifier.predict_scores()
      • BertClassifier.save()
    • BertClassifierNN
      • BertClassifierNN.forward()
    • TokenizedDataset
  • belt_nlp.bert_truncated module
    • BertClassifierTruncated
  • belt_nlp.bert_with_pooling module
    • BertClassifierWithPooling
      • BertClassifierWithPooling.collate_fn_pooled_tokens()
  • belt_nlp.exceptions module
    • InconsistentSplittingParamsException
  • belt_nlp.splitting module
    • add_padding_tokens()
    • add_special_tokens_at_beginning_and_end()
    • check_split_parameters_consistency()
    • split_overlapping()
    • split_tokens_into_smaller_chunks()
    • stack_tokens_from_all_chunks()
    • tokenize_text_with_truncation()
    • tokenize_whole_text()
    • transform_list_of_texts()
    • transform_single_text()
  • belt_nlp.tensor_utils module
    • list_of_tensors_deep_equal()
  • Module contents