Downloads · 30 days
11
31% of all-time downloads
Aakash22134Restarant-Search-Distaled_Bert
Aakash22134/Aakash22134Restarant-Search-Distaled_Bert
Aakash22134Restarant-Search-Distaled_Bert is a token classification model from Aakash22134. Use it when you need labels on individual words, such as names. The card lists the license as apache-2.0.
Fine-tuned DistilBERT for Named Entity Recognition in restaurant search queries.
All-time downloads
35
Public
Parameters
66.4M
266 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors266 MB · 100%
At a glance
- Task
- Token Classification
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Oct 1, 2025
- Updated
- Oct 1, 2025
- SHA
- 122e3f12
Try a prompt
Model card
From the Hugging Face model README
Restaurant Search NER with DistilBERT
Fine-tuned DistilBERT for Named Entity Recognition in restaurant search queries.
Performance
- F1 Score: 79.85%
- Precision: 78.30%
- Recall: 81.46%
- Accuracy: 91.96%
Training Results
| Epoch | Training Loss | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|
| 1 | 0.633 | 0.306 | 0.731 | 0.794 | 0.762 | 0.909 |
| 2 | 0.249 | 0.286 | 0.770 | 0.809 | 0.789 | 0.916 |
| 3 | 0.203 | 0.284 | 0.783 | 0.815 | 0.799 | 0.920 |
Entity Types
Rating, Location, Amenity, Cuisine, Dish, Hours, Price, Restaurant_Name
Usage
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained('Aakash22134/Aakash22134Restarant-Search-Distaled_Bert')
model = AutoModelForTokenClassification.from_pretrained('Aakash22134/Aakash22134Restarant-Search-Distaled_Bert')
- Task
- Token Classification
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Oct 1, 2025
- Updated
- Oct 1, 2025