File size: 911 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 |
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)
def test_bm25_sparse_vs_rank_bm25_on_scifact(self):
self.compare_with_rank_bm25("scifact", corpus_subsample=2000, queries_subsample=200)
def test_bm25_sparse_vs_rank_bm25_on_scidocs(self):
self.compare_with_rank_bm25("scidocs", corpus_subsample=2000, queries_subsample=200)
# fiqa
def test_bm25_sparse_vs_rank_bm25_on_fiqa(self):
self.compare_with_rank_bm25("fiqa", corpus_subsample=2000, queries_subsample=200)
# arguana
def test_bm25_sparse_vs_rank_bm25_on_arguana(self):
self.compare_with_rank_bm25("arguana", queries_subsample=100, corpus_subsample=1000)
if __name__ == '__main__':
unittest.main()
|