|
--- |
|
library_name: xpmir |
|
--- |
|
# Runs an experiment |
|
|
|
None |
|
|
|
## Using the model) |
|
The model can be loaded with [experimaestro IR](https://experimaestro-ir.readthedocs.io/en/latest/) |
|
|
|
```py |
|
from xpmir.models import AutoModel |
|
|
|
# Model that can be re-used in experiments |
|
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE") |
|
|
|
# Use this if you want to actually use the model |
|
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_DistilMSE", as_instance=True) |
|
model.initialize(None) |
|
model.rsv("walgreens store sales average", "The average Walgreens salary ranges from approximately $15,000 per year for Customer Service Associate / Cashier to $179,900 per year for District Manager...") |
|
``` |
|
|
|
## Results |
|
| Dataset | AP | P@20 | RR | RR@10 | nDCG | nDCG@10 | nDCG@20 | |
|
|----| ---|------|------|------|------|------|------| |
|
| trec2019 | 0.5102 | 0.7360 | 0.9612 | 0.9612 | 0.7407 | 0.7300 | 0.7097 | |
|
| msmarco_dev | 0.3623 | 0.0384 | 0.3673 | 0.3560 | 0.4870 | 0.4207 | 0.4451 | |
|
|