Update model metadata
Browse files
README.md
CHANGED
@@ -1,5 +1,16 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
4 |
# Cross-Encoder for MS MARCO - EN-DE
|
5 |
|
@@ -15,7 +26,9 @@ The training code is available in this repository, see `train_script.py`.
|
|
15 |
When you have [SentenceTransformers](https://www.sbert.net/) installed, you can use the model like this:
|
16 |
```python
|
17 |
from sentence_transformers import CrossEncoder
|
|
|
18 |
model = CrossEncoder('model_name', max_length=512)
|
|
|
19 |
query = 'How many people live in Berlin?'
|
20 |
docs = ['Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.', 'New York City is famous for the Metropolitan Museum of Art.']
|
21 |
pairs = [(query, doc) for doc in docs]
|
@@ -69,4 +82,4 @@ We also check the performance of bi-encoders using the same evaluation: The retr
|
|
69 |
|
70 |
|
71 |
|
72 |
-
Note: Docs / Sec gives the number of (query, document) pairs we can re-rank within a second on a V100 GPU.
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
datasets:
|
4 |
+
- sentence-transformers/msmarco
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
- de
|
8 |
+
base_model:
|
9 |
+
- microsoft/Multilingual-MiniLM-L12-H384
|
10 |
+
pipeline_tag: text-ranking
|
11 |
+
library_name: sentence-transformers
|
12 |
+
tags:
|
13 |
+
- transformers
|
14 |
---
|
15 |
# Cross-Encoder for MS MARCO - EN-DE
|
16 |
|
|
|
26 |
When you have [SentenceTransformers](https://www.sbert.net/) installed, you can use the model like this:
|
27 |
```python
|
28 |
from sentence_transformers import CrossEncoder
|
29 |
+
|
30 |
model = CrossEncoder('model_name', max_length=512)
|
31 |
+
|
32 |
query = 'How many people live in Berlin?'
|
33 |
docs = ['Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.', 'New York City is famous for the Metropolitan Museum of Art.']
|
34 |
pairs = [(query, doc) for doc in docs]
|
|
|
82 |
|
83 |
|
84 |
|
85 |
+
Note: Docs / Sec gives the number of (query, document) pairs we can re-rank within a second on a V100 GPU.
|