Hubert Fine Tuned Hungarian Squadv2

🚀 huBERT Fine-tuned on Hungarian SQuADv2

This is a fine-tuned version of the huBERT base model on the Hungarian SQuADv2 dataset, designed for question-answering tasks.

🚀 Quick Start

✨ Features

• The model is a fine-tuned version of the huBERT base model (cased) on the SQuADv2 dataset.
• It uses the Hungarian SQuADv2 dataset, which is a machine-translated version of the SQuAD dataset using the Google Translate API.

📦 Installation

No specific installation steps are provided in the original document.

💻 Usage Examples

Basic Usage

from transformers import pipeline
qa_pipeline = pipeline(
    "question-answering",
    model="mcsabai/huBert-fine-tuned-hungarian-squadv2",
    tokenizer="mcsabai/huBert-fine-tuned-hungarian-squadv2",
    topk = 1,
    handle_impossible_answer = True
)
predictions = qa_pipeline({
    'context': "Máté vagyok és Budapesten élek már több mint 4 éve.",
    'question': "Hol lakik Máté?"
})
print(predictions)
# output:
# {'score': 0.9892364144325256, 'start': 16, 'end': 26, 'answer': 'Budapesten'}

Parameter Explanation

• topk (int, optional, defaults to 1): The number of answers to return (will be chosen by order of likelihood). Note that we return less than topk answers if there are not enough options available within the context.
• handle_impossible_answer (bool, optional, defaults to False): Whether or not we accept impossible as an answer.

📚 Documentation

• huBert model + Tokenizer: https://huggingface.co/SZTAKI-HLT/hubert-base-cc
• Hungarian SQUADv2 dataset: Machine Translated SQuAD dataset (Google Translate API)

📄 License

No license information is provided in the original document.

Reference

[1] https://rajpurkar.github.io/SQuAD-explorer/