hezar.models.text_classification.roberta package

Submodules

• hezar.models.text_classification.roberta.roberta_text_classification module
  • RobertaClassificationHead
    • RobertaClassificationHead.forward()
  • RobertaTextClassification
    • RobertaTextClassification.compute_loss()
    • RobertaTextClassification.forward()
    • RobertaTextClassification.post_process()
    • RobertaTextClassification.preprocess()
    • RobertaTextClassification.required_backends
    • RobertaTextClassification.skip_keys_on_load
• hezar.models.text_classification.roberta.roberta_text_classification_config module
  • RobertaTextClassificationConfig
    • RobertaTextClassificationConfig.attention_probs_dropout_prob
    • RobertaTextClassificationConfig.bos_token_id
    • RobertaTextClassificationConfig.classifier_dropout
    • RobertaTextClassificationConfig.eos_token_id
    • RobertaTextClassificationConfig.gradient_checkpointing
    • RobertaTextClassificationConfig.hidden_act
    • RobertaTextClassificationConfig.hidden_dropout_prob
    • RobertaTextClassificationConfig.hidden_size
    • RobertaTextClassificationConfig.id2label
    • RobertaTextClassificationConfig.initializer_range
    • RobertaTextClassificationConfig.intermediate_size
    • RobertaTextClassificationConfig.layer_norm_eps
    • RobertaTextClassificationConfig.max_position_embeddings
    • RobertaTextClassificationConfig.name
    • RobertaTextClassificationConfig.num_attention_heads
    • RobertaTextClassificationConfig.num_hidden_layers
    • RobertaTextClassificationConfig.num_labels
    • RobertaTextClassificationConfig.pad_token_id
    • RobertaTextClassificationConfig.position_embedding_type
    • RobertaTextClassificationConfig.task
    • RobertaTextClassificationConfig.type_vocab_size
    • RobertaTextClassificationConfig.use_cache
    • RobertaTextClassificationConfig.vocab_size

Module contents