neurae/dnd_style_intents
Viewer • Updated • 163k • 304 • 6
How to use neurae/albert-dnd-intents with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="neurae/albert-dnd-intents") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("neurae/albert-dnd-intents")
model = AutoModelForSequenceClassification.from_pretrained("neurae/albert-dnd-intents")This is albert base tuned with optimal lr, lr scheduler and weight decay on dnd-style-intents dataset.
| parametrs | value |
|---|---|
| learning rate | 5e-5 |
| lr scheduler | linear |
| weight decay | 0 |
Model has next metrics on test data from dataset
| metric | value |
|---|---|
| accuracy | 0.981 |
| Macro F1 | 0.979 |
| Micro F1 | 0.985 |