Electra Base De Squad2

🚀 德國問答模型

我們發佈了基於德國問答數據集（deQuAD）微調的德國問答模型，該數據集包含 130k 個訓練問答對和 11k 個測試問答對。

🚀 快速開始

我們基於自己的德國問答數據集（deQuAD）對德國問答模型進行了微調。該數據集包含 130k 個訓練問答對和 11k 個測試問答對。

✨ 主要特性

• 基於 electra-base-german-uncased 語言模型。
• 支持德語問答任務。
• 使用 deQuAD2.0 訓練集（約 42MB）進行訓練，deQuAD2.0 測試集（約 4MB）進行評估。
• 在 8xV100 GPU 基礎設施上進行訓練。

📚 詳細文檔

評估

我們在 deQuAD 測試數據上對一些德語語言模型的問答性能進行了基準測試。微調後的 electra-base-german-uncased 模型在精確匹配（Exact Match）和 F1 分數上表現最佳。

💻 使用示例

基礎用法

from transformers import pipeline

qa_pipeline = pipeline(
    "question-answering",
    model="deutsche-telekom/electra-base-de-squad2",
    tokenizer="deutsche-telekom/electra-base-de-squad2"
)

contexts = ['''Die Robert Bosch GmbH ist ein im Jahr 1886 von Robert Bosch gegründetes multinationales deutsches Unternehmen. 
Es ist tätig als Automobilzulieferer, Hersteller von Gebrauchsgütern und Industrie- und Gebäudetechnik und darüber hinaus 
in der automatisierten Verpackungstechnik, wo Bosch den führenden Platz einnimmt. Die Robert Bosch GmbH und ihre rund 460 
Tochter- und Regionalgesellschaften in mehr als 60 Ländern bilden die Bosch-Gruppe. Der Sitz der Geschäftsführung befindet 
sich auf der Schillerhöhe in Gerlingen, der Firmensitz in Stuttgart. Seit dem 1. Juli 2012 ist Volkmar Denner Vorsitzender 
der Geschäftsführung. Im Jahr 2015 konnte Bosch die Spitzenposition zurückgewinnen. Die Automobilsparte war im Jahr 2018 
für 61 % des Konzernumsatzes von Bosch verantwortlich. Das Unternehmen hatte im Jahr 2018 in Deutschland an 85 Standorten 
139.400 Mitarbeiter.''']*2

questions = ["Wer leitet die Robert Bosch GmbH?", 
            "Wer begründete die Robert Bosch GmbH?"]

qa_pipeline(context=contexts, question=questions)

輸出示例

[{'score': 0.9537325501441956,
  'start': 577,
  'end': 591,
  'answer': 'Volkmar Denner'},
 {'score': 0.8804352879524231,
  'start': 47,
  'end': 59,
  'answer': 'Robert Bosch'}]

📄 許可證

麻省理工學院許可證（MIT License）

版權所有 (c) 2021 Fang Xu, Deutsche Telekom AG