leondz/wnut_17
Updated • 2.19k • 19
How to use Gladiator/distilbert-base-uncased_ner_wnut_17 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="Gladiator/distilbert-base-uncased_ner_wnut_17") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("Gladiator/distilbert-base-uncased_ner_wnut_17")
model = AutoModelForTokenClassification.from_pretrained("Gladiator/distilbert-base-uncased_ner_wnut_17", device_map="auto")This model is a fine-tuned version of distilbert-base-uncased on the wnut_17 dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| No log | 1.0 | 213 | 0.2367 | 0.6879 | 0.4270 | 0.5269 | 0.9455 |
| No log | 2.0 | 426 | 0.2272 | 0.6913 | 0.4928 | 0.5754 | 0.9533 |
| 0.173 | 3.0 | 639 | 0.2393 | 0.6788 | 0.5132 | 0.5845 | 0.9553 |
| 0.173 | 4.0 | 852 | 0.2338 | 0.6541 | 0.5610 | 0.6040 | 0.9557 |
| 0.0489 | 5.0 | 1065 | 0.2400 | 0.6701 | 0.5467 | 0.6021 | 0.9559 |