Closes #270 | Create dataset loader for NERGrit

cahya-wirawan commented 1 year ago

Please name your PR after the issue it closes. You can use the following line: "Closes #ISSUE-NUMBER" where you replace the ISSUE-NUMBER with the one corresponding to your dataset.

Checkbox

[x] Confirm that this PR is linked to the dataset issue.
[x] Create the dataloader script nusantara/nusa_datasets/my_dataset/my_dataset.py (please use only lowercase and underscore for dataset naming).
[x] Provide values for the _CITATION, _DATASETNAME, _DESCRIPTION, _HOMEPAGE, _LICENSE, _URLs, _SUPPORTED_TASKS, _SOURCE_VERSION, and _NUSANTARA_VERSION variables.
[x] Implement _info(), _split_generators() and _generate_examples() in dataloader script.
[x] Make sure that the BUILDER_CONFIGS class attribute is a list with at least one NusantaraConfig for the source schema and one for a nusantara schema.
[x] Confirm dataloader script works with datasets.load_dataset function.
[x] Confirm that your dataloader script passes the test suite run with python -m tests.test_nusantara --path=nusantara/nusa_datasets/my_dataset/my_dataset.py.
[] If my dataset is local, I have provided an output of the unit-tests in the PR (please copy paste). This is OPTIONAL for public datasets, as we can test these without access to the data files.

cahya-wirawan commented 1 year ago

This nergrit dataset is more comprehensive than the indonlu_nergrit dataset. It also contains three subset datasets NER, Sentiment, and Statement. Following is the comparison:

indonlu_nergrit:

It has only one subset dataset (NER)
NER dataset has three entities (Person, Organization, and Place)
NER dataset has 1672:209:209 (train:dev:test) entries.

nergrit:

it has three subset datasets (NER, Sentiment, and Statement)
NER dataset has 18 entities (Person, Organization, Location, Date, Geopolitical Entity,...)
NER dataset has 12551:2526:2402 (train:dev:test) entries.
Sentiment dataset has 7458:780:2314 (train:dev:test) entries.
Statement dataset has 2413:175:334 (train:dev:test) entries.