README.md

---
license: apache-2.0
tags:
- question-answering
- squad
- transformers
- pytorch
- evaluation
- hf-course
- fine-tuned
datasets:
- squad
metrics:
- exact_match
- f1
model-index:
  - name: QA-SQuAD-BERT
    results:
      - task:
          type: question-answering
          name: Question Answering
        dataset:
          name: SQuAD v1.1
          type: squad
        metrics:
          - name: Exact Match
            type: exact_match
            value: 82.7
          - name: F1
            type: f1
            value: 87.0039
---

# QA-SQuAD-BERT

A BERT-based model fine-tuned on SQuAD v1.1 for extractive QA

## Model Description

This model is based on bert-base-uncased and was fine-tuned on the **SQuAD v1.1** dataset for extractive question answering. It takes a question and a context passage as input and predicts the span of text in the passage that most likely answers the question.

The model was trained using the Hugging Face 🤗 Transformers library.

## Intended Uses & Limitations

### Intended Uses

- Extractive question answering on Wikipedia-style passages.
- As a downstream component in information retrieval pipelines.
- Educational purposes or experimentation with fine-tuning on QA tasks.

### Limitations

- The model may not generalize well to out-of-domain datasets.
- It does not handle unanswerable questions (not trained on SQuAD v2.0).
- It may produce incorrect or misleading answers if context is ambiguous.

## Training Details

- **Base model**: bert-base-uncased
- **Dataset**: [SQuAD v1.1](https://huggingface.co/datasets/squad)
- **Epochs**: 3
- **Batch size**: 8
- **Learning rate**: 2e-5
- **Optimizer**: AdamW
- **Max length**: 384
- **Hardware used**: Colab/GPU T4

## Evaluation Results

The model was evaluated on the SQuAD v1.1 development set using the standard metrics: Exact Match (EM) and F1.

| Metric       | Score |
|--------------|-------|
| Exact Match  | 82.7 |
| F1           | 87.0039 |

## How to Use

You can load this model using the `pipeline` API:

```python
from transformers import pipeline

qa_pipeline = pipeline("question-answering", model="tmt3103/SQuAD_BERT")
result = qa_pipeline({
    "context": "Hugging Face is creating a tool that democratizes AI.",
    "question": "What is Hugging Face creating?"
})
print(result)