hezar.models.text_classification package

Subpackages

• hezar.models.text_classification.bert package
  • Submodules
    • hezar.models.text_classification.bert.bert_text_classification module
      • BertTextClassification
    • hezar.models.text_classification.bert.bert_text_classification_config module
      • BertTextClassificationConfig
  • Module contents
• hezar.models.text_classification.distilbert package
  • Submodules
    • hezar.models.text_classification.distilbert.distilbert_text_classification module
      • DistilBertTextClassification
    • hezar.models.text_classification.distilbert.distilbert_text_classification_config module
      • DistilBertTextClassificationConfig
  • Module contents
• hezar.models.text_classification.roberta package
  • Submodules
    • hezar.models.text_classification.roberta.roberta_text_classification module
      • RobertaClassificationHead
      • RobertaTextClassification
    • hezar.models.text_classification.roberta.roberta_text_classification_config module
      • RobertaTextClassificationConfig
  • Module contents

Module contents