Instructions to use webis/tiny-bert-ranker with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use webis/tiny-bert-ranker with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("webis/tiny-bert-ranker") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores) - Notebooks
- Google Colab
- Kaggle
File size: 1,140 Bytes
2f908c4 f99bd25 2f908c4 | 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 26 27 28 29 | ---
language:
- en
license: mit
library_name: sentence-transformers
pipeline_tag: text-ranking
---
# tiny-bert-ranker model card
This model is a fine-tuned version of [prajjwal1/bert-tiny](https://web.archive.org/web/20240315094214/https://huggingface.co/prajjwal1/bert-tiny)
as part of our submission to [ReNeuIR 2024](https://web.archive.org/web/20240704171521/https://reneuir.org/shared_task.html).
## Model Details
### Model Description
<!-- Provide a longer summary of what this model is. -->
The model is based on the pre-trained [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny). It is fine-tuned on a 1GB subset of data
extracted from msmarco's [Train Triples Small](https://web.archive.org/web/20231209043304/https://microsoft.github.io/msmarco/Datasets.html).
Tiny-bert-ranker is part of our investigation into the tradeoffs between efficiency and effectiveness in ranking models.
This approach does not involve BM25 score injection or distillation.
- **Developed by:** Team FSU at ReNeuIR 2024
- **Model type:** sequence-to-sequence model
- **License:** mit
- **Finetuned from model:** prajjwal1/bert-tiny
|