capreolus-ir / capreolus

A toolkit for end-to-end neural ad hoc retrieval
https://capreolus.ai
Apache License 2.0
95 stars 32 forks source link

How to use capreolus for Robust evaluation? #197

Closed yiyaxiaozhi closed 2 years ago

yiyaxiaozhi commented 2 years ago

I obtain the model checkpoint on Robust04 fold-2 by the following command: python -m capreolus.run rerank.traineval with file=docs/reproduction/config_paradept.txt fold=s2 but while running the prediction, the progress is killed by accident. Therefore it did not display the test result. What command should I use to load the checkpoint and run the fold-2 test?

andrewyates commented 2 years ago

If training completed, you can replace rerank.traineval with rerank.evaluate to run only the evaluation. This will only work if all parts of training finished, which includes predictions on the dev set, for example.