Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForQuestionAnswering
3
4tokenizer = AutoTokenizer.from_pretrained('Mizuiro-sakura/deberta-v2-tiny-japanese-finetuned-QA')
5model=AutoModelForQuestionAnswering.from_pretrained('Mizuiro-sakura/deberta-v2-tiny-japanese-finetuned-QA') # 学習済みモデルの読み込み
6
7text={
8 'context':'私の名前はEIMIです。好きな食べ物は苺です。 趣味は皆さんと会話することです。',
9 'question' :'好きな食べ物は何ですか'
10}
11
12input_ids=tokenizer.encode(text['question'],text['context']) # tokenizerで形態素解析しつつコードに変換する
13output= model(torch.tensor([input_ids])) # 学習済みモデルを用いて解析
14prediction = tokenizer.decode(input_ids[torch.argmax(output.start_logits): torch.argmax(output.end_logits)]) # 答えに該当する部分を抜き取る
15print(prediction)