eriktks/conll2003
Updated • 31.9k • 177
How to use JohnLei/bert-base-ner-5 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="JohnLei/bert-base-ner-5") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("JohnLei/bert-base-ner-5")
model = AutoModelForTokenClassification.from_pretrained("JohnLei/bert-base-ner-5", device_map="auto")This model is a fine-tuned version of bert-base-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 | 2.0874 | 0.0189 | 0.0873 | 0.0311 | 0.2981 |
| No log | 2.0 | 6 | 1.9620 | 0.0195 | 0.0552 | 0.0288 | 0.5350 |
| No log | 3.0 | 9 | 1.9119 | 0.0205 | 0.0443 | 0.0280 | 0.6152 |
Base model
google-bert/bert-base-cased