Skip to content

Commit

Permalink
deploy: 613fd34
Browse files Browse the repository at this point in the history
  • Loading branch information
arxyzan committed Jan 2, 2024
0 parents commit 5f2d8f7
Show file tree
Hide file tree
Showing 572 changed files with 170,894 additions and 0 deletions.
4 changes: 4 additions & 0 deletions .buildinfo
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
config: 46dac3e782f238b726e25024d3fa6112
tags: 645f666f9bcd5a90fca523b33c5a78b7
Binary file added .doctrees/contributing.doctree
Binary file not shown.
Binary file added .doctrees/environment.pickle
Binary file not shown.
Binary file added .doctrees/get_started/index.doctree
Binary file not shown.
Binary file added .doctrees/get_started/installation.doctree
Binary file not shown.
Binary file added .doctrees/get_started/overview.doctree
Binary file not shown.
Binary file added .doctrees/get_started/quick_tour.doctree
Binary file not shown.
Binary file added .doctrees/guide/advanced_training.doctree
Binary file not shown.
Binary file added .doctrees/guide/hezar_architecture.doctree
Binary file not shown.
Binary file added .doctrees/guide/index.doctree
Binary file not shown.
Binary file added .doctrees/guide/models_advanced.doctree
Binary file not shown.
Binary file added .doctrees/guide/trainer_in_depth.doctree
Binary file not shown.
Binary file added .doctrees/index.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.builders.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.configs.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.constants.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.data.datasets.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.data.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.embeddings.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.accuracy.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.bleu.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.cer.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.f1.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.metric.doctree
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.recall.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.rouge.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.seqeval.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.metrics.wer.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.models.backbone.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.models.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.models.image2text.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.models.model.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.preprocessors.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.registry.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.trainer.doctree
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.trainer.trainer.doctree
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.audio_utils.doctree
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.file_utils.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.hub_utils.doctree
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.image_utils.doctree
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/hezar.utils.logging.doctree
Binary file not shown.
Binary file not shown.
Binary file added .doctrees/source/index.doctree
Binary file not shown.
Binary file added .doctrees/source/modules.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/datasets.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/embeddings.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/index.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/models.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/preprocessors.doctree
Binary file not shown.
Binary file added .doctrees/tutorial/training.doctree
Binary file not shown.
Empty file added .nojekyll
Empty file.
667 changes: 667 additions & 0 deletions _modules/hezar/builders.html

Large diffs are not rendered by default.

986 changes: 986 additions & 0 deletions _modules/hezar/configs.html

Large diffs are not rendered by default.

727 changes: 727 additions & 0 deletions _modules/hezar/constants.html

Large diffs are not rendered by default.

836 changes: 836 additions & 0 deletions _modules/hezar/data/data_collators.html

Large diffs are not rendered by default.

605 changes: 605 additions & 0 deletions _modules/hezar/data/datasets/dataset.html

Large diffs are not rendered by default.

617 changes: 617 additions & 0 deletions _modules/hezar/data/datasets/image_captioning_dataset.html

Large diffs are not rendered by default.

697 changes: 697 additions & 0 deletions _modules/hezar/data/datasets/ocr_dataset.html

Large diffs are not rendered by default.

676 changes: 676 additions & 0 deletions _modules/hezar/data/datasets/sequence_labeling_dataset.html

Large diffs are not rendered by default.

643 changes: 643 additions & 0 deletions _modules/hezar/data/datasets/text_classification_dataset.html

Large diffs are not rendered by default.

650 changes: 650 additions & 0 deletions _modules/hezar/data/datasets/text_summarization_dataset.html

Large diffs are not rendered by default.

900 changes: 900 additions & 0 deletions _modules/hezar/embeddings/embedding.html

Large diffs are not rendered by default.

755 changes: 755 additions & 0 deletions _modules/hezar/embeddings/fasttext.html

Large diffs are not rendered by default.

772 changes: 772 additions & 0 deletions _modules/hezar/embeddings/word2vec.html

Large diffs are not rendered by default.

580 changes: 580 additions & 0 deletions _modules/hezar/metrics/accuracy.html

Large diffs are not rendered by default.

580 changes: 580 additions & 0 deletions _modules/hezar/metrics/bleu.html

Large diffs are not rendered by default.

614 changes: 614 additions & 0 deletions _modules/hezar/metrics/cer.html

Large diffs are not rendered by default.

603 changes: 603 additions & 0 deletions _modules/hezar/metrics/f1.html

Large diffs are not rendered by default.

535 changes: 535 additions & 0 deletions _modules/hezar/metrics/metric.html

Large diffs are not rendered by default.

608 changes: 608 additions & 0 deletions _modules/hezar/metrics/precision.html

Large diffs are not rendered by default.

608 changes: 608 additions & 0 deletions _modules/hezar/metrics/recall.html

Large diffs are not rendered by default.

598 changes: 598 additions & 0 deletions _modules/hezar/metrics/rouge.html

Large diffs are not rendered by default.

619 changes: 619 additions & 0 deletions _modules/hezar/metrics/seqeval.html

Large diffs are not rendered by default.

592 changes: 592 additions & 0 deletions _modules/hezar/metrics/wer.html

Large diffs are not rendered by default.

588 changes: 588 additions & 0 deletions _modules/hezar/models/backbone/bert/bert.html

Large diffs are not rendered by default.

520 changes: 520 additions & 0 deletions _modules/hezar/models/backbone/bert/bert_config.html

Large diffs are not rendered by default.

575 changes: 575 additions & 0 deletions _modules/hezar/models/backbone/distilbert/distilbert.html

Large diffs are not rendered by default.

518 changes: 518 additions & 0 deletions _modules/hezar/models/backbone/distilbert/distilbert_config.html

Large diffs are not rendered by default.

587 changes: 587 additions & 0 deletions _modules/hezar/models/backbone/roberta/roberta.html

Large diffs are not rendered by default.

522 changes: 522 additions & 0 deletions _modules/hezar/models/backbone/roberta/roberta_config.html

Large diffs are not rendered by default.

570 changes: 570 additions & 0 deletions _modules/hezar/models/backbone/vit/vit.html

Large diffs are not rendered by default.

516 changes: 516 additions & 0 deletions _modules/hezar/models/backbone/vit/vit_config.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

534 changes: 534 additions & 0 deletions _modules/hezar/models/image2text/crnn/crnn_decode_utils.html

Large diffs are not rendered by default.

633 changes: 633 additions & 0 deletions _modules/hezar/models/image2text/crnn/crnn_image2text.html

Large diffs are not rendered by default.

513 changes: 513 additions & 0 deletions _modules/hezar/models/image2text/crnn/crnn_image2text_config.html

Large diffs are not rendered by default.

608 changes: 608 additions & 0 deletions _modules/hezar/models/image2text/trocr/trocr_image2text.html

Large diffs are not rendered by default.

570 changes: 570 additions & 0 deletions _modules/hezar/models/image2text/trocr/trocr_image2text_config.html

Large diffs are not rendered by default.

608 changes: 608 additions & 0 deletions _modules/hezar/models/image2text/vit_gpt2/vit_gpt2_image2text.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

1,062 changes: 1,062 additions & 0 deletions _modules/hezar/models/model.html

Large diffs are not rendered by default.

610 changes: 610 additions & 0 deletions _modules/hezar/models/model_outputs.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

1,236 changes: 1,236 additions & 0 deletions _modules/hezar/models/speech_recognition/whisper/whisper_tokenizer.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

Large diffs are not rendered by default.

604 changes: 604 additions & 0 deletions _modules/hezar/models/text_generation/gpt2/gpt2_text_generation.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

623 changes: 623 additions & 0 deletions _modules/hezar/models/text_generation/t5/t5_text_generation.html

Large diffs are not rendered by default.

Large diffs are not rendered by default.

841 changes: 841 additions & 0 deletions _modules/hezar/preprocessors/audio_feature_extractor.html

Large diffs are not rendered by default.

761 changes: 761 additions & 0 deletions _modules/hezar/preprocessors/image_processor.html

Large diffs are not rendered by default.

660 changes: 660 additions & 0 deletions _modules/hezar/preprocessors/preprocessor.html

Large diffs are not rendered by default.

655 changes: 655 additions & 0 deletions _modules/hezar/preprocessors/text_normalizer.html

Large diffs are not rendered by default.

616 changes: 616 additions & 0 deletions _modules/hezar/preprocessors/tokenizers/bpe.html

Large diffs are not rendered by default.

621 changes: 621 additions & 0 deletions _modules/hezar/preprocessors/tokenizers/sentencepiece_bpe.html

Large diffs are not rendered by default.

619 changes: 619 additions & 0 deletions _modules/hezar/preprocessors/tokenizers/sentencepiece_unigram.html

Large diffs are not rendered by default.

1,338 changes: 1,338 additions & 0 deletions _modules/hezar/preprocessors/tokenizers/tokenizer.html

Large diffs are not rendered by default.

604 changes: 604 additions & 0 deletions _modules/hezar/preprocessors/tokenizers/wordpiece.html

Large diffs are not rendered by default.

699 changes: 699 additions & 0 deletions _modules/hezar/registry.html

Large diffs are not rendered by default.

715 changes: 715 additions & 0 deletions _modules/hezar/trainer/metrics_handlers.html

Large diffs are not rendered by default.

1,282 changes: 1,282 additions & 0 deletions _modules/hezar/trainer/trainer.html

Large diffs are not rendered by default.

702 changes: 702 additions & 0 deletions _modules/hezar/trainer/trainer_utils.html

Large diffs are not rendered by default.

1,066 changes: 1,066 additions & 0 deletions _modules/hezar/utils/audio_utils.html

Large diffs are not rendered by default.

652 changes: 652 additions & 0 deletions _modules/hezar/utils/common_utils.html

Large diffs are not rendered by default.

724 changes: 724 additions & 0 deletions _modules/hezar/utils/data_utils.html

Large diffs are not rendered by default.

552 changes: 552 additions & 0 deletions _modules/hezar/utils/file_utils.html

Large diffs are not rendered by default.

672 changes: 672 additions & 0 deletions _modules/hezar/utils/hub_utils.html

Large diffs are not rendered by default.

757 changes: 757 additions & 0 deletions _modules/hezar/utils/image_utils.html

Large diffs are not rendered by default.

556 changes: 556 additions & 0 deletions _modules/hezar/utils/integration_utils.html

Large diffs are not rendered by default.

518 changes: 518 additions & 0 deletions _modules/hezar/utils/logging.html

Large diffs are not rendered by default.

679 changes: 679 additions & 0 deletions _modules/hezar/utils/registry_utils.html

Large diffs are not rendered by default.

577 changes: 577 additions & 0 deletions _modules/index.html

Large diffs are not rendered by default.

102 changes: 102 additions & 0 deletions _sources/contributing.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,102 @@
# Contributing to Hezar
Welcome to Hezar! We greatly appreciate your interest in contributing to this project and helping us make it even more
valuable to the Persian community. Whether you're a developer, researcher, or enthusiast, your contributions are
invaluable in helping us grow and improve Hezar.

Before you start contributing, please take a moment to review the following guidelines.

## Code of Conduct

This project and its community adhere to
the [Contributor Code of Conduct](https://github.com/hezarai/hezar/blob/main/CODE_OF_CONDUCT.md).

## How to Contribute

### Reporting Bugs

If you come across a bug or unexpected behavior, please help us by reporting it.
Use the [GitHub Issue Tracker](https://github.com/hezarai/hezar/issues) to create a detailed bug report.
Include information such as:

- A clear and descriptive title.
- Steps to reproduce the bug.
- Expected behavior.
- Actual behavior.
- Your operating system and Python version.

### Adding features

Have a great idea for a new feature or improvement? We'd love to hear it. You can open an issue and add your suggestion
with a clear description and further suggestions on how it can be implemented. Also, if you already can implement it
yourself, just follow the instructions on how you can send a PR.

### Adding/Improving documents

Have a suggestion to enhance our documentation or want to contribute entirely new sections? We welcome your input!<br>
Here's how you can get involved:<br>
Docs website is deployed here: [https://hezarai.github.io/hezar](https://hezarai.github.io/hezar) and the source for the
docs are located at the [docs](https://github.com/hezarai/hezar/tree/main/docs) folder in the root of the repo. Feel
free to apply your changes or add new docs to this section. Notice that docs are written in Markdown format. In case you have
added new files to this section, you must include them in the `index.md` file in the same folder. For example, if you've
added the file `new_doc.md` to the `get_started` folder, you have to modify `get_started/index.md` and put your file
name there.

### Commit guidelines

#### Functional best practices

- Ensure only one "logical change" per commit for efficient review and flaw identification.
- Smaller code changes facilitate quicker reviews and easier troubleshooting using Git's bisect capability.
- Avoid mixing whitespace changes with functional code changes.
- Avoid mixing two unrelated functional changes.
- Refrain from sending large new features in a single giant commit.

#### Styling best practices

- Use imperative mood in the subject (e.g., "Add support for ..." not "Adding support or added support") .
- Keep the subject line short and concise, preferably less than 50 characters.
- Capitalize the subject line and do not end it with a period.
- Wrap body lines at 72 characters.
- Use the body to explain what and why a change was made.
- Do not explain the "how" in the commit message; reserve it for documentation or code.
- For commits referencing an issue or pull request, write the proper commit subject followed by the reference in
parentheses (e.g., "Add NFKC normalizer (#9999)").
- Reference codes & paths in back quotes (e.g., `variable`, `method()`, `Class()`, `file.py`).
- Preferably use the following [gitmoji](https://gitmoji.dev/) compatible codes at the beginning of your commit message:

| Emoji Code | Emoji | Description | Example Commit |
|----------------------|-------|----------------------------------------------|----------------------------------------------------------------|
| `:bug:` | 🐛 | Fix a bug or issue | `:bug: Fix issue with image loading in DataLoader` |
| `:sparkles:` | ✨ | Add feature or improvements | `:sparkles: Introduce support for text summarization` |
| `:recycle:` | ♻️ | Refactor code (backward compatible refactor) | `:recycle: Refactor data preprocessing utilities` |
| `:memo:` | 📝 | Add or change docs | `:memo: Update documentation for text classification` |
| `:pencil2:` | ✏️ | Minor change or improvement | `:pencil2: Improve logging in Trainer` |
| `:fire:` | 🔥 | Remove code or file | `:fire: Remove outdated utility function` |
| `:boom:` | 💥 | Introduce breaking changes | `:boom: Update API, requires modification in existing scripts` |
| `:test_tube:` | 🧪 | Test-related changes | `:test_tube: Add unit tests for data loading functions` |
| `:bookmark:` | 🔖 | Version release | `:bookmark: Release v1.0.0` |
| `:adhesive_bandage:` | 🩹 | Non-critical fix | `:adhesive_bandage: Fix minor issue in BPE tokenizer` |

## Sending a PR

In order to apply any change to the repo, you have to follow these step:

1. Fork the Hezar repository.
2. Create a new branch for your feature, bug fix, etc.
3. Make your changes.
4. Update the documentation to reflect your changes.
5. Ensure your code adheres to the [Google Python Style Guide](https://google.github.io/styleguide/pyguide.html).
6. Format the code using `ruff` (`ruff check --fix .`)
7. Write tests to ensure the functionality if needed.
8. Run tests and make sure all of them pass. (Skip this step if your changes do not involve codes)
9. Open a pull request from your fork and the PR template will be automatically loaded to help you do the rest.
10. Be responsive to feedback and comments during the review process.
11. Thanks for contributing to the Hezar project.😉❤️

## License

By contributing to Hezar, you agree that your contributions will be licensed under
the [Apache 2.0 License](https://github.com/hezarai/hezar/blob/main/LICENSE).

We look forward to your contributions and appreciate your efforts in making Hezar a powerful AI tool for the Persian
community!
8 changes: 8 additions & 0 deletions _sources/get_started/index.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# Get Started
```{toctree}
:maxdepth: 1

overview.md
installation.md
quick_tour.md
```
41 changes: 41 additions & 0 deletions _sources/get_started/installation.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
# Installation

## Install from PyPi
Installing Hezar is as easy as any other Python library! Most of the requirements are cross-platform and installing
them on any machine is a piece of cake!

```
pip install hezar
```
### Installation variations
Hezar is packed with a lot of tools that are dependent on other packages. Most of the
time you might not want everything to be installed, hence, providing multiple variations of
Hezar so that the installation is light and fast for general use.

You can install optional dependencies for each mode like so:
```
pip install hezar[nlp] # For natural language processing
pip install hezar[vision] # For computer vision and image processing
pip install hezar[audio] # For audio and speech processing
pip install hezar[embeddings] # For word embeddings
```
Or you can also install everything using:
```
pip install hezar[all]
```
## Install from source
Also, you can install the dev version of the library using the source:
```
pip install git+https://github.com/hezarai/hezar.git
```

## Test installation
From a Python console or in CLI just import `hezar` and check the version:
```python
import hezar

print(hezar.__version__)
```
```
0.23.1
```
20 changes: 20 additions & 0 deletions _sources/get_started/overview.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
# Overview

Welcome to Hezar! A library that makes state-of-the-art machine learning as easy as possible aimed for the Persian
language, built by the Persian community!

In Hezar, the primary goal is to provide plug-and-play AI/ML utilities so that you don't need to know much about what's
going on under the hood. Hezar is not just a model library, but instead it's packed with every aspect you need for any
ML pipeline like datasets, trainers, preprocessors, feature extractors, etc.

Hezar is a library that:
- brings together all the best works in AI for Persian
- makes using AI models as easy as a couple of lines of code
- seamlessly integrates with Hugging Face Hub for all of its models
- has a highly developer-friendly interface
- has a task-based model interface which is more convenient for general users.
- is packed with additional tools like word embeddings, tokenizers, feature extractors, etc.
- comes with a lot of supplementary ML tools for deployment, benchmarking, optimization, etc.
- and more!

To find out more, just take the [quick tour](quick_tour.md)!
214 changes: 214 additions & 0 deletions _sources/get_started/quick_tour.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,214 @@
# Quick Tour
## Models
There's a bunch of ready to use trained models for different tasks on the Hub!

**🤗Hugging Face Hub Page**: [https://huggingface.co/hezarai](https://huggingface.co/hezarai)

Let's walk you through some examples!

- **Text Classification (sentiment analysis, categorization, etc)**
```python
from hezar.models import Model

example = ["هزار، کتابخانه‌ای کامل برای به کارگیری آسان هوش مصنوعی"]
model = Model.load("hezarai/bert-fa-sentiment-dksf")
outputs = model.predict(example)
print(outputs)
```
```
[[{'label': 'positive', 'score': 0.812910258769989}]]
```
- **Sequence Labeling (POS, NER, etc.)**
```python
from hezar.models import Model

pos_model = Model.load("hezarai/bert-fa-pos-lscp-500k") # Part-of-speech
ner_model = Model.load("hezarai/bert-fa-ner-arman") # Named entity recognition
inputs = ["شرکت هوش مصنوعی هزار"]
pos_outputs = pos_model.predict(inputs)
ner_outputs = ner_model.predict(inputs)
print(f"POS: {pos_outputs}")
print(f"NER: {ner_outputs}")
```
```
POS: [[{'token': 'شرکت', 'label': 'Ne'}, {'token': 'هوش', 'label': 'Ne'}, {'token': 'مصنوعی', 'label': 'AJe'}, {'token': 'هزار', 'label': 'NUM'}]]
NER: [[{'token': 'شرکت', 'label': 'B-org'}, {'token': 'هوش', 'label': 'I-org'}, {'token': 'مصنوعی', 'label': 'I-org'}, {'token': 'هزار', 'label': 'I-org'}]]
```
- **Language Modeling (Mask Filling)**
```python
from hezar.models import Model

roberta_mask_filling = Model.load("hezarai/roberta-fa-mask-filling")
inputs = ["سلام بچه ها حالتون <mask>"]
outputs = roberta_mask_filling.predict(inputs, top_k=1)
print(outputs)
```
```
[[{'token': 'چطوره', 'sequence': 'سلام بچه ها حالتون چطوره', 'token_id': 34505, 'score': 0.2230483442544937}]]
```
- **Speech Recognition**
```python
from hezar.models import Model

whisper = Model.load("hezarai/whisper-small-fa")
transcripts = whisper.predict("examples/assets/speech_example.mp3")
print(transcripts)
```
```
[{'text': 'و این تنها محدود به محیط کار نیست'}]
```
- **Image to Text (OCR)**
```python
from hezar.models import Model
# OCR with TrOCR
model = Model.load("hezarai/trocr-base-fa-v2")
texts = model.predict(["examples/assets/ocr_example.jpg"])
print(f"TrOCR Output: {texts}")

# OCR with CRNN
model = Model.load("hezarai/crnn-fa-printed-96-long")
texts = model.predict("examples/assets/ocr_example.jpg")
print(f"CRNN Output: {texts}")
```
```
TrOCR Output: [{'text': 'چه میشه کرد، باید صبر کنیم'}]
CRNN Output: [{'text': 'چه میشه کرد، باید صبر کنیم'}]
```
![](https://raw.githubusercontent.com/hezarai/hezar/main/examples/assets/ocr_example.jpg)

- **Image to Text (License Plate Recognition)**
```python
from hezar.models import Model

model = Model.load("hezarai/crnn-fa-64x256-license-plate-recognition")
plate_text = model.predict("assets/license_plate_ocr_example.jpg")
print(plate_text) # Persian text of mixed numbers and characters might not show correctly in the console
```
```
[{'text': '۵۷س۷۷۹۷۷'}]
```
![](https://raw.githubusercontent.com/hezarai/hezar/main/examples/assets/license_plate_ocr_example.jpg)

- **Image to Text (Image Captioning)**
```python
from hezar.models import Model

model = Model.load("hezarai/vit-roberta-fa-image-captioning-flickr30k")
texts = model.predict("examples/assets/image_captioning_example.jpg")
print(texts)
```
```
[{'text': 'سگی با توپ تنیس در دهانش می دود.'}]
```
![](https://raw.githubusercontent.com/hezarai/hezar/main/examples/assets/image_captioning_example.jpg)

We constantly keep working on adding and training new models and this section will hopefully be expanding over time ;)
## Word Embeddings
- **FastText**
```python
from hezar.embeddings import Embedding

fasttext = Embedding.load("hezarai/fasttext-fa-300")
most_similar = fasttext.most_similar("هزار")
print(most_similar)
```
```
[{'score': 0.7579, 'word': 'میلیون'},
{'score': 0.6943, 'word': '21هزار'},
{'score': 0.6861, 'word': 'میلیارد'},
{'score': 0.6825, 'word': '26هزار'},
{'score': 0.6803, 'word': '٣هزار'}]
```
- **Word2Vec (Skip-gram)**
```python
from hezar.embeddings import Embedding

word2vec = Embedding.load("hezarai/word2vec-skipgram-fa-wikipedia")
most_similar = word2vec.most_similar("هزار")
print(most_similar)
```
```
[{'score': 0.7885, 'word': 'چهارهزار'},
{'score': 0.7788, 'word': '۱۰هزار'},
{'score': 0.7727, 'word': 'دویست'},
{'score': 0.7679, 'word': 'میلیون'},
{'score': 0.7602, 'word': 'پانصد'}]
```
- **Word2Vec (CBOW)**
```python
from hezar.embeddings import Embedding

word2vec = Embedding.load("hezarai/word2vec-cbow-fa-wikipedia")
most_similar = word2vec.most_similar("هزار")
print(most_similar)
```
```
[{'score': 0.7407, 'word': 'دویست'},
{'score': 0.7400, 'word': 'میلیون'},
{'score': 0.7326, 'word': 'صد'},
{'score': 0.7276, 'word': 'پانصد'},
{'score': 0.7011, 'word': 'سیصد'}]
```
For a full guide on the embeddings module, see the [embeddings tutorial](https://hezarai.github.io/hezar/tutorial/embeddings.html).
## Datasets
You can load any of the datasets on the [Hub](https://huggingface.co/hezarai) like below:
```python
from hezar.data import Dataset

sentiment_dataset = Dataset.load("hezarai/sentiment-dksf") # A TextClassificationDataset instance
lscp_dataset = Dataset.load("hezarai/lscp-pos-500k") # A SequenceLabelingDataset instance
xlsum_dataset = Dataset.load("hezarai/xlsum-fa") # A TextSummarizationDataset instance
alpr_ocr_dataset = Dataset.load("hezarai/persian-license-plate-v1") # An OCRDataset instance
...
```
The returned dataset objects from `load()` are PyTorch Dataset wrappers for specific tasks and can be used by a data loader out-of-the-box!

You can also load Hezar's datasets using 🤗Datasets:
```python
from datasets import load_dataset

dataset = load_dataset("hezarai/sentiment-dksf")
```
For a full guide on Hezar's datasets, see the [datasets tutorial](https://hezarai.github.io/hezar/tutorial/datasets.html).
## Training
Hezar makes it super easy to train models using out-of-the-box models and datasets provided in the library.

```python
from hezar.models import BertSequenceLabeling, BertSequenceLabelingConfig
from hezar.data import Dataset
from hezar.trainer import Trainer, TrainerConfig
from hezar.preprocessors import Preprocessor

base_model_path = "hezarai/bert-base-fa"
dataset_path = "hezarai/lscp-pos-500k"

train_dataset = Dataset.load(dataset_path, split="train", tokenizer_path=base_model_path)
eval_dataset = Dataset.load(dataset_path, split="test", tokenizer_path=base_model_path)

model = BertSequenceLabeling(BertSequenceLabelingConfig(id2label=train_dataset.config.id2label))
preprocessor = Preprocessor.load(base_model_path)

train_config = TrainerConfig(
output_dir="bert-fa-pos-lscp-500k",
task="sequence_labeling",
device="cuda",
init_weights_from=base_model_path,
batch_size=8,
num_epochs=5,
metrics=["seqeval"],
)

trainer = Trainer(
config=train_config,
model=model,
train_dataset=train_dataset,
eval_dataset=eval_dataset,
data_collator=train_dataset.data_collator,
preprocessor=preprocessor,
)
trainer.train()

trainer.push_to_hub("bert-fa-pos-lscp-500k") # push model, config, preprocessor, trainer files and configs
```

Want to go deeper? Check out the [guides](../guide/index.md).
2 changes: 2 additions & 0 deletions _sources/guide/advanced_training.md.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
# Advanced Training
Docs coming soon, stay tuned!
Loading

0 comments on commit 5f2d8f7

Please sign in to comment.