hezar.models.text_classification.distilbert package

Submodules

• hezar.models.text_classification.distilbert.distilbert_text_classification module
  • DistilBertTextClassification
    • DistilBertTextClassification.compute_loss()
    • DistilBertTextClassification.forward()
    • DistilBertTextClassification.post_process()
    • DistilBertTextClassification.preprocess()
    • DistilBertTextClassification.required_backends
• hezar.models.text_classification.distilbert.distilbert_text_classification_config module
  • DistilBertTextClassificationConfig
    • DistilBertTextClassificationConfig.activation
    • DistilBertTextClassificationConfig.attention_dropout
    • DistilBertTextClassificationConfig.dim
    • DistilBertTextClassificationConfig.dropout
    • DistilBertTextClassificationConfig.hidden_dim
    • DistilBertTextClassificationConfig.id2label
    • DistilBertTextClassificationConfig.initializer_range
    • DistilBertTextClassificationConfig.max_position_embeddings
    • DistilBertTextClassificationConfig.n_heads
    • DistilBertTextClassificationConfig.n_layers
    • DistilBertTextClassificationConfig.name
    • DistilBertTextClassificationConfig.num_labels
    • DistilBertTextClassificationConfig.output_past
    • DistilBertTextClassificationConfig.pad_token_id
    • DistilBertTextClassificationConfig.qa_dropout
    • DistilBertTextClassificationConfig.seq_classif_dropout
    • DistilBertTextClassificationConfig.task
    • DistilBertTextClassificationConfig.tie_weights_
    • DistilBertTextClassificationConfig.vocab_size

Module contents