tomaarsen HF Staff commited on
Commit
533a95b
·
verified ·
1 Parent(s): 970944b

Update model metadata

Browse files
Files changed (1) hide show
  1. README.md +14 -1
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.