Named Entity Recognition

from transformers import pipeline

# NER tiếng Việt với PhoBERT
ner = pipeline("token-classification",
               model="NlpHUST/ner-vietnamese-electra-base",
               aggregation_strategy="simple")

text = "Nguyễn Quang Hải ghi bàn cho CLB Hà Nội tại Mỹ Đình"
entities = ner(text)

for ent in entities:
    print(f"  {ent['word']:>20} → {ent['entity_group']} "
          f"({ent['score']:.2%})")
# Nguyễn Quang Hải → PER (98.5%)
#        CLB Hà Nội → ORG (96.2%)
#          Mỹ Đình → LOC (97.8%)

# Ứng dụng: tự động trích xuất từ tin tức VnExpress
# → Xây knowledge graph: Quang Hải --chơi cho--> CLB Hà Nội
#                         CLB Hà Nội --sân nhà--> Mỹ Đình