eriktks/conll2003
Updated • 31.9k • 177
How to use JohnLei/bert-large-5-ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="JohnLei/bert-large-5-ner") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("JohnLei/bert-large-5-ner")
model = AutoModelForTokenClassification.from_pretrained("JohnLei/bert-large-5-ner", device_map="auto")This model is a fine-tuned version of bert-large-cased on the conll2003 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 | 3 | 1.6779 | 0.0398 | 0.0825 | 0.0537 | 0.6719 |
| No log | 2.0 | 6 | 1.4283 | 0.0579 | 0.0527 | 0.0551 | 0.7628 |
| No log | 3.0 | 9 | 1.3380 | 0.0572 | 0.0409 | 0.0477 | 0.7758 |
Base model
google-bert/bert-large-cased