task_categories:token-classification language:fa region:us
"""

_DOWNLOAD_URLS = {
    "train": "https://huggingface.co/datasets/hezarai/arman-ner/resolve/main/arman-ner_train.csv",
    "test": "https://huggingface.co/datasets/hezarai/arman-ner/resolve/main/arman-ner_test.csv",
}


class ArmanNERConfig(datasets.BuilderConfig):
    def __init__(self, **kwargs):
        super(ArmanNERConfig, self).__init__(**kwargs)


class ArmanNER(datasets.GeneratorBasedBuilder):
    BUILDER_CONFIGS = [
        ArmanNERConfig(
            name="Arman-NER",
            version=datasets.Version("1.0.0"),
            description=_DESCRIPTION,
        ),
    ]

    def _info(self):
        return datasets.DatasetInfo(
            description=_DESCRIPTION,
            features=datasets.Features(
                {
                    "tokens": datasets.Sequence(datasets.Value("string")),
                    "ner_tags": datasets.Sequence(
                        datasets.features.ClassLabel(
                            names=[
                                "O",
                                "B-pro",
                                "I-pro",
                                "B-pers",
                                "I-pers",
                                "B-org",
                                "I-org",
                                "B-loc",
                                "I-loc",
                                "B-fac",
                                "I-fac",
                                "B-event",
                                "I-event"
                            ]
                        )
                    ),
                }
            ),
            homepage="https://huggingface.co/datasets/hezarai/arman-ner",
            citation=_CITATION,
        )

    def _split_generators(self, dl_manager):
@inproceedings{poostchi-etal-2018-bilstm,
    title = "{B}i{LSTM}-{CRF} for {P}ersian Named-Entity Recognition {A}rman{P}erso{NERC}orpus: the First Entity-Annotated {P}ersian Dataset",
    author = "Poostchi, Hanieh  and
      Zare Borzeshi, Ehsan  and
      Piccardi, Massimo",
    booktitle = "Proceedings of the Eleventh International Conference on Language Resources and Evaluation ({LREC} 2018)",
    month = may,
    year = "2018",
    address = "Miyazaki, Japan",
    publisher = "European Language Resources Association (ELRA)",
    url = "https://aclanthology.org/L18-1701",
}
---
task_categories:
- token-classification
language:
- fa
pretty_name: ARMAN-NER
---