Roberta Large Squad2

🚀 Roberta large model for QA (SQuAD 2.0)

このモデルは質問応答タスクに特化しており、roberta-large をベースに構築されています。SQuAD 2.0データセットで訓練され、高精度な質問応答を実現します。

🚀 クイックスタート

このモデルは質問応答タスクに使用できます。以下に使用方法を示します。

💻 使用例

基本的な使用法

from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

# Load model & tokenizer
roberta_model = AutoModelForQuestionAnswering.from_pretrained('navteca/roberta-large-squad2')
roberta_tokenizer = AutoTokenizer.from_pretrained('navteca/roberta-large-squad2')

# Get predictions
nlp = pipeline('question-answering', model=roberta_model, tokenizer=roberta_tokenizer)

result = nlp({
    'question': 'How many people live in Berlin?',
    'context': 'Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.'
})

print(result)

#{
#  "answer": "3,520,031"
#  "end": 36,
#  "score": 0.96186668,
#  "start": 27,
#}

📚 ドキュメント

データセット

ライセンス

このモデルはMITライセンスの下で提供されています。