TharushiDinushika/modern-bert-finetuned-query-classification

BERT Fine-tuned for Query Classification

This model is a fine-tuned version of bert-base-uncased on a query classification dataset.

Model description

The model was fine-tuned on queries to classify them into specific categories.

Training and evaluation data

The model was trained on [describe your dataset here].

Training procedure

The model was trained with the following hyperparameters:

• Learning rate: 2e-05
• Batch size: 8
• Number of epochs: 10
• Optimizer: AdamW
• Weight decay: 0.01

Evaluation results

The model achieved the following results on the validation set:

• Accuracy: 0.9796
• F1 Score: 0.9796

Uses and limitations

[Discuss the intended uses and limitations of your model]