philschmid commited on
Commit
23d3bb5
·
1 Parent(s): 5c7a9c6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -12
README.md CHANGED
@@ -40,25 +40,27 @@ should probably proofread and complete it, then remove this comment. -->
40
  # distilroberta-base-ner-wikiann
41
 
42
  This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the wikiann dataset.
43
- It achieves the following results on the evaluation set:
44
- - Loss: 0.3156
45
- - Precision: 0.8332
46
- - Recall: 0.8424
47
- - F1: 0.8378
48
- - Accuracy: 0.9193
49
 
50
- ## Model description
51
 
52
- More information needed
 
 
 
 
53
 
54
- ## Intended uses & limitations
 
 
55
 
56
- More information needed
 
57
 
58
- ## Training and evaluation data
 
59
 
60
- More information needed
61
 
 
62
  ## Training procedure
63
 
64
  ### Training hyperparameters
@@ -75,6 +77,19 @@ The following hyperparameters were used during training:
75
 
76
  ### Training results
77
 
 
 
 
 
 
 
 
 
 
 
 
 
 
78
 
79
 
80
  ### Framework versions
 
40
  # distilroberta-base-ner-wikiann
41
 
42
  This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the wikiann dataset.
 
 
 
 
 
 
43
 
 
44
 
45
+ eval F1-Score: **83,78** (CoNLL-03)
46
+ test F1-Score: **83,76** (CoNLL-03)
47
+
48
+
49
+ ## Model Usage
50
 
51
+ ```python
52
+ from transformers import AutoTokenizer, AutoModelForTokenClassification
53
+ from transformers import pipeline
54
 
55
+ tokenizer = AutoTokenizer.from_pretrained("philschmid/distilroberta-base-ner-wikiann")
56
+ model = AutoModelForTokenClassification.from_pretrained("philschmid/distilroberta-base-ner-wikiann")
57
 
58
+ nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
59
+ example = "My name is Philipp and live in Germany"
60
 
61
+ nlp(example)
62
 
63
+ ```
64
  ## Training procedure
65
 
66
  ### Training hyperparameters
 
77
 
78
  ### Training results
79
 
80
+ It achieves the following results on the evaluation set:
81
+ - Loss: 0.3156
82
+ - Precision: 0.8332
83
+ - Recall: 0.8424
84
+ - F1: 0.8378
85
+ - Accuracy: 0.9193
86
+
87
+ It achieves the following results on the test set:
88
+ - Loss: 0.3023
89
+ - Precision: 0.8301
90
+ - Recall: 0.8452
91
+ - F1: 0.8376
92
+ - Accuracy: 0.92
93
 
94
 
95
  ### Framework versions