unimelb-nlp/wikiann
Viewer • Updated • 2M • 45.2k • 121
How to use lakshaywadhwa1993/ner_marathi_bert with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="lakshaywadhwa1993/ner_marathi_bert") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("lakshaywadhwa1993/ner_marathi_bert")
model = AutoModelForTokenClassification.from_pretrained("lakshaywadhwa1993/ner_marathi_bert")This model is a fine-tuned version of bert-base-multilingual-cased on the wikiann dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Overall Precision | Overall Recall | Overall F1 | Overall Accuracy | Loc F1 | Org F1 | Per F1 |
|---|---|---|---|---|---|---|---|---|---|---|
| 0.2961 | 3.19 | 1000 | 0.3496 | 0.8720 | 0.8841 | 0.8780 | 0.9229 | 0.8599 | 0.8210 | 0.9343 |
| 0.0613 | 6.39 | 2000 | 0.3606 | 0.8939 | 0.9030 | 0.8984 | 0.9347 | 0.8823 | 0.8555 | 0.9435 |