Update README.md
Browse files
README.md
CHANGED
@@ -1,9 +1,16 @@
|
|
1 |
---
|
2 |
library_name: transformers
|
3 |
base_model: "cross-encoder/ms-marco-MiniLM-L-12-v2"
|
4 |
-
|
5 |
-
-
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
tags: ["cross-encoder", "search", "product-search"]
|
8 |
---
|
9 |
|
@@ -32,12 +39,12 @@ queries = [
|
|
32 |
"mustard blouse"
|
33 |
]
|
34 |
documents = [
|
35 |
-
"
|
36 |
-
"
|
37 |
-
"
|
38 |
-
"
|
39 |
-
"
|
40 |
-
"
|
41 |
]
|
42 |
|
43 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
|
|
1 |
---
|
2 |
library_name: transformers
|
3 |
base_model: "cross-encoder/ms-marco-MiniLM-L-12-v2"
|
4 |
+
model-index:
|
5 |
+
- name: esci-ms-marco-MiniLM-L-12-v2
|
6 |
+
results:
|
7 |
+
- task:
|
8 |
+
type: ranking
|
9 |
+
metrics:
|
10 |
+
- type: mrr@10
|
11 |
+
value: 91.74
|
12 |
+
- type: ndcg@10
|
13 |
+
value: 84.83
|
14 |
tags: ["cross-encoder", "search", "product-search"]
|
15 |
---
|
16 |
|
|
|
39 |
"mustard blouse"
|
40 |
]
|
41 |
documents = [
|
42 |
+
"Nike Air Max, with air cushion",
|
43 |
+
"Adidas Ultraboost, the best boost you can get",
|
44 |
+
"Women's sandals wide width 9",
|
45 |
+
"Girl's surf backpack",
|
46 |
+
"Fresh watermelon, all you can eat",
|
47 |
+
"Floral yellow dress with frills and lace"
|
48 |
]
|
49 |
|
50 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|