Skip to content

Commit 8fc9e71

Browse files
🎨 ⚡️ Limit stanza nlp pipeline to tokenize
1 parent df50738 commit 8fc9e71

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

benchmarks/stanza_benchmark.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import stanza
22
# stanza.download('en')
3-
nlp = stanza.Pipeline('en')
3+
nlp = stanza.Pipeline(lang='en', processors='tokenize')
44

55
from .english_golden_rules import GOLDEN_EN_RULES
66

@@ -20,10 +20,10 @@ def stanza_english_benchmark(golden_rules):
2020
import timeit
2121
time_taken = timeit.timeit("stanza_english_benchmark(GOLDEN_EN_RULES)",
2222
setup="from __main__ import stanza_english_benchmark, GOLDEN_EN_RULES",
23-
number=10)
23+
number=100)
2424
print(f'{stanza.__name__} - v{stanza.__version__}')
2525
print(f'GRS score: {percent_score:0.2f}%')
2626
print(f'Speed: {time_taken*1000:>10.2f} ms')
2727
# stanza - v1.0.1
2828
# GRS score: 72.92%
29-
# Speed: 109839.63 ms
29+
# Speed: 120803.37 ms

0 commit comments

Comments
 (0)