Skip to content

Commit 94f2817

Browse files
authored
Update GetTopK.py
1 parent ba3c1f9 commit 94f2817

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

GetTopK.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
################################################################################################################
2-
# Author: Suthee Chaidaroon
2+
# Author: Suthee Un
33
44
################################################################################################################
55
import numpy as np
@@ -128,4 +128,4 @@ def GetTopK_UsingCosineSim(outfn, queries, documents, TopK, queryBatchSize=10, d
128128
out_fn = "bm25/{}_test_top101.txt".format(dataset)
129129

130130
print("save the result to {}".format(out_fn))
131-
GetTopK_UsingCosineSim(out_fn, query_corpus, data.train, TopK=101, queryBatchSize=500, docBatchSize=100)
131+
GetTopK_UsingCosineSim(out_fn, query_corpus, data.train, TopK=101, queryBatchSize=500, docBatchSize=100)

0 commit comments

Comments
 (0)