34 lines
852 B
Python
34 lines
852 B
Python
import difflib
|
|
|
|
from xiaomusic.utils import (
|
|
find_best_match,
|
|
keyword_detection,
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
user_input = "八年的爱"
|
|
s1 = "冰冰超人 - 八年的爱新版"
|
|
s2 = "冰冰超人 - 八年的爱"
|
|
r1 = difflib.SequenceMatcher(None, s1, user_input).ratio()
|
|
r2 = difflib.SequenceMatcher(None, s2, user_input).ratio()
|
|
print(s1, r1)
|
|
print(s2, r2)
|
|
|
|
s3 = "其他"
|
|
str_list = [s2, s1, s3]
|
|
matches, remains = keyword_detection(user_input, str_list, n=10)
|
|
print(matches, remains)
|
|
|
|
extra_search_index = {}
|
|
extra_search_index["1"] = s1
|
|
extra_search_index["2"] = s2
|
|
extra_search_index["3"] = s3
|
|
real_names = find_best_match(
|
|
user_input,
|
|
str_list,
|
|
cutoff=0.4,
|
|
n=100,
|
|
extra_search_index=extra_search_index,
|
|
)
|
|
print(real_names)
|