Bart Large Finetuned Squad2

🚀 roberta-large-finetuned-squad2

このモデルは質問応答タスクを解決するためのもので、特定のデータセットで微調整され、高い性能を発揮します。

🚀 クイックスタート

このセクションでは、モデルの概要、使用方法、トレーニング手順、評価結果について説明します。

✨ 主な機能

• タグ: pytorch, question-answering
• データセット: squad2
• 評価指標: exact, f1

対応する論文は こちら (モデル) と こちら (データ) で見ることができます。

📦 インストール

このモデルを使用するには、transformers ライブラリが必要です。以下のコマンドでインストールできます。

pip install transformers

💻 使用例

基本的な使用法

from transformers.pipelines import pipeline

model_name = "phiyodr/bart-large-finetuned-squad2"
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
inputs = {
    'question': 'What discipline did Winkelmann create?',
    'context': 'Johann Joachim Winckelmann was a German art historian and archaeologist. He was a pioneering Hellenist who first articulated the difference between Greek, Greco-Roman and Roman art. "The prophet and founding hero of modern archaeology", Winckelmann was one of the founders of scientific archaeology and first applied the categories of style on a large, systematic basis to the history of art. '
}
nlp(inputs)

🔧 技術詳細

トレーニング手順

{
    "base_model": "facebook/bart-large",
    "do_lower_case": True,
    "learning_rate": 3e-5,
    "num_train_epochs": 4,
    "max_seq_length": 384,
    "doc_stride": 128,
    "max_query_length": 64,
    "batch_size": 96 
}

評価結果

• データ: dev-v2.0.json
• スクリプト: evaluate-v2.0.py (元のスクリプトは こちら から)

{
  "exact": 81.96748926134929,
  "f1": 85.93825235371045,
  "total": 11873,
  "HasAns_exact": 78.71120107962213,
  "HasAns_f1": 86.6641144054667,
  "HasAns_total": 5928,
  "NoAns_exact": 85.21446593776282,
  "NoAns_f1": 85.21446593776282,
  "NoAns_total": 5945
}