Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Hezar Documentation
Logo
Hezar Documentation
  • Get Started
    • Overview
    • Installation
    • Quick Tour
  • Tutorial
    • Models
    • Datasets
    • Embeddings
    • Preprocessors
    • Training Recipes
      • Trainer Overview
      • Text Classification (Sentiment Analysis)
      • Sequence Labeling (POS, NER, etc.)
      • Image to Text (Image Captioning)
      • Image to Text (License Plate Recognition)
      • Speech Recognition
  • Developer Guides
    • Hazer’s Architecture
    • Advanced Guide on Models
    • Dataset Processors
    • Advanced Training
  • Reference API
    • hezar package
      • hezar.data package
        • hezar.data.datasets package
          • hezar.data.datasets.dataset module
          • hezar.data.datasets.image_captioning_dataset module
          • hezar.data.datasets.ocr_dataset module
          • hezar.data.datasets.sequence_labeling_dataset module
          • hezar.data.datasets.speech_recognition_dataset module
          • hezar.data.datasets.text_classification_dataset module
          • hezar.data.datasets.text_summarization_dataset module
        • hezar.data.data_collators module
        • hezar.data.data_samplers module
        • hezar.data.dataset_processors module
      • hezar.embeddings package
        • hezar.embeddings.embedding module
        • hezar.embeddings.fasttext module
        • hezar.embeddings.word2vec module
      • hezar.metrics package
        • hezar.metrics.accuracy module
        • hezar.metrics.bleu module
        • hezar.metrics.cer module
        • hezar.metrics.f1 module
        • hezar.metrics.metric module
        • hezar.metrics.precision module
        • hezar.metrics.recall module
        • hezar.metrics.rouge module
        • hezar.metrics.seqeval module
        • hezar.metrics.wer module
      • hezar.models package
        • hezar.models.backbone package
          • hezar.models.backbone.bert package
            • hezar.models.backbone.bert.bert module
            • hezar.models.backbone.bert.bert_config module
          • hezar.models.backbone.distilbert package
            • hezar.models.backbone.distilbert.distilbert module
            • hezar.models.backbone.distilbert.distilbert_config module
          • hezar.models.backbone.roberta package
            • hezar.models.backbone.roberta.roberta module
            • hezar.models.backbone.roberta.roberta_config module
          • hezar.models.backbone.vit package
            • hezar.models.backbone.vit.vit module
            • hezar.models.backbone.vit.vit_config module
        • hezar.models.image2text package
          • hezar.models.image2text.beit_roberta package
            • hezar.models.image2text.beit_roberta.beit_roberta_image2text module
            • hezar.models.image2text.beit_roberta.beit_roberta_image2text_config module
          • hezar.models.image2text.crnn package
            • hezar.models.image2text.crnn.crnn_decode_utils module
            • hezar.models.image2text.crnn.crnn_image2text module
            • hezar.models.image2text.crnn.crnn_image2text_config module
          • hezar.models.image2text.trocr package
            • hezar.models.image2text.trocr.trocr_image2text module
            • hezar.models.image2text.trocr.trocr_image2text_config module
          • hezar.models.image2text.vit_gpt2 package
            • hezar.models.image2text.vit_gpt2.vit_gpt2_image2text module
            • hezar.models.image2text.vit_gpt2.vit_gpt2_image2text_config module
          • hezar.models.image2text.vit_roberta package
            • hezar.models.image2text.vit_roberta.vit_roberta_image2text module
            • hezar.models.image2text.vit_roberta.vit_roberta_image2text_config module
        • hezar.models.mask_filling package
          • hezar.models.mask_filling.bert package
            • hezar.models.mask_filling.bert.bert_mask_filling module
            • hezar.models.mask_filling.bert.bert_mask_filling_config module
          • hezar.models.mask_filling.distilbert package
            • hezar.models.mask_filling.distilbert.distilbert_mask_filling module
            • hezar.models.mask_filling.distilbert.distilbert_mask_filling_config module
          • hezar.models.mask_filling.roberta package
            • hezar.models.mask_filling.roberta.roberta_mask_filling module
            • hezar.models.mask_filling.roberta.roberta_mask_filling_config module
        • hezar.models.sequence_labeling package
          • hezar.models.sequence_labeling.bert package
            • hezar.models.sequence_labeling.bert.bert_sequence_labeling module
            • hezar.models.sequence_labeling.bert.bert_sequence_labeling_config module
          • hezar.models.sequence_labeling.distilbert package
            • hezar.models.sequence_labeling.distilbert.distilbert_sequence_labeling module
            • hezar.models.sequence_labeling.distilbert.distilbert_sequence_labeling_config module
          • hezar.models.sequence_labeling.roberta package
            • hezar.models.sequence_labeling.roberta.roberta_sequence_labeling module
            • hezar.models.sequence_labeling.roberta.roberta_sequence_labeling_config module
        • hezar.models.speech_recognition package
          • hezar.models.speech_recognition.whisper package
            • hezar.models.speech_recognition.whisper.whisper_feature_extractor module
            • hezar.models.speech_recognition.whisper.whisper_speech_recognition module
            • hezar.models.speech_recognition.whisper.whisper_speech_recognition_config module
            • hezar.models.speech_recognition.whisper.whisper_tokenizer module
        • hezar.models.text_classification package
          • hezar.models.text_classification.bert package
            • hezar.models.text_classification.bert.bert_text_classification module
            • hezar.models.text_classification.bert.bert_text_classification_config module
          • hezar.models.text_classification.distilbert package
            • hezar.models.text_classification.distilbert.distilbert_text_classification module
            • hezar.models.text_classification.distilbert.distilbert_text_classification_config module
          • hezar.models.text_classification.roberta package
            • hezar.models.text_classification.roberta.roberta_text_classification module
            • hezar.models.text_classification.roberta.roberta_text_classification_config module
        • hezar.models.text_detection package
          • hezar.models.text_detection.craft package
            • hezar.models.text_detection.craft.craft_image_processor module
            • hezar.models.text_detection.craft.craft_text_detection module
            • hezar.models.text_detection.craft.craft_text_detection_config module
            • hezar.models.text_detection.craft.craft_utils module
        • hezar.models.text_generation package
          • hezar.models.text_generation.gpt2 package
            • hezar.models.text_generation.gpt2.gpt2_text_generation module
            • hezar.models.text_generation.gpt2.gpt2_text_generation_config module
          • hezar.models.text_generation.t5 package
            • hezar.models.text_generation.t5.t5_text_generation module
            • hezar.models.text_generation.t5.t5_text_generation_config module
        • hezar.models.model module
        • hezar.models.model_outputs module
      • hezar.preprocessors package
        • hezar.preprocessors.tokenizers package
          • hezar.preprocessors.tokenizers.bpe module
          • hezar.preprocessors.tokenizers.sentencepiece_bpe module
          • hezar.preprocessors.tokenizers.sentencepiece_unigram module
          • hezar.preprocessors.tokenizers.tokenizer module
          • hezar.preprocessors.tokenizers.wordpiece module
        • hezar.preprocessors.audio_feature_extractor module
        • hezar.preprocessors.image_processor module
        • hezar.preprocessors.preprocessor module
        • hezar.preprocessors.text_normalizer module
      • hezar.trainer package
        • hezar.trainer.metrics_handlers module
        • hezar.trainer.trainer module
        • hezar.trainer.trainer_utils module
      • hezar.utils package
        • hezar.utils.audio_utils module
        • hezar.utils.common_utils module
        • hezar.utils.data_utils module
        • hezar.utils.hub_utils module
        • hezar.utils.image_utils module
        • hezar.utils.integration_utils module
        • hezar.utils.logging module
        • hezar.utils.registry_utils module
      • hezar.builders module
      • hezar.configs module
      • hezar.constants module
      • hezar.registry module
  • Contributing to Hezar
Back to top
View this page
Edit this page

Reference API¶

The docs here are mostly generated from the official source code on the main branch.

  • hezar package
    • Subpackages
    • Submodules
    • Module contents
Next
hezar package
Previous
Advanced Training
Copyright © 2023, Hezar AI Team & contributors
Made with Sphinx and @pradyunsg's Furo