hezar.models.text_classification.bert package

Submodules

• hezar.models.text_classification.bert.bert_text_classification module
  • BertTextClassification
    • BertTextClassification.compute_loss()
    • BertTextClassification.forward()
    • BertTextClassification.post_process()
    • BertTextClassification.preprocess()
    • BertTextClassification.required_backends
    • BertTextClassification.skip_keys_on_load
• hezar.models.text_classification.bert.bert_text_classification_config module
  • BertTextClassificationConfig
    • BertTextClassificationConfig.attention_probs_dropout_prob
    • BertTextClassificationConfig.classifier_dropout
    • BertTextClassificationConfig.hidden_act
    • BertTextClassificationConfig.hidden_dropout_prob
    • BertTextClassificationConfig.hidden_size
    • BertTextClassificationConfig.id2label
    • BertTextClassificationConfig.initializer_range
    • BertTextClassificationConfig.intermediate_size
    • BertTextClassificationConfig.layer_norm_eps
    • BertTextClassificationConfig.max_position_embeddings
    • BertTextClassificationConfig.name
    • BertTextClassificationConfig.num_attention_heads
    • BertTextClassificationConfig.num_hidden_layers
    • BertTextClassificationConfig.num_labels
    • BertTextClassificationConfig.pad_token_id
    • BertTextClassificationConfig.position_embedding_type
    • BertTextClassificationConfig.task
    • BertTextClassificationConfig.type_vocab_size
    • BertTextClassificationConfig.use_cache
    • BertTextClassificationConfig.vocab_size

Module contents