task_categories:text-retrieval region:us pyterrier pyterrier-artifact pyterrier-artifact.dense_index pyterrier-artifact.dense_index.flex
---
# pretty_name: "" # Example: "MS MARCO Terrier Index"
tags:
- pyterrier
- pyterrier-artifact
- pyterrier-artifact.dense_index
- pyterrier-artifact.dense_index.flex
task_categories:
- text-retrieval
viewer: false
---

# scifact.retromae.flex

## Description

RetroMAE index for SciFact

## Usage

```python
# Load the artifact
import pyterrier_alpha as pta
artifact = pta.Artifact.from_hf('pyterrier/scifact.retromae.flex')
artifact.np_retriever()
```

## Benchmarks

`scifact/test`

| name      |   nDCG@10 |   R@1000 |
|:----------|----------:|---------:|
| np (flat) |    0.6474 |     0.98 |

## Reproduction

```python
import pyterrier as pt
from tqdm import tqdm
import ir_datasets
from pyterrier_dr import FlexIndex, RetroMAE
pipeline = RetroMAE.msmarco_distill() >> FlexIndex("scifact.retromae.flex")
dataset = ir_datasets.load('beir/scifact')
docs = ({'docno': d.doc_id, 'text': d.default_text()} for d in tqdm(dataset.docs))
pipeline.index(docs)
```

## Metadata

```
{
  "type": "dense_index",
  "format": "flex",
  "vec_size": 768,
  "doc_count": 5183
}
```
