Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

docs: Resolve typos, missing import #3443

Merged
merged 1 commit into from
Jul 26, 2023
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions docs/_source/guides/train_a_model.md
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ Options:

```python
import argilla as rg
from argilla.training import ArgillaTrainer
from datasets import load_dataset

dataset_rg = rg.DatasetForTokenClassification.from_datasets(
Expand Down Expand Up @@ -126,18 +127,18 @@ It is possible to directly include train-test splits to the `prepare_for_trainin
*TextClassification*

For text classification tasks, it flattens the inputs into separate columns of the returned dataset and converts the annotations of your records into integers and writes them in a label column:
By passing the `framework` variable as `setfit`, `transformers`, `spark-nlp` or `spacy`. This task requires a `DatastForTextClassification`.
By passing the `framework` variable as `setfit`, `transformers`, `spark-nlp` or `spacy`. This task requires a `DatasetForTextClassification`.


*TokenClassification*

For token classification tasks, it converts the annotations of a record into integers representing BIO tags and writes them in a `ner_tags` column:
By passing the `framework` variable as `transformers`, `spark-nlp` or `spacy`. This task requires a `DatastForTokenClassification`.
By passing the `framework` variable as `transformers`, `spark-nlp` or `spacy`. This task requires a `DatasetForTokenClassification`.

*Text2Text*

For text generation tasks like `summarization` and translation tasks, it converts the annotations of a record `text` and `target` columns.
By passing the `framework` variable as `transformers` and `spark-nlp`. This task requires a `DatastForText2Text`.
By passing the `framework` variable as `transformers` and `spark-nlp`. This task requires a `DatasetForText2Text`.

*Feedback*
For feedback-oriented datasets, we currently rely on a fully customizable workflow, which means automation is limited and yet to be thought out.
Expand Down
Loading