Views
No views yet
1from transformers import pipeline
2from pprint import pprint
3
4ner_pipeline = pipeline(
5 model="llm-book/bert-base-japanese-v3-ner-wikipedia-dataset",
6 aggregation_strategy="simple",
7)
8text = "大谷翔平は岩手県水沢市出身のプロ野球選手"
9# text中の固有表現を抽出
10pprint(ner_pipeline(text))
11# [{'end': None,
12# 'entity_group': '人名',
13# 'score': 0.99823624,
14# 'start': None,
15# 'word': '大谷 翔平'},
16# {'end': None,
17# 'entity_group': '地名',
18# 'score': 0.9986874,
19# 'start': None,
20# 'word': '岩手 県 水沢 市'}]