Text Ranking
sentence-transformers
Safetensors
xlm-roberta
cross-encoder
reranker
Generated from Trainer
dataset_size:769572
loss:BinaryCrossEntropyLoss
Eval Results (legacy)
text-embeddings-inference
Instructions to use steerrec/bge-reranker-v2-m3-query-note with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use steerrec/bge-reranker-v2-m3-query-note with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("steerrec/bge-reranker-v2-m3-query-note") 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
Add new CrossEncoder model
Browse files- .gitattributes +1 -0
- README.md +549 -0
- config.json +33 -0
- config_sentence_transformers.json +11 -0
- model.safetensors +3 -0
- modules.json +8 -0
- sentence_bert_config.json +10 -0
- sentencepiece.bpe.model +3 -0
- special_tokens_map.json +51 -0
- tokenizer.json +3 -0
- tokenizer_config.json +56 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,549 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- sentence-transformers
|
| 4 |
+
- cross-encoder
|
| 5 |
+
- reranker
|
| 6 |
+
- generated_from_trainer
|
| 7 |
+
- dataset_size:769572
|
| 8 |
+
- loss:BinaryCrossEntropyLoss
|
| 9 |
+
base_model: BAAI/bge-reranker-v2-m3
|
| 10 |
+
pipeline_tag: text-ranking
|
| 11 |
+
library_name: sentence-transformers
|
| 12 |
+
metrics:
|
| 13 |
+
- map
|
| 14 |
+
- mrr@10
|
| 15 |
+
- ndcg@10
|
| 16 |
+
model-index:
|
| 17 |
+
- name: CrossEncoder based on BAAI/bge-reranker-v2-m3
|
| 18 |
+
results:
|
| 19 |
+
- task:
|
| 20 |
+
type: cross-encoder-reranking
|
| 21 |
+
name: Cross Encoder Reranking
|
| 22 |
+
dataset:
|
| 23 |
+
name: query note test
|
| 24 |
+
type: query_note_test
|
| 25 |
+
metrics:
|
| 26 |
+
- type: map
|
| 27 |
+
value: 0.45654587062161356
|
| 28 |
+
name: Map
|
| 29 |
+
- type: mrr@10
|
| 30 |
+
value: 0.5283685579312956
|
| 31 |
+
name: Mrr@10
|
| 32 |
+
- type: ndcg@10
|
| 33 |
+
value: 0.527108528024005
|
| 34 |
+
name: Ndcg@10
|
| 35 |
+
- task:
|
| 36 |
+
type: cross-encoder-reranking
|
| 37 |
+
name: Cross Encoder Reranking
|
| 38 |
+
dataset:
|
| 39 |
+
name: query note val
|
| 40 |
+
type: query_note_val
|
| 41 |
+
metrics:
|
| 42 |
+
- type: map
|
| 43 |
+
value: 0.45724051914340746
|
| 44 |
+
name: Map
|
| 45 |
+
- type: mrr@10
|
| 46 |
+
value: 0.5246903296465979
|
| 47 |
+
name: Mrr@10
|
| 48 |
+
- type: ndcg@10
|
| 49 |
+
value: 0.5328170850876233
|
| 50 |
+
name: Ndcg@10
|
| 51 |
+
---
|
| 52 |
+
|
| 53 |
+
# CrossEncoder based on BAAI/bge-reranker-v2-m3
|
| 54 |
+
|
| 55 |
+
This is a [Cross Encoder](https://www.sbert.net/docs/cross_encoder/usage/usage.html) model finetuned from [BAAI/bge-reranker-v2-m3](https://huggingface.co/BAAI/bge-reranker-v2-m3) using the [sentence-transformers](https://www.SBERT.net) library. It computes scores for pairs of texts, which can be used for text reranking and semantic search.
|
| 56 |
+
|
| 57 |
+
## Model Details
|
| 58 |
+
|
| 59 |
+
### Model Description
|
| 60 |
+
- **Model Type:** Cross Encoder
|
| 61 |
+
- **Base model:** [BAAI/bge-reranker-v2-m3](https://huggingface.co/BAAI/bge-reranker-v2-m3) <!-- at revision 953dc6f6f85a1b2dbfca4c34a2796e7dde08d41e -->
|
| 62 |
+
- **Maximum Sequence Length:** 512 tokens
|
| 63 |
+
- **Number of Output Labels:** 1 label
|
| 64 |
+
- **Supported Modality:** Text
|
| 65 |
+
<!-- - **Training Dataset:** Unknown -->
|
| 66 |
+
<!-- - **Language:** Unknown -->
|
| 67 |
+
<!-- - **License:** Unknown -->
|
| 68 |
+
|
| 69 |
+
### Model Sources
|
| 70 |
+
|
| 71 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 72 |
+
- **Documentation:** [Cross Encoder Documentation](https://www.sbert.net/docs/cross_encoder/usage/usage.html)
|
| 73 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
|
| 74 |
+
- **Hugging Face:** [Cross Encoders on Hugging Face](https://huggingface.co/models?library=sentence-transformers&other=cross-encoder)
|
| 75 |
+
|
| 76 |
+
### Full Model Architecture
|
| 77 |
+
|
| 78 |
+
```
|
| 79 |
+
CrossEncoder(
|
| 80 |
+
(0): Transformer({'transformer_task': 'sequence-classification', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'logits'}}, 'module_output_name': 'scores', 'architecture': 'XLMRobertaForSequenceClassification'})
|
| 81 |
+
)
|
| 82 |
+
```
|
| 83 |
+
|
| 84 |
+
## Usage
|
| 85 |
+
|
| 86 |
+
### Direct Usage (Sentence Transformers)
|
| 87 |
+
|
| 88 |
+
First install the Sentence Transformers library:
|
| 89 |
+
|
| 90 |
+
```bash
|
| 91 |
+
pip install -U sentence-transformers
|
| 92 |
+
```
|
| 93 |
+
|
| 94 |
+
Then you can load this model and run inference.
|
| 95 |
+
```python
|
| 96 |
+
from sentence_transformers import CrossEncoder
|
| 97 |
+
|
| 98 |
+
# Download from the 🤗 Hub
|
| 99 |
+
model = CrossEncoder("steerrec/bge-reranker-v2-m3-query-note")
|
| 100 |
+
# Get scores for pairs of inputs
|
| 101 |
+
pairs = [
|
| 102 |
+
['infp男和infj女', '谁懂啊啊啊‼️💗INFP×INFJ相处好戳我\n绿老头和小蝴蝶都是高敏易碎人群,消息秒回啊,朋友圈点赞啊,这种细微且偏爱的行为,都能让对方感到安心和减少内耗。\n小蝴蝶的话,有事情就不要藏着掖着,对绿老头也不要有过多的试探行为,有问题就直接问,直接说,只要把事情拖到情绪都上来了,那这问题就更难解决了。因为绿老头只是看起来淡然,但其实他们的内心是很火热躁动的,他们只是不好意思主动表达。绿老头得学会调整,明知道小蝴蝶容易想多内耗,那就多主动表现情绪,不要让小蝴蝶觉得你对他淡淡的,要是能偶尔打个只球,明确的表达一下内心,那就更好了。#心理[话题]# #mbti[话题]# #infp[话题]# #infp日常[话题]# #infp和infj[话题]# #INFJ[话题]# #infj的世界[话题]# #我的日常[话题]# #温暖治愈[话题]# '],
|
| 103 |
+
['电动车远光灯刺眼反击', '电动车在这些情况下是全责哦!注意了哦!\n骑电动车一定要小心这些情况哦![暗中观察R]#交通事故理赔[话题]# #交通事故[话题]# #法律求助[话题]# #电动车[话题]# #法律咨询[话题]# #交通安全[话题]# '],
|
| 104 |
+
['电动车远光灯刺眼反击', '支付宝上这个骑行险有用吗\n#车险[话题]# #电动车骑行险[话题]# #支付宝[话题]# '],
|
| 105 |
+
['牛奶品牌排名', '注意❗注意❗杯子到货,买就送📢\n杯子已经到货啦📣\n依然还是35.9到手2瓶4斤装的鲜奶,在额外赠送2个mini玻璃杯哦~\n顺丰冷链发货,包邮到家📦\n\xa0#乍甸牛奶[话题]#\xa0\xa0#云南游[话题]#\xa0\xa0#鲜奶[话题]#\xa0\xa0#可爱杯子[话题]#\xa0\xa0#杯子分享[话题]#\xa0\xa0#杯子控必入系列[话题]#\xa0\xa0#我就是个杯子控[话题]#\xa0\xa0#杯子[话题]#\xa0\xa0#鲜奶酸奶怎么挑[话��]#\xa0\xa0#鲜奶推荐[话题]#\xa0\xa0#鲜牛乳[话题]#\xa0\xa0#乍甸牛奶福利种草官[话题]#\xa0\xa0#乍甸牛奶也很好喝[话题]#\xa0\xa0#乍甸鲜奶[话题]#\xa0\t\n'],
|
| 106 |
+
['mbti人格', 'INFP小蝴蝶🦋请谨慎破防😅\n感觉本infp的确心灵上有些许脆弱,在面对朋友或者家人,别人的一句批评或者不认同,会影响我一天的心情~感觉这个习惯跟刻在骨子里一样,一边安慰自己,却一边焦虑😮\u200d💨很难想象有时候自己却很开朗乐观,其实内心很脆弱,一点就破⊙﹏⊙\n\t\n内容纯属娱乐🌚如有雷同纯属巧合🌚\n请大家对号入座哈哈哈哈😂#MBTI16型人格[话题]# #mbti梗图[话题]# #infp精神世界[话题]# #infp日常[话题]# '],
|
| 107 |
+
]
|
| 108 |
+
scores = model.predict(pairs)
|
| 109 |
+
print(scores)
|
| 110 |
+
# [0.3757 0.2134 0.1656 0.179 0.272 ]
|
| 111 |
+
|
| 112 |
+
# Or rank different texts based on similarity to a single text
|
| 113 |
+
ranks = model.rank(
|
| 114 |
+
'infp男和infj女',
|
| 115 |
+
[
|
| 116 |
+
'谁懂啊啊啊‼️💗INFP×INFJ相处好戳我\n绿老头和小蝴蝶都是高敏易碎人群,消息秒回啊,朋友圈点赞啊,这种细微且偏爱的行为,都能让对方感到安心和减少内耗。\n小蝴蝶的话,有事情就不要藏着掖着,对绿老头也不要有过多的试探行为,有问题就直接问,直接说,只要把事情拖到情绪都上来了,那这问题就更难解决了。因为绿老头只是看起来淡然,但其实他们的内心是很火热躁动的,他们只是不好意思主动表达。绿老头得学会调整,明知道小蝴蝶容易想多内耗,那就多主动表现情绪,不要让小蝴蝶觉得你对他淡淡的,要是能偶尔打个只球,明确的表达一下内心,那就更好了。#心理[话题]# #mbti[话题]# #infp[话题]# #infp日常[话题]# #infp和infj[话题]# #INFJ[话题]# #infj的世界[话题]# #我的日常[话题]# #温暖治愈[话题]# ',
|
| 117 |
+
'电动车在这些情况下是全责哦!注意了哦!\n骑电动车一定要小心这些情况哦![暗中观察R]#交通事故理赔[话题]# #交通事故[话题]# #法律求助[话题]# #电动车[话题]# #法律咨询[话题]# #交通安全[话题]# ',
|
| 118 |
+
'支付宝上这个骑行险有用吗\n#车险[话题]# #电动车骑行险[话题]# #支付宝[话题]# ',
|
| 119 |
+
'注意❗注意❗杯子到货,买就送📢\n杯子已经到货啦📣\n依然还是35.9到手2瓶4斤装的鲜奶,在额外赠送2个mini玻璃杯哦~\n顺丰冷链发货,包邮到家📦\n\xa0#乍甸牛奶[话题]#\xa0\xa0#云南游[话题]#\xa0\xa0#鲜奶[话题]#\xa0\xa0#可爱杯子[话题]#\xa0\xa0#杯子分享[话题]#\xa0\xa0#杯子控必入系列[话题]#\xa0\xa0#我就是个杯子控[话题]#\xa0\xa0#杯子[话题]#\xa0\xa0#鲜奶酸奶怎么挑[话题]#\xa0\xa0#鲜奶推荐[话题]#\xa0\xa0#鲜牛乳[话题]#\xa0\xa0#乍甸牛奶福利种草官[话题]#\xa0\xa0#乍甸牛奶也很好喝[话题]#\xa0\xa0#乍甸鲜奶[话题]#\xa0\t\n',
|
| 120 |
+
'INFP小蝴蝶🦋请谨慎破防😅\n感觉本infp的确心灵上有些许脆弱,在面对朋友或者家人,别人的一句批评或者不认同,会影响我一天的心情~感觉这个习惯跟刻在骨子里一样,一边安慰自己,却一边焦虑😮\u200d💨很难想象有时候自己却很开朗乐观,其实内心很脆弱,一点就破⊙﹏⊙\n\t\n内容纯属娱乐🌚如有雷同纯属巧合🌚\n请大家对号入座哈哈哈哈😂#MBTI16型人格[话题]# #mbti梗图[话题]# #infp精神世界[话题]# #infp日常[话题]# ',
|
| 121 |
+
]
|
| 122 |
+
)
|
| 123 |
+
# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]
|
| 124 |
+
```
|
| 125 |
+
|
| 126 |
+
<!--
|
| 127 |
+
### Direct Usage (Transformers)
|
| 128 |
+
|
| 129 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 130 |
+
|
| 131 |
+
</details>
|
| 132 |
+
-->
|
| 133 |
+
|
| 134 |
+
<!--
|
| 135 |
+
### Downstream Usage (Sentence Transformers)
|
| 136 |
+
|
| 137 |
+
You can finetune this model on your own dataset.
|
| 138 |
+
|
| 139 |
+
<details><summary>Click to expand</summary>
|
| 140 |
+
|
| 141 |
+
</details>
|
| 142 |
+
-->
|
| 143 |
+
|
| 144 |
+
<!--
|
| 145 |
+
### Out-of-Scope Use
|
| 146 |
+
|
| 147 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 148 |
+
-->
|
| 149 |
+
|
| 150 |
+
## Evaluation
|
| 151 |
+
|
| 152 |
+
### Metrics
|
| 153 |
+
|
| 154 |
+
#### Cross Encoder Reranking
|
| 155 |
+
|
| 156 |
+
* Datasets: `query_note_test` and `query_note_val`
|
| 157 |
+
* Evaluated with [<code>CrossEncoderRerankingEvaluator</code>](https://sbert.net/docs/package_reference/cross_encoder/evaluation.html#sentence_transformers.cross_encoder.evaluation.CrossEncoderRerankingEvaluator) with these parameters:
|
| 158 |
+
```json
|
| 159 |
+
{
|
| 160 |
+
"at_k": 10
|
| 161 |
+
}
|
| 162 |
+
```
|
| 163 |
+
|
| 164 |
+
| Metric | query_note_test | query_note_val |
|
| 165 |
+
|:------------|:----------------|:---------------|
|
| 166 |
+
| map | 0.4565 | 0.4572 |
|
| 167 |
+
| mrr@10 | 0.5284 | 0.5247 |
|
| 168 |
+
| **ndcg@10** | **0.5271** | **0.5328** |
|
| 169 |
+
|
| 170 |
+
<!--
|
| 171 |
+
## Bias, Risks and Limitations
|
| 172 |
+
|
| 173 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 174 |
+
-->
|
| 175 |
+
|
| 176 |
+
<!--
|
| 177 |
+
### Recommendations
|
| 178 |
+
|
| 179 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 180 |
+
-->
|
| 181 |
+
|
| 182 |
+
## Training Details
|
| 183 |
+
|
| 184 |
+
### Training Dataset
|
| 185 |
+
|
| 186 |
+
#### Unnamed Dataset
|
| 187 |
+
|
| 188 |
+
* Size: 769,572 training samples
|
| 189 |
+
* Columns: <code>sentence1</code>, <code>sentence2</code>, and <code>label</code>
|
| 190 |
+
* Approximate statistics based on the first 100 samples:
|
| 191 |
+
| | sentence1 | sentence2 | label |
|
| 192 |
+
|:---------|:---------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:--------------------------------------------------------------|
|
| 193 |
+
| type | string | string | float |
|
| 194 |
+
| modality | text | text | |
|
| 195 |
+
| details | <ul><li>min: 5 tokens</li><li>mean: 7.92 tokens</li><li>max: 13 tokens</li></ul> | <ul><li>min: 18 tokens</li><li>mean: 208.42 tokens</li><li>max: 512 tokens</li></ul> | <ul><li>min: 0.0</li><li>mean: 0.0</li><li>max: 0.0</li></ul> |
|
| 196 |
+
* Samples:
|
| 197 |
+
| sentence1 | sentence2 | label |
|
| 198 |
+
|:-------------------------|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:-----------------|
|
| 199 |
+
| <code>infp男和infj女</code> | <code>谁懂啊啊啊‼️💗INFP×INFJ相处好戳我<br>绿老头和小蝴蝶都是高敏易碎人群,消息秒回啊,朋友圈点赞啊,这种细微且偏爱的行为,都能让对方感到安心和减少内耗。<br>小蝴蝶的话,有事情就不要藏着掖着,对绿老头也不要有过多的试探行为,有问题就直接问,直接说,只要把事情拖到情绪都上来了,那这问题就更难解决了。因为绿老头只是看起来淡然,但其实他们的内心是很火热躁动的,他们只是不好意思主动表达。绿老头得学会调整,明知道小蝴蝶容易想多内耗,那就多主动表现情绪,不要让小蝴蝶觉得你对他淡淡的,要是能偶尔打个只球,明确的表达一下内心,那就更好了。#心理[话题]# #mbti[话题]# #infp[话题]# #infp日常[话题]# #infp和infj[话题]# #INFJ[话题]# #infj的世界[话题]# #我的日常[话题]# #温暖治愈[话题]# </code> | <code>0.0</code> |
|
| 200 |
+
| <code>电动车远光灯刺眼反击</code> | <code>电动车在这些情况下是全责哦!注意了哦!<br>骑电动车一定要小心这些情况哦![暗中观察R]#交通事故理赔[话题]# #交通事故[话题]# #法律求助[话题]# #电动车[话题]# #法律咨询[话题]# #交通安全[话题]# </code> | <code>0.0</code> |
|
| 201 |
+
| <code>电动车远光灯刺眼反击</code> | <code>支付宝上这个骑行险有用吗<br>#车险[话题]# #电动车骑行险[话题]# #支付宝[话题]# </code> | <code>0.0</code> |
|
| 202 |
+
* Loss: [<code>BinaryCrossEntropyLoss</code>](https://sbert.net/docs/package_reference/cross_encoder/losses.html#binarycrossentropyloss) with these parameters:
|
| 203 |
+
```json
|
| 204 |
+
{
|
| 205 |
+
"activation_fn": "torch.nn.modules.linear.Identity",
|
| 206 |
+
"pos_weight": null
|
| 207 |
+
}
|
| 208 |
+
```
|
| 209 |
+
|
| 210 |
+
### Training Hyperparameters
|
| 211 |
+
#### Non-Default Hyperparameters
|
| 212 |
+
|
| 213 |
+
- `per_device_train_batch_size`: 32
|
| 214 |
+
- `learning_rate`: 0.0001
|
| 215 |
+
- `weight_decay`: 0.01
|
| 216 |
+
- `num_train_epochs`: 1
|
| 217 |
+
- `lr_scheduler_type`: cosine
|
| 218 |
+
- `warmup_ratio`: 0.05
|
| 219 |
+
- `seed`: 3407
|
| 220 |
+
- `bf16`: True
|
| 221 |
+
|
| 222 |
+
#### All Hyperparameters
|
| 223 |
+
<details><summary>Click to expand</summary>
|
| 224 |
+
|
| 225 |
+
- `overwrite_output_dir`: False
|
| 226 |
+
- `do_predict`: False
|
| 227 |
+
- `prediction_loss_only`: True
|
| 228 |
+
- `per_device_train_batch_size`: 32
|
| 229 |
+
- `per_device_eval_batch_size`: 8
|
| 230 |
+
- `per_gpu_train_batch_size`: None
|
| 231 |
+
- `per_gpu_eval_batch_size`: None
|
| 232 |
+
- `gradient_accumulation_steps`: 1
|
| 233 |
+
- `eval_accumulation_steps`: None
|
| 234 |
+
- `torch_empty_cache_steps`: None
|
| 235 |
+
- `learning_rate`: 0.0001
|
| 236 |
+
- `weight_decay`: 0.01
|
| 237 |
+
- `adam_beta1`: 0.9
|
| 238 |
+
- `adam_beta2`: 0.999
|
| 239 |
+
- `adam_epsilon`: 1e-08
|
| 240 |
+
- `max_grad_norm`: 1.0
|
| 241 |
+
- `num_train_epochs`: 1
|
| 242 |
+
- `max_steps`: -1
|
| 243 |
+
- `lr_scheduler_type`: cosine
|
| 244 |
+
- `lr_scheduler_kwargs`: {}
|
| 245 |
+
- `warmup_ratio`: 0.05
|
| 246 |
+
- `warmup_steps`: 0
|
| 247 |
+
- `log_level`: passive
|
| 248 |
+
- `log_level_replica`: warning
|
| 249 |
+
- `log_on_each_node`: True
|
| 250 |
+
- `logging_nan_inf_filter`: True
|
| 251 |
+
- `save_safetensors`: True
|
| 252 |
+
- `save_on_each_node`: False
|
| 253 |
+
- `save_only_model`: False
|
| 254 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 255 |
+
- `no_cuda`: False
|
| 256 |
+
- `use_cpu`: False
|
| 257 |
+
- `use_mps_device`: False
|
| 258 |
+
- `seed`: 3407
|
| 259 |
+
- `data_seed`: None
|
| 260 |
+
- `jit_mode_eval`: False
|
| 261 |
+
- `use_ipex`: False
|
| 262 |
+
- `bf16`: True
|
| 263 |
+
- `fp16`: False
|
| 264 |
+
- `fp16_opt_level`: O1
|
| 265 |
+
- `half_precision_backend`: auto
|
| 266 |
+
- `bf16_full_eval`: False
|
| 267 |
+
- `fp16_full_eval`: False
|
| 268 |
+
- `tf32`: None
|
| 269 |
+
- `local_rank`: 0
|
| 270 |
+
- `ddp_backend`: None
|
| 271 |
+
- `tpu_num_cores`: None
|
| 272 |
+
- `tpu_metrics_debug`: False
|
| 273 |
+
- `debug`: []
|
| 274 |
+
- `dataloader_drop_last`: False
|
| 275 |
+
- `dataloader_num_workers`: 0
|
| 276 |
+
- `dataloader_prefetch_factor`: None
|
| 277 |
+
- `past_index`: -1
|
| 278 |
+
- `disable_tqdm`: False
|
| 279 |
+
- `remove_unused_columns`: True
|
| 280 |
+
- `label_names`: None
|
| 281 |
+
- `load_best_model_at_end`: False
|
| 282 |
+
- `ignore_data_skip`: False
|
| 283 |
+
- `fsdp`: []
|
| 284 |
+
- `fsdp_min_num_params`: 0
|
| 285 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 286 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
| 287 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 288 |
+
- `parallelism_config`: None
|
| 289 |
+
- `deepspeed`: None
|
| 290 |
+
- `label_smoothing_factor`: 0.0
|
| 291 |
+
- `optim`: adamw_torch_fused
|
| 292 |
+
- `optim_args`: None
|
| 293 |
+
- `adafactor`: False
|
| 294 |
+
- `group_by_length`: False
|
| 295 |
+
- `length_column_name`: length
|
| 296 |
+
- `ddp_find_unused_parameters`: None
|
| 297 |
+
- `ddp_bucket_cap_mb`: None
|
| 298 |
+
- `ddp_broadcast_buffers`: False
|
| 299 |
+
- `dataloader_pin_memory`: True
|
| 300 |
+
- `dataloader_persistent_workers`: False
|
| 301 |
+
- `skip_memory_metrics`: True
|
| 302 |
+
- `use_legacy_prediction_loop`: False
|
| 303 |
+
- `push_to_hub`: False
|
| 304 |
+
- `resume_from_checkpoint`: None
|
| 305 |
+
- `hub_model_id`: None
|
| 306 |
+
- `hub_strategy`: every_save
|
| 307 |
+
- `hub_private_repo`: None
|
| 308 |
+
- `hub_always_push`: False
|
| 309 |
+
- `hub_revision`: None
|
| 310 |
+
- `gradient_checkpointing`: False
|
| 311 |
+
- `gradient_checkpointing_kwargs`: None
|
| 312 |
+
- `include_inputs_for_metrics`: False
|
| 313 |
+
- `include_for_metrics`: []
|
| 314 |
+
- `eval_do_concat_batches`: True
|
| 315 |
+
- `fp16_backend`: auto
|
| 316 |
+
- `push_to_hub_model_id`: None
|
| 317 |
+
- `push_to_hub_organization`: None
|
| 318 |
+
- `mp_parameters`:
|
| 319 |
+
- `auto_find_batch_size`: False
|
| 320 |
+
- `full_determinism`: False
|
| 321 |
+
- `torchdynamo`: None
|
| 322 |
+
- `ray_scope`: last
|
| 323 |
+
- `ddp_timeout`: 1800
|
| 324 |
+
- `torch_compile`: False
|
| 325 |
+
- `torch_compile_backend`: None
|
| 326 |
+
- `torch_compile_mode`: None
|
| 327 |
+
- `include_tokens_per_second`: False
|
| 328 |
+
- `include_num_input_tokens_seen`: False
|
| 329 |
+
- `neftune_noise_alpha`: None
|
| 330 |
+
- `optim_target_modules`: None
|
| 331 |
+
- `batch_eval_metrics`: False
|
| 332 |
+
- `eval_on_start`: False
|
| 333 |
+
- `use_liger_kernel`: False
|
| 334 |
+
- `liger_kernel_config`: None
|
| 335 |
+
- `eval_use_gather_object`: False
|
| 336 |
+
- `average_tokens_across_devices`: False
|
| 337 |
+
- `prompts`: None
|
| 338 |
+
- `batch_sampler`: batch_sampler
|
| 339 |
+
- `multi_dataset_batch_sampler`: proportional
|
| 340 |
+
- `router_mapping`: {}
|
| 341 |
+
- `learning_rate_mapping`: {}
|
| 342 |
+
|
| 343 |
+
</details>
|
| 344 |
+
|
| 345 |
+
### Training Logs
|
| 346 |
+
<details><summary>Click to expand</summary>
|
| 347 |
+
|
| 348 |
+
| Epoch | Step | Training Loss | query_note_test_ndcg@10 | query_note_val_ndcg@10 |
|
| 349 |
+
|:------:|:-----:|:-------------:|:-----------------------:|:----------------------:|
|
| 350 |
+
| -1 | -1 | - | 0.5206 | - |
|
| 351 |
+
| 0.7048 | 16950 | 0.6932 | - | - |
|
| 352 |
+
| 0.7069 | 17000 | 0.5323 | - | - |
|
| 353 |
+
| 0.7089 | 17050 | 0.5209 | - | - |
|
| 354 |
+
| 0.7110 | 17100 | 0.5281 | - | - |
|
| 355 |
+
| 0.7131 | 17150 | 0.5267 | - | - |
|
| 356 |
+
| 0.7152 | 17200 | 0.5124 | - | - |
|
| 357 |
+
| 0.7173 | 17250 | 0.5199 | - | - |
|
| 358 |
+
| 0.7193 | 17300 | 0.5189 | - | - |
|
| 359 |
+
| 0.7214 | 17350 | 0.5059 | - | - |
|
| 360 |
+
| 0.7235 | 17400 | 0.5326 | - | - |
|
| 361 |
+
| 0.7256 | 17450 | 0.5159 | - | - |
|
| 362 |
+
| 0.7277 | 17500 | 0.5246 | - | - |
|
| 363 |
+
| 0.7297 | 17550 | 0.5128 | - | - |
|
| 364 |
+
| 0.7318 | 17600 | 0.5078 | - | - |
|
| 365 |
+
| 0.7339 | 17650 | 0.4966 | - | - |
|
| 366 |
+
| 0.7360 | 17700 | 0.506 | - | - |
|
| 367 |
+
| 0.7380 | 17750 | 0.499 | - | - |
|
| 368 |
+
| 0.7401 | 17800 | 0.5069 | - | - |
|
| 369 |
+
| 0.7422 | 17850 | 0.5387 | - | - |
|
| 370 |
+
| 0.7443 | 17900 | 0.5124 | - | - |
|
| 371 |
+
| 0.7464 | 17950 | 0.522 | - | - |
|
| 372 |
+
| 0.7484 | 18000 | 0.5103 | - | - |
|
| 373 |
+
| 0.7505 | 18050 | 0.5217 | - | - |
|
| 374 |
+
| 0.7526 | 18100 | 0.4939 | - | - |
|
| 375 |
+
| 0.7547 | 18150 | 0.5151 | - | - |
|
| 376 |
+
| 0.7568 | 18200 | 0.4804 | - | - |
|
| 377 |
+
| 0.7588 | 18250 | 0.4969 | - | - |
|
| 378 |
+
| 0.7609 | 18300 | 0.5277 | - | - |
|
| 379 |
+
| 0.7630 | 18350 | 0.5143 | - | - |
|
| 380 |
+
| 0.7651 | 18400 | 0.5063 | - | - |
|
| 381 |
+
| 0.7672 | 18450 | 0.4899 | - | - |
|
| 382 |
+
| 0.7692 | 18500 | 0.5144 | - | - |
|
| 383 |
+
| 0.7713 | 18550 | 0.528 | - | - |
|
| 384 |
+
| 0.7734 | 18600 | 0.5032 | - | - |
|
| 385 |
+
| 0.7755 | 18650 | 0.4956 | - | - |
|
| 386 |
+
| 0.7775 | 18700 | 0.5144 | - | - |
|
| 387 |
+
| 0.7796 | 18750 | 0.5145 | - | - |
|
| 388 |
+
| 0.7817 | 18800 | 0.4971 | - | - |
|
| 389 |
+
| 0.7838 | 18850 | 0.5188 | - | - |
|
| 390 |
+
| 0.7859 | 18900 | 0.501 | - | - |
|
| 391 |
+
| 0.7879 | 18950 | 0.4892 | - | - |
|
| 392 |
+
| 0.7900 | 19000 | 0.4752 | - | - |
|
| 393 |
+
| 0.7921 | 19050 | 0.4984 | - | - |
|
| 394 |
+
| 0.7942 | 19100 | 0.5001 | - | - |
|
| 395 |
+
| 0.7963 | 19150 | 0.4809 | - | - |
|
| 396 |
+
| 0.7983 | 19200 | 0.5085 | - | - |
|
| 397 |
+
| 0.8004 | 19250 | 0.5122 | - | - |
|
| 398 |
+
| 0.8025 | 19300 | 0.5122 | - | - |
|
| 399 |
+
| 0.8046 | 19350 | 0.4909 | - | - |
|
| 400 |
+
| 0.8067 | 19400 | 0.5341 | - | - |
|
| 401 |
+
| 0.8087 | 19450 | 0.5147 | - | - |
|
| 402 |
+
| 0.8108 | 19500 | 0.5095 | - | - |
|
| 403 |
+
| 0.8129 | 19550 | 0.4945 | - | - |
|
| 404 |
+
| 0.8150 | 19600 | 0.4971 | - | - |
|
| 405 |
+
| 0.8170 | 19650 | 0.4967 | - | - |
|
| 406 |
+
| 0.8191 | 19700 | 0.5108 | - | - |
|
| 407 |
+
| 0.8212 | 19750 | 0.4983 | - | - |
|
| 408 |
+
| 0.8233 | 19800 | 0.5154 | - | - |
|
| 409 |
+
| 0.8254 | 19850 | 0.5214 | - | - |
|
| 410 |
+
| 0.8274 | 19900 | 0.4953 | - | - |
|
| 411 |
+
| 0.8295 | 19950 | 0.5079 | - | - |
|
| 412 |
+
| 0.8316 | 20000 | 0.5252 | - | - |
|
| 413 |
+
| 0.8337 | 20050 | 0.4966 | - | - |
|
| 414 |
+
| 0.8358 | 20100 | 0.492 | - | - |
|
| 415 |
+
| 0.8378 | 20150 | 0.5065 | - | - |
|
| 416 |
+
| 0.8399 | 20200 | 0.4825 | - | - |
|
| 417 |
+
| 0.8420 | 20250 | 0.4879 | - | - |
|
| 418 |
+
| 0.8441 | 20300 | 0.5351 | - | - |
|
| 419 |
+
| 0.8462 | 20350 | 0.4904 | - | - |
|
| 420 |
+
| 0.8482 | 20400 | 0.5141 | - | - |
|
| 421 |
+
| 0.8503 | 20450 | 0.5146 | - | - |
|
| 422 |
+
| 0.8524 | 20500 | 0.508 | - | - |
|
| 423 |
+
| 0.8545 | 20550 | 0.5271 | - | - |
|
| 424 |
+
| 0.8565 | 20600 | 0.5057 | - | - |
|
| 425 |
+
| 0.8586 | 20650 | 0.4757 | - | - |
|
| 426 |
+
| 0.8607 | 20700 | 0.5151 | - | - |
|
| 427 |
+
| 0.8628 | 20750 | 0.486 | - | - |
|
| 428 |
+
| 0.8649 | 20800 | 0.4908 | - | - |
|
| 429 |
+
| 0.8669 | 20850 | 0.5287 | - | - |
|
| 430 |
+
| 0.8690 | 20900 | 0.5223 | - | - |
|
| 431 |
+
| 0.8711 | 20950 | 0.5086 | - | - |
|
| 432 |
+
| 0.8732 | 21000 | 0.5066 | - | - |
|
| 433 |
+
| 0.8753 | 21050 | 0.5042 | - | - |
|
| 434 |
+
| 0.8773 | 21100 | 0.5032 | - | - |
|
| 435 |
+
| 0.8794 | 21150 | 0.5123 | - | - |
|
| 436 |
+
| 0.8815 | 21200 | 0.4825 | - | - |
|
| 437 |
+
| 0.8836 | 21250 | 0.5222 | - | - |
|
| 438 |
+
| 0.8857 | 21300 | 0.5044 | - | - |
|
| 439 |
+
| 0.8877 | 21350 | 0.5034 | - | - |
|
| 440 |
+
| 0.8898 | 21400 | 0.5193 | - | - |
|
| 441 |
+
| 0.8919 | 21450 | 0.4975 | - | - |
|
| 442 |
+
| 0.8940 | 21500 | 0.4754 | - | - |
|
| 443 |
+
| 0.8960 | 21550 | 0.5209 | - | - |
|
| 444 |
+
| 0.8981 | 21600 | 0.5024 | - | - |
|
| 445 |
+
| 0.9002 | 21650 | 0.5206 | - | - |
|
| 446 |
+
| 0.9023 | 21700 | 0.5032 | - | - |
|
| 447 |
+
| 0.9044 | 21750 | 0.5264 | - | - |
|
| 448 |
+
| 0.9064 | 21800 | 0.499 | - | - |
|
| 449 |
+
| 0.9085 | 21850 | 0.4967 | - | - |
|
| 450 |
+
| 0.9106 | 21900 | 0.491 | - | - |
|
| 451 |
+
| 0.9127 | 21950 | 0.5056 | - | - |
|
| 452 |
+
| 0.9148 | 22000 | 0.4996 | - | - |
|
| 453 |
+
| 0.9168 | 22050 | 0.4994 | - | - |
|
| 454 |
+
| 0.9189 | 22100 | 0.5254 | - | - |
|
| 455 |
+
| 0.9210 | 22150 | 0.5034 | - | - |
|
| 456 |
+
| 0.9231 | 22200 | 0.5123 | - | - |
|
| 457 |
+
| 0.9252 | 22250 | 0.4956 | - | - |
|
| 458 |
+
| 0.9272 | 22300 | 0.5194 | - | - |
|
| 459 |
+
| 0.9293 | 22350 | 0.474 | - | - |
|
| 460 |
+
| 0.9314 | 22400 | 0.4842 | - | - |
|
| 461 |
+
| 0.9335 | 22450 | 0.4914 | - | - |
|
| 462 |
+
| 0.9356 | 22500 | 0.4925 | - | - |
|
| 463 |
+
| 0.9376 | 22550 | 0.4938 | - | - |
|
| 464 |
+
| 0.9397 | 22600 | 0.5086 | - | - |
|
| 465 |
+
| 0.9418 | 22650 | 0.457 | - | - |
|
| 466 |
+
| 0.9439 | 22700 | 0.5185 | - | - |
|
| 467 |
+
| 0.9459 | 22750 | 0.5268 | - | - |
|
| 468 |
+
| 0.9480 | 22800 | 0.4872 | - | - |
|
| 469 |
+
| 0.9501 | 22850 | 0.5048 | - | - |
|
| 470 |
+
| 0.9522 | 22900 | 0.5103 | - | - |
|
| 471 |
+
| 0.9543 | 22950 | 0.5236 | - | - |
|
| 472 |
+
| 0.9563 | 23000 | 0.5049 | - | - |
|
| 473 |
+
| 0.9584 | 23050 | 0.5041 | - | - |
|
| 474 |
+
| 0.9605 | 23100 | 0.5066 | - | - |
|
| 475 |
+
| 0.9626 | 23150 | 0.5206 | - | - |
|
| 476 |
+
| 0.9647 | 23200 | 0.4732 | - | - |
|
| 477 |
+
| 0.9667 | 23250 | 0.4881 | - | - |
|
| 478 |
+
| 0.9688 | 23300 | 0.5099 | - | - |
|
| 479 |
+
| 0.9709 | 23350 | 0.5226 | - | - |
|
| 480 |
+
| 0.9730 | 23400 | 0.5322 | - | - |
|
| 481 |
+
| 0.9751 | 23450 | 0.4993 | - | - |
|
| 482 |
+
| 0.9771 | 23500 | 0.4856 | - | - |
|
| 483 |
+
| 0.9792 | 23550 | 0.4727 | - | - |
|
| 484 |
+
| 0.9813 | 23600 | 0.5093 | - | - |
|
| 485 |
+
| 0.9834 | 23650 | 0.5073 | - | - |
|
| 486 |
+
| 0.9854 | 23700 | 0.5153 | - | - |
|
| 487 |
+
| 0.9875 | 23750 | 0.4979 | - | - |
|
| 488 |
+
| 0.9896 | 23800 | 0.4961 | - | - |
|
| 489 |
+
| 0.9917 | 23850 | 0.5093 | - | - |
|
| 490 |
+
| 0.9938 | 23900 | 0.4811 | - | - |
|
| 491 |
+
| 0.9958 | 23950 | 0.5008 | - | - |
|
| 492 |
+
| 0.9979 | 24000 | 0.5151 | - | - |
|
| 493 |
+
| 1.0 | 24050 | 0.5318 | - | 0.5328 |
|
| 494 |
+
| -1 | -1 | - | 0.5271 | - |
|
| 495 |
+
|
| 496 |
+
</details>
|
| 497 |
+
|
| 498 |
+
### Training Time
|
| 499 |
+
- **Training**: 57.2 minutes
|
| 500 |
+
|
| 501 |
+
### Framework Versions
|
| 502 |
+
- Python: 3.12.3
|
| 503 |
+
- Sentence Transformers: 5.5.1
|
| 504 |
+
- Transformers: 4.56.2
|
| 505 |
+
- PyTorch: 2.11.0+cu128
|
| 506 |
+
- Accelerate: 1.13.0
|
| 507 |
+
- Datasets: 4.3.0
|
| 508 |
+
- Tokenizers: 0.22.2
|
| 509 |
+
|
| 510 |
+
## Additional Resources
|
| 511 |
+
|
| 512 |
+
- [Training and Finetuning Reranker Models with Sentence Transformers](https://huggingface.co/blog/train-reranker): the end-to-end guide for training or finetuning Cross Encoder (reranker) models.
|
| 513 |
+
- [Multimodal Embedding & Reranker Models with Sentence Transformers](https://huggingface.co/blog/multimodal-sentence-transformers): use text, image, audio, and video reranker models through the same API.
|
| 514 |
+
- [Training and Finetuning Multimodal Embedding & Reranker Models with Sentence Transformers](https://huggingface.co/blog/train-multimodal-sentence-transformers): training multimodal Cross Encoders.
|
| 515 |
+
|
| 516 |
+
## Citation
|
| 517 |
+
|
| 518 |
+
### BibTeX
|
| 519 |
+
|
| 520 |
+
#### Sentence Transformers
|
| 521 |
+
```bibtex
|
| 522 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 523 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 524 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 525 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 526 |
+
month = "11",
|
| 527 |
+
year = "2019",
|
| 528 |
+
publisher = "Association for Computational Linguistics",
|
| 529 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 530 |
+
}
|
| 531 |
+
```
|
| 532 |
+
|
| 533 |
+
<!--
|
| 534 |
+
## Glossary
|
| 535 |
+
|
| 536 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 537 |
+
-->
|
| 538 |
+
|
| 539 |
+
<!--
|
| 540 |
+
## Model Card Authors
|
| 541 |
+
|
| 542 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 543 |
+
-->
|
| 544 |
+
|
| 545 |
+
<!--
|
| 546 |
+
## Model Card Contact
|
| 547 |
+
|
| 548 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 549 |
+
-->
|
config.json
ADDED
|
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"XLMRobertaForSequenceClassification"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"bos_token_id": 0,
|
| 7 |
+
"classifier_dropout": null,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 2,
|
| 10 |
+
"hidden_act": "gelu",
|
| 11 |
+
"hidden_dropout_prob": 0.1,
|
| 12 |
+
"hidden_size": 1024,
|
| 13 |
+
"id2label": {
|
| 14 |
+
"0": "LABEL_0"
|
| 15 |
+
},
|
| 16 |
+
"initializer_range": 0.02,
|
| 17 |
+
"intermediate_size": 4096,
|
| 18 |
+
"label2id": {
|
| 19 |
+
"LABEL_0": 0
|
| 20 |
+
},
|
| 21 |
+
"layer_norm_eps": 1e-05,
|
| 22 |
+
"max_position_embeddings": 8194,
|
| 23 |
+
"model_type": "xlm-roberta",
|
| 24 |
+
"num_attention_heads": 16,
|
| 25 |
+
"num_hidden_layers": 24,
|
| 26 |
+
"output_past": true,
|
| 27 |
+
"pad_token_id": 1,
|
| 28 |
+
"position_embedding_type": "absolute",
|
| 29 |
+
"transformers_version": "4.56.2",
|
| 30 |
+
"type_vocab_size": 1,
|
| 31 |
+
"use_cache": true,
|
| 32 |
+
"vocab_size": 250002
|
| 33 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"pytorch": "2.11.0+cu128",
|
| 4 |
+
"sentence_transformers": "5.5.1",
|
| 5 |
+
"transformers": "4.56.2"
|
| 6 |
+
},
|
| 7 |
+
"activation_fn": "torch.nn.modules.activation.Sigmoid",
|
| 8 |
+
"default_prompt_name": null,
|
| 9 |
+
"model_type": "CrossEncoder",
|
| 10 |
+
"prompts": {}
|
| 11 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:965ad8042c9f8d96bb7406e42110d3d5579e4f3c811977191f5fd77d6f24ad2d
|
| 3 |
+
size 2271071852
|
modules.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.base.modules.transformer.Transformer"
|
| 7 |
+
}
|
| 8 |
+
]
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"transformer_task": "sequence-classification",
|
| 3 |
+
"modality_config": {
|
| 4 |
+
"text": {
|
| 5 |
+
"method": "forward",
|
| 6 |
+
"method_output_name": "logits"
|
| 7 |
+
}
|
| 8 |
+
},
|
| 9 |
+
"module_output_name": "scores"
|
| 10 |
+
}
|
sentencepiece.bpe.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
|
| 3 |
+
size 5069051
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token": {
|
| 3 |
+
"content": "<s>",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"cls_token": {
|
| 10 |
+
"content": "<s>",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"eos_token": {
|
| 17 |
+
"content": "</s>",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"mask_token": {
|
| 24 |
+
"content": "<mask>",
|
| 25 |
+
"lstrip": true,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"pad_token": {
|
| 31 |
+
"content": "<pad>",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
},
|
| 37 |
+
"sep_token": {
|
| 38 |
+
"content": "</s>",
|
| 39 |
+
"lstrip": false,
|
| 40 |
+
"normalized": false,
|
| 41 |
+
"rstrip": false,
|
| 42 |
+
"single_word": false
|
| 43 |
+
},
|
| 44 |
+
"unk_token": {
|
| 45 |
+
"content": "<unk>",
|
| 46 |
+
"lstrip": false,
|
| 47 |
+
"normalized": false,
|
| 48 |
+
"rstrip": false,
|
| 49 |
+
"single_word": false
|
| 50 |
+
}
|
| 51 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf44dabfaa82b1276a7af64a2ea2c76c047d560cf7bfb5711d6135382372c93d
|
| 3 |
+
size 17083153
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"added_tokens_decoder": {
|
| 3 |
+
"0": {
|
| 4 |
+
"content": "<s>",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false,
|
| 9 |
+
"special": true
|
| 10 |
+
},
|
| 11 |
+
"1": {
|
| 12 |
+
"content": "<pad>",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"2": {
|
| 20 |
+
"content": "</s>",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"3": {
|
| 28 |
+
"content": "<unk>",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"250001": {
|
| 36 |
+
"content": "<mask>",
|
| 37 |
+
"lstrip": true,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"bos_token": "<s>",
|
| 45 |
+
"clean_up_tokenization_spaces": true,
|
| 46 |
+
"cls_token": "<s>",
|
| 47 |
+
"eos_token": "</s>",
|
| 48 |
+
"extra_special_tokens": {},
|
| 49 |
+
"mask_token": "<mask>",
|
| 50 |
+
"model_max_length": 512,
|
| 51 |
+
"pad_token": "<pad>",
|
| 52 |
+
"sep_token": "</s>",
|
| 53 |
+
"sp_model_kwargs": {},
|
| 54 |
+
"tokenizer_class": "XLMRobertaTokenizer",
|
| 55 |
+
"unk_token": "<unk>"
|
| 56 |
+
}
|