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
Created README.md
#1
by Integer-Ctrl - opened
README.md
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
license: mit
|
| 5 |
+
---
|
| 6 |
+
# tiny-bert-ranker model card
|
| 7 |
+
|
| 8 |
+
This model is a fine-tuned version of [prajjwal1/bert-tiny](https://web.archive.org/web/20240315094214/https://huggingface.co/prajjwal1/bert-tiny)
|
| 9 |
+
as part of our submission to [ReNeuIR 2024](https://web.archive.org/web/20240704171521/https://reneuir.org/shared_task.html).
|
| 10 |
+
|
| 11 |
+
## Model Details
|
| 12 |
+
|
| 13 |
+
### Model Description
|
| 14 |
+
|
| 15 |
+
<!-- Provide a longer summary of what this model is. -->
|
| 16 |
+
|
| 17 |
+
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
|
| 18 |
+
extracted from msmarco's [Train Triples Small](https://web.archive.org/web/20231209043304/https://microsoft.github.io/msmarco/Datasets.html).
|
| 19 |
+
|
| 20 |
+
Tiny-bert-ranker is part of our investigation into the tradeoffs between efficiency and effectiveness in ranking models.
|
| 21 |
+
This approach does not involve BM25 score injection or distillation.
|
| 22 |
+
|
| 23 |
+
- **Developed by:** Team FSU at ReNeuIR 2024
|
| 24 |
+
- **Model type:** sequence-to-sequence model
|
| 25 |
+
- **License:** mit
|
| 26 |
+
- **Finetuned from model:** prajjwal1/bert-tiny
|