22f3001251/dlgenai-bert-roberta

BERT-base MCQ Answer Selection

Fine-tuned BERT-base-uncased to select the correct answer option (A–E) for multiple-choice questions.

Files

• —model.safetensors — fine-tuned weights
• —config.json — model configuration
• —tokenizer.json / tokenizer_config.json — tokenizer

Training details

• —Base model: google-bert/bert-base-uncased
• —Task: rank 5 answer options per prompt (top-1 accuracy + mAP@3)
• —Format: prompt + 5 options, each option scored via the [CLS] embedding

Usage

from transformers import AutoModelForSequenceClassification, AutoTokenizer
from safetensors.torch import load_file

# Load tokenizer
tokenizer = AutoTokenizer.from_pretrained("22f3001251/dlgenai-bert-roberta")

See training/bert_roberta.ipynb in the companion repo for the full training and inference flow.