修改qa脚本
parent
471b62b729
commit
fb7d3b3603
@ -0,0 +1,19 @@
|
|||||||
|
# another_file.py
|
||||||
|
import time
|
||||||
|
|
||||||
|
start_time = time.time()
|
||||||
|
from qa_amend import QAService, match_query
|
||||||
|
|
||||||
|
kb_name = 'my_kb_test'
|
||||||
|
device = None
|
||||||
|
qa_service = QAService(kb_name, device)
|
||||||
|
|
||||||
|
query = "你有哪里不舒服?"
|
||||||
|
top_k = 1
|
||||||
|
score_threshold = 0.1
|
||||||
|
|
||||||
|
match_query(qa_service, query, top_k, 1 - score_threshold)
|
||||||
|
|
||||||
|
end_time = time.time()
|
||||||
|
elapsed_time = end_time - start_time
|
||||||
|
print(f"总耗时: {elapsed_time} 秒")
|
Loading…
Reference in New Issue