diff options
Diffstat (limited to 'hydro_roll/models')
| -rw-r--r-- | hydro_roll/models/__init__.py | 0 | ||||
| -rw-r--r-- | hydro_roll/models/hola.pkl | bin | 0 -> 209 bytes | |||
| -rw-r--r-- | hydro_roll/models/utils.py | 14 |
3 files changed, 14 insertions, 0 deletions
diff --git a/hydro_roll/models/__init__.py b/hydro_roll/models/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/hydro_roll/models/__init__.py diff --git a/hydro_roll/models/hola.pkl b/hydro_roll/models/hola.pkl Binary files differnew file mode 100644 index 0000000..b9940fa --- /dev/null +++ b/hydro_roll/models/hola.pkl diff --git a/hydro_roll/models/utils.py b/hydro_roll/models/utils.py new file mode 100644 index 0000000..73e7ba0 --- /dev/null +++ b/hydro_roll/models/utils.py @@ -0,0 +1,14 @@ +import difflib +import pickle + +def find_max_similarity(input_string, string_list): + max_similarity = 0 + max_string = "" + + for string in string_list: + similarity = difflib.SequenceMatcher(None, input_string, string).quick_ratio() + if similarity > max_similarity: + max_similarity = similarity + max_string = string + + return max_string, max_similarity
\ No newline at end of file |