birgermoell commited on
Commit
397397b
1 Parent(s): fec41d6

Updated model

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. eval_results.json +12 -0
README.md CHANGED
@@ -86,9 +86,9 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
86
 
87
  from transformers import pipeline
88
 
89
- tokenizer = AutoTokenizer.from_pretrained("philschmid/distilroberta-base-ner-wikiann")
90
 
91
- model = AutoModelForTokenClassification.from_pretrained("philschmid/distilroberta-base-ner-wikiann")
92
 
93
  nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
94
 
86
 
87
  from transformers import pipeline
88
 
89
+ tokenizer = AutoTokenizer.from_pretrained("birgermoell/ner-swedish-wikiann")
90
 
91
+ model = AutoModelForTokenClassification.from_pretrained("birgermoell/ner-swedish-wikiann")
92
 
93
  nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
94
 
eval_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 3.0,
3
+ "eval_accuracy": 0.9848601595201899,
4
+ "eval_f1": 0.9434072987346416,
5
+ "eval_loss": 0.08223789930343628,
6
+ "eval_precision": 0.939307624890447,
7
+ "eval_recall": 0.9475429160834009,
8
+ "eval_runtime": 148.4881,
9
+ "eval_samples": 10000,
10
+ "eval_samples_per_second": 67.345,
11
+ "eval_steps_per_second": 8.418
12
+ }