Update README.md

README.md

@@ -39,25 +39,31 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilroberta-base-ner-wikiann-conll2003-4-class
 
-This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the wikiann-conll2003 dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.0705
-- Precision: 0.9492
-- Recall: 0.9585
-- F1: 0.9539
-- Accuracy: 0.9882
+This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the wikiann and conll2003 dataset. It consists out of the classes of conll2003. 
+
+O (0), B-PER (1), I-PER (2), B-ORG (3), I-ORG (4) B-LOC (5), I-LOC (6) B-MISC (7), I-MISC (8).
+
+eval F1-Score: **95,39** (merged dataset)   
+test F1-Score: **90,75** (merged dataset) 
+
+eval F1-Score: **95,39** (CoNLL-03)   
+test F1-Score: **90,75** (CoNLL-03) 
 
-## Model description
+## Model Usage
 
-More information needed
+```python
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers import pipeline
 
-## Intended uses & limitations
+tokenizer = AutoTokenizer.from_pretrained("philschmid/distilroberta-base-ner-wikiann-conll2003-4-class")
+model = AutoModelForTokenClassification.from_pretrained("philschmid/distilroberta-base-ner-wikiann-conll2003-4-class")
 
-More information needed
+nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
+example = "My name is Philipp and live in Germany"
 
-## Training and evaluation data
+nlp(example)
 
-More information needed
+```
 
 ## Training procedure
 
@@ -75,6 +81,20 @@ The following hyperparameters were used during training:
 
 ### Training results
 
+It achieves the following results on the evaluation set:
+- Loss: 0.0705
+- Precision: 0.9492
+- Recall: 0.9585
+- F1: 0.9539
+- Accuracy: 0.9882
+
+It achieves the following results on the test set:
+- Loss: 0.239
+- Precision: 0.8984
+- Recall: 0.9168
+- F1: 0.9075
+- Accuracy: 0.9741
+
 
 
 ### Framework versions