Commit
·
e21d0e1
1
Parent(s):
a5515f4
try better aggregation scheme now that I understand what it does
Browse files- helpers.py +2 -2
helpers.py
CHANGED
|
@@ -2,11 +2,11 @@ from transformers import pipeline
|
|
| 2 |
|
| 3 |
baseline_classifier = pipeline("ner",
|
| 4 |
model="Dagobert42/xlnet-base-cased-biored-augmented",
|
| 5 |
-
aggregation_strategy="
|
| 6 |
)
|
| 7 |
augmented_classifier = pipeline("ner",
|
| 8 |
model="Dagobert42/xlnet-base-cased-biored-finetuned",
|
| 9 |
-
aggregation_strategy="
|
| 10 |
)
|
| 11 |
|
| 12 |
def annotate_sentence(sentence, predictions):
|
|
|
|
| 2 |
|
| 3 |
baseline_classifier = pipeline("ner",
|
| 4 |
model="Dagobert42/xlnet-base-cased-biored-augmented",
|
| 5 |
+
aggregation_strategy="average"
|
| 6 |
)
|
| 7 |
augmented_classifier = pipeline("ner",
|
| 8 |
model="Dagobert42/xlnet-base-cased-biored-finetuned",
|
| 9 |
+
aggregation_strategy="average"
|
| 10 |
)
|
| 11 |
|
| 12 |
def annotate_sentence(sentence, predictions):
|