Sahajtomar's picture
Update README.md
429a906
metadata
language: de
tags:
  - pytorch
  - tf
  - Gelectra
datasets:
  - mlqa
metrics:
  - f1
  - em

QA Model trained on MLQA dataset for german langauge.

MODEL used for fine tuning is GELECTRA Large by deepset.ai

MLQA DEV (german)

EM: 64.27
F1: 77.39

XQUAD TEST (german)

EM: 66.38
F1: 82.25

Hyperparameters

per_gpu_train_batch_size 4
per_gpu_eval_batch_size 32
gradient_accumulation_steps 8
learning_rate 3e-5
num_train_epochs 1.0
max_seq_length 384
doc_stride 128

Model inferencing:

!pip install -q transformers
from transformers import pipeline

qa_pipeline = pipeline(
    "question-answering",
    model="Sahajtomar/GELECTRAQA",
    tokenizer="Sahajtomar/GELECTRAQA"
)

qa_pipeline({
    'context': "Vor einigen Jahren haben Wissenschaftler ein wichtiges Mutagen identifiziert, das in unseren eigenen Zellen liegt: APOBEC, ein Protein, das normalerweise als Schutzmittel gegen Virusinfektionen fungiert. Heute hat ein Team von Schweizer und russischen Wissenschaftlern unter der Leitung von Sergey Nikolaev, Genetiker an der Universität Genf (UNIGE) in der Schweiz, entschlüsselt, wie APOBEC eine Schwäche unseres DNA-Replikationsprozesses ausnutzt, um Mutationen in unserem Genom zu induzieren.",
    'question': "Welches Mutagen schützt vor Virusinfektionen?"
    
})

# output

{'answer': 'APOBEC', 'end': 121, 'score': 0.987, 'start': 115} 

## Even complex queries can be answered pretty well

qa_pipeline({
    "context": "Es wird erwartet, dass sich schwarze Löcher mit Sternmasse bilden,
                wenn sehr massive Sterne am Ende ihres Lebenszyklus 
                zusammenbrechen. Nachdem sich ein Schwarzes Loch gebildet hat, 
                kann es weiter wachsen,indem es Masse aus seiner Umgebung 
                absorbiert. Durch Absorption anderer Sterne und Verschmelzung mit 
                anderen Schwarzen Löchern können sich  supermassereiche Schwarze 
                Löcher mit Millionen von Sonnenmassen (M☉) bilden.  Es besteht 
                Konsens darüber, dass in den Zentren der meisten Galaxien
                supermassereiche Schwarze Löcher existieren.",
    'question': "Wie Sonnenmassen entstehen?"


})

#output

{'answer': 'Durch Absorption anderer Sterne und Verschmelzung mit anderen Schwarzen Löchern',
 'end': 332,
 'score': 0.23970196,
 'start': 253}