Add NuMER as dataset for training and update expected output
Browse files
README.md
CHANGED
@@ -7,7 +7,7 @@ datasets:
|
|
7 |
- wjbmattingly/gliner-bird-diet-synthetic
|
8 |
- knowledgator/GLINER-multi-task-synthetic-data
|
9 |
- urchade/pile-mistral-v0.1
|
10 |
-
-
|
11 |
pipeline_tag: token-classification
|
12 |
base_model:
|
13 |
- answerdotai/ModernBERT-base
|
@@ -50,22 +50,22 @@ labels = ["person", "award", "date", "competitions", "teams"]
|
|
50 |
entities = model.predict_entities(text, labels)
|
51 |
|
52 |
for entity in entities:
|
53 |
-
print(entity["text"], "
|
54 |
```
|
55 |
|
56 |
```
|
57 |
-
Cristiano Ronaldo dos Santos Aveiro => person
|
58 |
-
5 February 1985 => date
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
UEFA Champions Leagues => competitions
|
65 |
-
UEFA European Championship => competitions
|
66 |
-
UEFA Nations League => competitions
|
67 |
-
Champions League => competitions
|
68 |
-
European Championship => competitions
|
69 |
```
|
70 |
|
71 |
## Named Entity Recognition benchmark result
|
|
|
7 |
- wjbmattingly/gliner-bird-diet-synthetic
|
8 |
- knowledgator/GLINER-multi-task-synthetic-data
|
9 |
- urchade/pile-mistral-v0.1
|
10 |
+
- numind/NuNER
|
11 |
pipeline_tag: token-classification
|
12 |
base_model:
|
13 |
- answerdotai/ModernBERT-base
|
|
|
50 |
entities = model.predict_entities(text, labels)
|
51 |
|
52 |
for entity in entities:
|
53 |
+
print(entity["text"], "<=>", entity["label"], "<=>", round(entity["score"], 3))
|
54 |
```
|
55 |
|
56 |
```
|
57 |
+
Cristiano Ronaldo dos Santos Aveiro => person <=> 0.552
|
58 |
+
5 February 1985 => date <=> 0.776
|
59 |
+
Saudi Pro League => teams <=> 0.146
|
60 |
+
Ballon d'Or awards => award <=> 0.343
|
61 |
+
UEFA Men's Player of the Year Awards => award <=> 0.56
|
62 |
+
European Golden Shoes => award <=> 0.482
|
63 |
+
33 trophies in his career, including seven league titles => award <=> 0.126
|
64 |
+
UEFA Champions Leagues => competitions <=> 0.196
|
65 |
+
UEFA European Championship => competitions <=> 0.271
|
66 |
+
UEFA Nations League => competitions <=> 0.273
|
67 |
+
Champions League => competitions <=> 0.172
|
68 |
+
European Championship => competitions <=> 0.22
|
69 |
```
|
70 |
|
71 |
## Named Entity Recognition benchmark result
|