Downloads · 30 days
34.2K
0% of all-time downloads
fahmisyaifudin/indobert_ner_p1
indobert_ner_p1 is a machine learning model from fahmisyaifudin. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This repository contains a fine-tuned IndoBERT model for Indonesian Named Entity Recognition, wrapped in a HuggingFace-compatible pipeline for easy usage.
Downloads · 30 days
34.2K
0% of all-time downloads
All-time downloads
8.5M
Public
Parameters
124M
1.5 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors495 MB · 100%
From the Hugging Face model README
This repository contains a fine-tuned IndoBERT model for Indonesian Named Entity Recognition, wrapped in a HuggingFace-compatible pipeline for easy usage.
from transformers import pipeline
ner = pipeline(
"ner",
model="fahmisyaifudin/indobert_ner_p1",
tokenizer="fahmisyaifudin/indobert_ner_p1",
aggregation_strategy="simple"
)
text = "Hadiri KTT G20, Momen Gibran Bersama Erdogan hingga PM Jepang"
entities = ner(text)
print(entities)
# Output: [{'entity_group': 'EVENT', 'score': np.float32(0.9938912), 'word': 'ktt g20', 'start': 7, 'end': 14}, {'entity_group': 'PERSON', 'score': np.float32(0.80416846), 'word': 'gibran', 'start': 22, 'end': 28}, {'entity_group': 'PERSON', 'score': np.float32(0.99820226), 'word': 'erdogan', 'start': 37, 'end': 44}, {'entity_group': 'REPRESENTATIVE_PERSON', 'score': np.float32(0.9186572), 'word': 'pm jepang', 'start': 52, 'end': 61}]