electra-dnd-intents / README.md
MichaelOnly's picture
Update README.md
76244ff
metadata
datasets:
  - neurae/dnd_style_intents
language:
  - en
pipeline_tag: text-classification
license: apache-2.0
metrics:
  - accuracy
  - f1

This is electra base tuned with optimal lr, lr scheduler and weight decay on dnd-style-intents dataset.

parametrs value
learning rate 6.6e-5
lr scheduler cosine with restarts
weight decay 0

Model has next metrics on test data from dataset

metric value
accuracy 0.978
Macro F1 0.976
Micro F1 0.978