File size: 1,095 Bytes
74b1bac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import unittest
from .. import BM25TestCase
class TestRankBM25Quick(BM25TestCase):
def test_bm25_sparse_vs_rank_bm25_on_nfcorpus(self):
self.compare_with_rank_bm25(
"nfcorpus", corpus_subsample=2000, queries_subsample=200, method="bm25+"
)
def test_bm25_sparse_vs_rank_bm25_on_scifact(self):
self.compare_with_rank_bm25(
"scifact", corpus_subsample=2000, queries_subsample=200, method="bm25+"
)
def test_bm25_sparse_vs_rank_bm25_on_scidocs(self):
self.compare_with_rank_bm25(
"scidocs", corpus_subsample=2000, queries_subsample=200, method="bm25+"
)
# fiqa
def test_bm25_sparse_vs_rank_bm25_on_fiqa(self):
self.compare_with_rank_bm25(
"fiqa", corpus_subsample=2000, queries_subsample=200, method="bm25+"
)
# arguana
def test_bm25_sparse_vs_rank_bm25_on_arguana(self):
self.compare_with_rank_bm25(
"arguana", queries_subsample=100, corpus_subsample=1000, method="bm25+"
)
if __name__ == "__main__":
unittest.main()
|