35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
import pandas as pd
|
|
import numpy as np
|
|
from benchmark import benchmark
|
|
|
|
setup ="""
|
|
from rapidfuzz import fuzz as rfuzz
|
|
from fuzzywuzzy import fuzz
|
|
import string
|
|
import random
|
|
random.seed(18)
|
|
characters = string.ascii_letters + string.digits + string.whitespace + string.punctuation
|
|
a = ''.join(random.choice(characters) for _ in range(64))
|
|
b_list = [''.join(random.choice(characters) for _ in range({0})) for _ in range({1})]
|
|
"""
|
|
|
|
lengths = list(range(64,512,2))
|
|
count = 2000
|
|
|
|
time_rapidfuzz = benchmark("rapidfuzz",
|
|
'[rfuzz.partial_ratio(a, b) for b in b_list]',
|
|
setup, lengths, count)
|
|
|
|
# this gets very slow, so only benchmark it for smaller values
|
|
time_fuzzywuzzy = benchmark("fuzzywuzzy",
|
|
'[fuzz.partial_ratio(a, b) for b in b_list]',
|
|
setup, list(range(64,256,2)), count) + [np.NaN] * 128
|
|
|
|
df = pd.DataFrame(data={
|
|
"length": lengths,
|
|
"rapidfuzz": time_rapidfuzz,
|
|
"fuzzywuzzy": time_fuzzywuzzy,
|
|
})
|
|
|
|
df.to_csv("results/partial_ratio_short_needle.csv", sep=',',index=False)
|