namdp-ptit
commited on
Commit
•
524a63b
1
Parent(s):
4d8600b
Update README.md
Browse files
README.md
CHANGED
@@ -14,9 +14,9 @@ widget:
|
|
14 |
- text: tỉnh nào có diện tích lớn nhất việt nam
|
15 |
output:
|
16 |
- label: nghệ an có diện tích lớn nhất việt nam
|
17 |
-
score: 0.
|
18 |
- label: bắc ninh có diện tích nhỏ nhất việt nam
|
19 |
-
score: 0.
|
20 |
---
|
21 |
|
22 |
# Reranker
|
@@ -50,31 +50,31 @@ from FlagEmbedding import FlagReranker
|
|
50 |
reranker = FlagReranker('namdp-ptit/ViRanker',
|
51 |
use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation
|
52 |
|
53 |
-
score = reranker.compute_score(['
|
54 |
-
print(score) #
|
55 |
|
56 |
# You can map the scores into 0-1 by set "normalize=True", which will apply sigmoid function to the score
|
57 |
-
score = reranker.compute_score(['
|
58 |
normalize=True)
|
59 |
-
print(score) # 0.
|
60 |
|
61 |
scores = reranker.compute_score(
|
62 |
[
|
63 |
-
['
|
64 |
-
['
|
65 |
]
|
66 |
)
|
67 |
-
print(scores) # [
|
68 |
|
69 |
# You can map the scores into 0-1 by set "normalize=True", which will apply sigmoid function to the score
|
70 |
scores = reranker.compute_score(
|
71 |
[
|
72 |
-
['
|
73 |
-
['
|
74 |
],
|
75 |
normalize=True
|
76 |
)
|
77 |
-
print(scores) # [0.
|
78 |
```
|
79 |
|
80 |
### Using Huggingface transformers
|
|
|
14 |
- text: tỉnh nào có diện tích lớn nhất việt nam
|
15 |
output:
|
16 |
- label: nghệ an có diện tích lớn nhất việt nam
|
17 |
+
score: 0.99999
|
18 |
- label: bắc ninh có diện tích nhỏ nhất việt nam
|
19 |
+
score: 0.0001
|
20 |
---
|
21 |
|
22 |
# Reranker
|
|
|
50 |
reranker = FlagReranker('namdp-ptit/ViRanker',
|
51 |
use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation
|
52 |
|
53 |
+
score = reranker.compute_score(['ai là vị vua cuối cùng của việt nam', 'vua bảo đại là vị vua cuối cùng của nước ta'])
|
54 |
+
print(score) # 13.71875
|
55 |
|
56 |
# You can map the scores into 0-1 by set "normalize=True", which will apply sigmoid function to the score
|
57 |
+
score = reranker.compute_score(['ai là vị vua cuối cùng của việt nam', 'vua bảo đại là vị vua cuối cùng của nước ta'],
|
58 |
normalize=True)
|
59 |
+
print(score) # 0.99999889840464
|
60 |
|
61 |
scores = reranker.compute_score(
|
62 |
[
|
63 |
+
['ai là vị vua cuối cùng của việt nam', 'vua bảo đại là vị vua cuối cùng của nước ta'],
|
64 |
+
['ai là vị vua cuối cùng của việt nam', 'lý nam đế là vị vua đầu tiên của nước ta']
|
65 |
]
|
66 |
)
|
67 |
+
print(scores) # [13.7265625, -8.53125]
|
68 |
|
69 |
# You can map the scores into 0-1 by set "normalize=True", which will apply sigmoid function to the score
|
70 |
scores = reranker.compute_score(
|
71 |
[
|
72 |
+
['ai là vị vua cuối cùng của việt nam', 'vua bảo đại là vị vua cuối của nước ta'],
|
73 |
+
['ai là vị vua cuối cùng của việt nam', 'lý nam đế là vị vua đầu tiên của nước ta']
|
74 |
],
|
75 |
normalize=True
|
76 |
)
|
77 |
+
print(scores) # [0.99999889840464, 0.00019716942196222918]
|
78 |
```
|
79 |
|
80 |
### Using Huggingface transformers
|