metadata
language: en
license: mit
datasets:
- your_dataset_name
tags:
- text-classification
- bert
- query-classification
metrics:
- accuracy: 0.975925925925926
- f1: 0.975935077462957
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: 5
- Optimizer: AdamW
- Weight decay: 0.01
Evaluation results
The model achieved the following results on the validation set:
- Accuracy: 0.9759
- F1 Score: 0.9759
Uses and limitations
[Discuss the intended uses and limitations of your model]