grit-id/id_nergrit_corpus
Updated • 191 • 7
How to use aadhistii/IndoBERT-NER with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="aadhistii/IndoBERT-NER") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("aadhistii/IndoBERT-NER")
model = AutoModelForTokenClassification.from_pretrained("aadhistii/IndoBERT-NER")This model is a fine-tuned version of indolem/indobert-base-uncased on dataset id_nergrit_corpus. It achieves the following results on the evaluation set:
Dataset Entities:
More information needed
More information needed
The following hyperparameters were used during training:
| Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
|---|---|---|---|---|---|---|
| 0.5182 | 0.2042 | 0.7770 | 0.8146 | 0.7954 | 0.9395 | 0 |
| 0.1907 | 0.1810 | 0.8020 | 0.8344 | 0.8179 | 0.9469 | 1 |
| 0.1471 | 0.1801 | 0.8077 | 0.8437 | 0.8253 | 0.9471 | 2 |
Base model
indolem/indobert-base-uncased