This page documents BM25 regression experiments for CIRAL (v1.0) — Swahili with document translations. To be clear, the queries are in English and the corpus is in English (translated with NLLB 1.3B).
The exact configurations for these regressions are stored in this YAML file. Note that this page is automatically generated from this template as part of Anserini's regression pipeline, so do not modify this page directly; modify the template instead.
From one of our Waterloo servers (e.g., orca
), the following command will perform the complete regression, end to end:
python src/main/python/run_regression.py --index --verify --search --regression ciral-v1.0-sw-en
Typical indexing command:
bin/run.sh io.anserini.index.IndexCollection \
-threads 16 \
-collection MrTyDiCollection \
-input /path/to/ciral-swahili-english \
-generator DefaultLuceneDocumentGenerator \
-index indexes/lucene-index.ciral-v1.0-sw-en/ \
-storePositions -storeDocvectors -storeRaw \
>& logs/log.ciral-swahili-english &
See this page for more details about the CIRAL corpus. For additional details, see explanation of common indexing options.
After indexing has completed, you should be able to perform retrieval as follows:
bin/run.sh io.anserini.search.SearchCollection \
-index indexes/lucene-index.ciral-v1.0-sw-en/ \
-topics tools/topics-and-qrels/topics.ciral-v1.0-sw-test-a.tsv \
-topicReader TsvInt \
-output runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt \
-bm25 -hits 1000 &
bin/run.sh io.anserini.search.SearchCollection \
-index indexes/lucene-index.ciral-v1.0-sw-en/ \
-topics tools/topics-and-qrels/topics.ciral-v1.0-sw-test-a.tsv \
-topicReader TsvInt \
-output runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt \
-bm25 -hits 1000 &
bin/run.sh io.anserini.search.SearchCollection \
-index indexes/lucene-index.ciral-v1.0-sw-en/ \
-topics tools/topics-and-qrels/topics.ciral-v1.0-sw-test-b.tsv \
-topicReader TsvInt \
-output runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-b.txt \
-bm25 -hits 1000 &
Evaluation can be performed using trec_eval
:
bin/trec_eval -c -m ndcg_cut.20 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-a.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt
bin/trec_eval -c -m recall.100 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-a.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt
bin/trec_eval -c -m ndcg_cut.20 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-a-pools.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt
bin/trec_eval -c -m recall.100 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-a-pools.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-a.txt
bin/trec_eval -c -m ndcg_cut.20 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-b.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-b.txt
bin/trec_eval -c -m recall.100 tools/topics-and-qrels/qrels.ciral-v1.0-sw-test-b.tsv runs/run.ciral-swahili-english.bm25-default.topics.ciral-v1.0-sw-test-b.txt
With the above commands, you should be able to reproduce the following results:
nDCG@20 | BM25 (default) |
---|---|
CIRAL Swahili: Test Set A (Shallow Judgements) | 0.2033 |
CIRAL Swahili: Test Set A (Pools) | 0.2327 |
CIRAL Swahili: Test Set B | 0.2582 |
R@100 | BM25 (default) |
CIRAL Swahili: Test Set A (Shallow Judgements) | 0.4786 |
CIRAL Swahili: Test Set A (Pools) | 0.3636 |
CIRAL Swahili: Test Set B | 0.4918 |