Jean-Baptiste commited on
Commit
3272069
1 Parent(s): 6de88b8

Update README.md

Browse files

Changing example with new aggregation_strategy parameter

Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -31,7 +31,7 @@ model = AutoModelForTokenClassification.from_pretrained("Jean-Baptiste/camembert
31
 
32
  from transformers import pipeline
33
 
34
- nlp = pipeline('ner', model=model, tokenizer=tokenizer, grouped_entities=True)
35
  nlp("Apple est créée le 1er avril 1976 dans le garage de la maison d'enfance de Steve Jobs à Los Altos en Californie par Steve Jobs, Steve Wozniak et Ronald Wayne14, puis constituée sous forme de société le 3 janvier 1977 à l'origine sous le nom d'Apple Computer, mais pour ses 30 ans et pour refléter la diversification de ses produits, le mot « computer » est retiré le 9 janvier 2015.")
36
 
37
 
@@ -91,17 +91,17 @@ Global
91
  By entity
92
  ```
93
  'LOC': {'precision': 0.8905576596578294,
94
- 'recall': 0.900554675118859,
95
- 'f1': 0.8955282684352223},
96
  'MISC': {'precision': 0.8175627240143369,
97
- 'recall': 0.8117437722419929,
98
- 'f1': 0.8146428571428571},
99
  'ORG': {'precision': 0.8099480326651819,
100
- 'recall': 0.8265151515151515,
101
- 'f1': 0.8181477315335584},
102
  'PER': {'precision': 0.9372509960159362,
103
- 'recall': 0.959812321501428,
104
- 'f1': 0.9483975005039308}
105
 
106
  ```
107
 
 
31
 
32
  from transformers import pipeline
33
 
34
+ nlp = pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy="simple")
35
  nlp("Apple est créée le 1er avril 1976 dans le garage de la maison d'enfance de Steve Jobs à Los Altos en Californie par Steve Jobs, Steve Wozniak et Ronald Wayne14, puis constituée sous forme de société le 3 janvier 1977 à l'origine sous le nom d'Apple Computer, mais pour ses 30 ans et pour refléter la diversification de ses produits, le mot « computer » est retiré le 9 janvier 2015.")
36
 
37
 
 
91
  By entity
92
  ```
93
  'LOC': {'precision': 0.8905576596578294,
94
+ \t\t'recall': 0.900554675118859,
95
+ \t\t'f1': 0.8955282684352223},
96
  'MISC': {'precision': 0.8175627240143369,
97
+ \t\t 'recall': 0.8117437722419929,
98
+ \t\t 'f1': 0.8146428571428571},
99
  'ORG': {'precision': 0.8099480326651819,
100
+ \t\t'recall': 0.8265151515151515,
101
+ \t\t'f1': 0.8181477315335584},
102
  'PER': {'precision': 0.9372509960159362,
103
+ \t\t'recall': 0.959812321501428,
104
+ \t\t'f1': 0.9483975005039308}
105
 
106
  ```
107