Views
No views yet
1from transformers import AutoTokenizer,pipeline, AutoModelForTokenClassification
2tokenizer = AutoTokenizer.from_pretrained('Mizuiro-sakura/deberta-v2-base-japanese-finetuned-ner')
3model=AutoModelForTokenClassification.from_pretrained('Mizuiro-sakura/deberta-v2-base-japanese-finetuned-ner') # 学習済みモデルの読み込み
4
5text=('昨日は東京で買い物をした')
6
7ner=pipeline('ner', model=model, tokenizer=tokenizer)
8
9result=ner(text)
10print(result) precision recall f1-score support
その他の組織名 0.73 0.75 0.74 238
イベント名 0.81 0.81 0.81 215
人名 0.84 0.87 0.85 547
地名 0.83 0.83 0.83 446
政治的組織名 0.82 0.85 0.83 263
施設名 0.74 0.86 0.80 241
法人名 0.81 0.82 0.82 487
製品名 0.68 0.73 0.71 252
micro avg 0.79 0.82 0.81 2689
macro avg 0.78 0.81 0.80 2689
weighted avg 0.79 0.82 0.81 2689