Sentence Similarity
sentence-transformers
Safetensors
gemma3_text
feature-extraction
dense
Generated from Trainer
dataset_size:5749
loss:CosineSimilarityLoss
Eval Results (legacy)
text-embeddings-inference
Instructions to use summerstars/MARK-Embedding with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use summerstars/MARK-Embedding with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("summerstars/MARK-Embedding") sentences = [ "Nterprise Linux Services is expected to be available before then end of this year.", "Beta versions of Nterprise Linux Services are expected to be available on certain HP ProLiant servers in July.", "Spain turning back the clock on siestas", "I don't like many flavored drinks." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- 1_Pooling/config.json +10 -0
- README.md +364 -42
- calibration.json +4 -0
- config_sentence_transformers.json +14 -0
- eval/similarity_evaluation_results.csv +4 -0
- model.safetensors +1 -1
- modules.json +14 -0
- sentence_bert_config.json +4 -0
- tokenizer.json +2 -2
- tokenizer_config.json +6 -0
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 256,
|
| 3 |
+
"pooling_mode_cls_token": false,
|
| 4 |
+
"pooling_mode_mean_tokens": true,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
+
}
|
README.md
CHANGED
|
@@ -1,66 +1,388 @@
|
|
| 1 |
---
|
| 2 |
-
language: ja
|
| 3 |
-
license: apache-2.0
|
| 4 |
tags:
|
| 5 |
- sentence-transformers
|
| 6 |
-
-
|
| 7 |
-
-
|
| 8 |
-
-
|
| 9 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
library_name: sentence-transformers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
-
#
|
| 14 |
|
| 15 |
-
|
| 16 |
-
SentenceTransformers 互換で、意味ベースの文章類似度計算や検索、クラスタリングなどに利用できます。
|
| 17 |
|
| 18 |
-
##
|
| 19 |
|
| 20 |
-
|
| 21 |
-
- **
|
| 22 |
-
- **
|
| 23 |
-
- **
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
|
|
|
| 28 |
|
| 29 |
-
##
|
| 30 |
|
| 31 |
-
|
| 32 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
|
| 34 |
-
#
|
| 35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
-
#
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
embeddings = model.encode(sentences)
|
|
|
|
|
|
|
| 40 |
|
| 41 |
-
#
|
| 42 |
-
|
| 43 |
-
print(
|
|
|
|
|
|
|
|
|
|
| 44 |
```
|
| 45 |
|
| 46 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
-
|
|
|
|
| 53 |
|
| 54 |
-
|
| 55 |
-
-
|
| 56 |
-
- 商用利用の場合はライセンス条件を確認してください。
|
| 57 |
|
| 58 |
-
|
|
|
|
| 59 |
|
| 60 |
-
|
| 61 |
-
-
|
| 62 |
|
| 63 |
-
|
|
|
|
| 64 |
|
| 65 |
-
|
| 66 |
-
|
|
|
|
| 1 |
---
|
|
|
|
|
|
|
| 2 |
tags:
|
| 3 |
- sentence-transformers
|
| 4 |
+
- sentence-similarity
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- dense
|
| 7 |
+
- generated_from_trainer
|
| 8 |
+
- dataset_size:5749
|
| 9 |
+
- loss:CosineSimilarityLoss
|
| 10 |
+
widget:
|
| 11 |
+
- source_sentence: Nterprise Linux Services is expected to be available before then
|
| 12 |
+
end of this year.
|
| 13 |
+
sentences:
|
| 14 |
+
- Beta versions of Nterprise Linux Services are expected to be available on certain
|
| 15 |
+
HP ProLiant servers in July.
|
| 16 |
+
- Spain turning back the clock on siestas
|
| 17 |
+
- I don't like many flavored drinks.
|
| 18 |
+
- source_sentence: Iran hopes nuclear talks will yield 'roadmap'
|
| 19 |
+
sentences:
|
| 20 |
+
- Iran Nuclear Talks in Geneva Spur High Hopes
|
| 21 |
+
- A black pet dog runs around in the garden of a house.
|
| 22 |
+
- The witness was a 27-year-old Kosovan parking attendant, who was paid by the News
|
| 23 |
+
of the World, the court heard.
|
| 24 |
+
- source_sentence: Hamas Urges Hizbullah to Pull Fighters Out of Syria
|
| 25 |
+
sentences:
|
| 26 |
+
- '"This was a persistent problem which has not been solved, mechanically and physically,"
|
| 27 |
+
said board member Steven Wallace.'
|
| 28 |
+
- A small dog jumps over a yellow beam.
|
| 29 |
+
- Hamas calls on Hezbollah to pull forces out of Syria
|
| 30 |
+
- source_sentence: Licensing revenue slid 21 percent, however, to $107.6 million.
|
| 31 |
+
sentences:
|
| 32 |
+
- Britain loses bid to deport radical cleric Abu Qatada
|
| 33 |
+
- A man sits on a bed very close to a small television.
|
| 34 |
+
- License sales, a key measure of demand, fell 21 percent to $107.6 million.
|
| 35 |
+
- source_sentence: Comcast Class A shares were up 8 cents at $30.50 in morning trading
|
| 36 |
+
on the Nasdaq Stock Market.
|
| 37 |
+
sentences:
|
| 38 |
+
- The stock rose 48 cents to $30 yesterday in Nasdaq Stock Market trading.
|
| 39 |
+
- 'Malaysia: Chinese satellite found object in ocean'
|
| 40 |
+
- A boy in a robe sits in a chair.
|
| 41 |
+
pipeline_tag: sentence-similarity
|
| 42 |
library_name: sentence-transformers
|
| 43 |
+
metrics:
|
| 44 |
+
- pearson_cosine
|
| 45 |
+
- spearman_cosine
|
| 46 |
+
model-index:
|
| 47 |
+
- name: SentenceTransformer
|
| 48 |
+
results:
|
| 49 |
+
- task:
|
| 50 |
+
type: semantic-similarity
|
| 51 |
+
name: Semantic Similarity
|
| 52 |
+
dataset:
|
| 53 |
+
name: Unknown
|
| 54 |
+
type: unknown
|
| 55 |
+
metrics:
|
| 56 |
+
- type: pearson_cosine
|
| 57 |
+
value: 0.4639747212598005
|
| 58 |
+
name: Pearson Cosine
|
| 59 |
+
- type: spearman_cosine
|
| 60 |
+
value: 0.4595105448711385
|
| 61 |
+
name: Spearman Cosine
|
| 62 |
---
|
| 63 |
|
| 64 |
+
# SentenceTransformer
|
| 65 |
|
| 66 |
+
This is a [sentence-transformers](https://www.SBERT.net) model trained. It maps sentences & paragraphs to a 256-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
|
|
|
| 67 |
|
| 68 |
+
## Model Details
|
| 69 |
|
| 70 |
+
### Model Description
|
| 71 |
+
- **Model Type:** Sentence Transformer
|
| 72 |
+
<!-- - **Base model:** [Unknown](https://huggingface.co/unknown) -->
|
| 73 |
+
- **Maximum Sequence Length:** 2048 tokens
|
| 74 |
+
- **Output Dimensionality:** 256 dimensions
|
| 75 |
+
- **Similarity Function:** Cosine Similarity
|
| 76 |
+
<!-- - **Training Dataset:** Unknown -->
|
| 77 |
+
<!-- - **Language:** Unknown -->
|
| 78 |
+
<!-- - **License:** Unknown -->
|
| 79 |
|
| 80 |
+
### Model Sources
|
| 81 |
|
| 82 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 83 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
| 84 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| 85 |
+
|
| 86 |
+
### Full Model Architecture
|
| 87 |
+
|
| 88 |
+
```
|
| 89 |
+
SentenceTransformer(
|
| 90 |
+
(0): Transformer({'max_seq_length': 2048, 'do_lower_case': False, 'architecture': 'Gemma3TextModel'})
|
| 91 |
+
(1): Pooling({'word_embedding_dimension': 256, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
|
| 92 |
+
)
|
| 93 |
+
```
|
| 94 |
+
|
| 95 |
+
## Usage
|
| 96 |
|
| 97 |
+
### Direct Usage (Sentence Transformers)
|
| 98 |
+
|
| 99 |
+
First install the Sentence Transformers library:
|
| 100 |
+
|
| 101 |
+
```bash
|
| 102 |
+
pip install -U sentence-transformers
|
| 103 |
+
```
|
| 104 |
+
|
| 105 |
+
Then you can load this model and run inference.
|
| 106 |
+
```python
|
| 107 |
+
from sentence_transformers import SentenceTransformer
|
| 108 |
|
| 109 |
+
# Download from the 🤗 Hub
|
| 110 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
| 111 |
+
# Run inference
|
| 112 |
+
sentences = [
|
| 113 |
+
'Comcast Class A shares were up 8 cents at $30.50 in morning trading on the Nasdaq Stock Market.',
|
| 114 |
+
'The stock rose 48 cents to $30 yesterday in Nasdaq Stock Market trading.',
|
| 115 |
+
'Malaysia: Chinese satellite found object in ocean',
|
| 116 |
+
]
|
| 117 |
embeddings = model.encode(sentences)
|
| 118 |
+
print(embeddings.shape)
|
| 119 |
+
# [3, 256]
|
| 120 |
|
| 121 |
+
# Get the similarity scores for the embeddings
|
| 122 |
+
similarities = model.similarity(embeddings, embeddings)
|
| 123 |
+
print(similarities)
|
| 124 |
+
# tensor([[1.0000, 0.5752, 0.2980],
|
| 125 |
+
# [0.5752, 1.0000, 0.2161],
|
| 126 |
+
# [0.2980, 0.2161, 1.0000]])
|
| 127 |
```
|
| 128 |
|
| 129 |
+
<!--
|
| 130 |
+
### Direct Usage (Transformers)
|
| 131 |
+
|
| 132 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 133 |
+
|
| 134 |
+
</details>
|
| 135 |
+
-->
|
| 136 |
+
|
| 137 |
+
<!--
|
| 138 |
+
### Downstream Usage (Sentence Transformers)
|
| 139 |
+
|
| 140 |
+
You can finetune this model on your own dataset.
|
| 141 |
+
|
| 142 |
+
<details><summary>Click to expand</summary>
|
| 143 |
+
|
| 144 |
+
</details>
|
| 145 |
+
-->
|
| 146 |
+
|
| 147 |
+
<!--
|
| 148 |
+
### Out-of-Scope Use
|
| 149 |
+
|
| 150 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 151 |
+
-->
|
| 152 |
+
|
| 153 |
+
## Evaluation
|
| 154 |
+
|
| 155 |
+
### Metrics
|
| 156 |
+
|
| 157 |
+
#### Semantic Similarity
|
| 158 |
|
| 159 |
+
* Evaluated with [<code>EmbeddingSimilarityEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.EmbeddingSimilarityEvaluator)
|
| 160 |
+
|
| 161 |
+
| Metric | Value |
|
| 162 |
+
|:--------------------|:-----------|
|
| 163 |
+
| pearson_cosine | 0.464 |
|
| 164 |
+
| **spearman_cosine** | **0.4595** |
|
| 165 |
+
|
| 166 |
+
<!--
|
| 167 |
+
## Bias, Risks and Limitations
|
| 168 |
+
|
| 169 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 170 |
+
-->
|
| 171 |
+
|
| 172 |
+
<!--
|
| 173 |
+
### Recommendations
|
| 174 |
+
|
| 175 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 176 |
+
-->
|
| 177 |
+
|
| 178 |
+
## Training Details
|
| 179 |
+
|
| 180 |
+
### Training Dataset
|
| 181 |
+
|
| 182 |
+
#### Unnamed Dataset
|
| 183 |
+
|
| 184 |
+
* Size: 5,749 training samples
|
| 185 |
+
* Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
|
| 186 |
+
* Approximate statistics based on the first 1000 samples:
|
| 187 |
+
| | sentence_0 | sentence_1 | label |
|
| 188 |
+
|:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:---------------------------------------------------------------|
|
| 189 |
+
| type | string | string | float |
|
| 190 |
+
| details | <ul><li>min: 6 tokens</li><li>mean: 14.76 tokens</li><li>max: 55 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 14.73 tokens</li><li>max: 57 tokens</li></ul> | <ul><li>min: 0.0</li><li>mean: 0.55</li><li>max: 1.0</li></ul> |
|
| 191 |
+
* Samples:
|
| 192 |
+
| sentence_0 | sentence_1 | label |
|
| 193 |
+
|:----------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|:--------------------------------|
|
| 194 |
+
| <code>Forecasters said warnings might go up for Cuba later Thursday.</code> | <code>Watches or warnings could be issued for eastern Cuba later on Thursday.</code> | <code>0.8</code> |
|
| 195 |
+
| <code>Death toll in Lebanon bombings rises to 47</code> | <code>1 suspect arrested after Lebanon car bombings kill 45</code> | <code>0.5599999904632569</code> |
|
| 196 |
+
| <code>Three dogs running on a racetrack.</code> | <code>Three dogs round a bend at a racetrack.</code> | <code>0.9600000381469727</code> |
|
| 197 |
+
* Loss: [<code>CosineSimilarityLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cosinesimilarityloss) with these parameters:
|
| 198 |
+
```json
|
| 199 |
+
{
|
| 200 |
+
"loss_fct": "torch.nn.modules.loss.MSELoss"
|
| 201 |
+
}
|
| 202 |
+
```
|
| 203 |
+
|
| 204 |
+
### Training Hyperparameters
|
| 205 |
+
#### Non-Default Hyperparameters
|
| 206 |
+
|
| 207 |
+
- `eval_strategy`: steps
|
| 208 |
+
- `per_device_train_batch_size`: 16
|
| 209 |
+
- `per_device_eval_batch_size`: 16
|
| 210 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 211 |
+
|
| 212 |
+
#### All Hyperparameters
|
| 213 |
+
<details><summary>Click to expand</summary>
|
| 214 |
+
|
| 215 |
+
- `overwrite_output_dir`: False
|
| 216 |
+
- `do_predict`: False
|
| 217 |
+
- `eval_strategy`: steps
|
| 218 |
+
- `prediction_loss_only`: True
|
| 219 |
+
- `per_device_train_batch_size`: 16
|
| 220 |
+
- `per_device_eval_batch_size`: 16
|
| 221 |
+
- `per_gpu_train_batch_size`: None
|
| 222 |
+
- `per_gpu_eval_batch_size`: None
|
| 223 |
+
- `gradient_accumulation_steps`: 1
|
| 224 |
+
- `eval_accumulation_steps`: None
|
| 225 |
+
- `torch_empty_cache_steps`: None
|
| 226 |
+
- `learning_rate`: 5e-05
|
| 227 |
+
- `weight_decay`: 0.0
|
| 228 |
+
- `adam_beta1`: 0.9
|
| 229 |
+
- `adam_beta2`: 0.999
|
| 230 |
+
- `adam_epsilon`: 1e-08
|
| 231 |
+
- `max_grad_norm`: 1
|
| 232 |
+
- `num_train_epochs`: 3
|
| 233 |
+
- `max_steps`: -1
|
| 234 |
+
- `lr_scheduler_type`: linear
|
| 235 |
+
- `lr_scheduler_kwargs`: {}
|
| 236 |
+
- `warmup_ratio`: 0.0
|
| 237 |
+
- `warmup_steps`: 0
|
| 238 |
+
- `log_level`: passive
|
| 239 |
+
- `log_level_replica`: warning
|
| 240 |
+
- `log_on_each_node`: True
|
| 241 |
+
- `logging_nan_inf_filter`: True
|
| 242 |
+
- `save_safetensors`: True
|
| 243 |
+
- `save_on_each_node`: False
|
| 244 |
+
- `save_only_model`: False
|
| 245 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 246 |
+
- `no_cuda`: False
|
| 247 |
+
- `use_cpu`: False
|
| 248 |
+
- `use_mps_device`: False
|
| 249 |
+
- `seed`: 42
|
| 250 |
+
- `data_seed`: None
|
| 251 |
+
- `jit_mode_eval`: False
|
| 252 |
+
- `use_ipex`: False
|
| 253 |
+
- `bf16`: False
|
| 254 |
+
- `fp16`: False
|
| 255 |
+
- `fp16_opt_level`: O1
|
| 256 |
+
- `half_precision_backend`: auto
|
| 257 |
+
- `bf16_full_eval`: False
|
| 258 |
+
- `fp16_full_eval`: False
|
| 259 |
+
- `tf32`: None
|
| 260 |
+
- `local_rank`: 0
|
| 261 |
+
- `ddp_backend`: None
|
| 262 |
+
- `tpu_num_cores`: None
|
| 263 |
+
- `tpu_metrics_debug`: False
|
| 264 |
+
- `debug`: []
|
| 265 |
+
- `dataloader_drop_last`: False
|
| 266 |
+
- `dataloader_num_workers`: 0
|
| 267 |
+
- `dataloader_prefetch_factor`: None
|
| 268 |
+
- `past_index`: -1
|
| 269 |
+
- `disable_tqdm`: False
|
| 270 |
+
- `remove_unused_columns`: True
|
| 271 |
+
- `label_names`: None
|
| 272 |
+
- `load_best_model_at_end`: False
|
| 273 |
+
- `ignore_data_skip`: False
|
| 274 |
+
- `fsdp`: []
|
| 275 |
+
- `fsdp_min_num_params`: 0
|
| 276 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 277 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
| 278 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 279 |
+
- `parallelism_config`: None
|
| 280 |
+
- `deepspeed`: None
|
| 281 |
+
- `label_smoothing_factor`: 0.0
|
| 282 |
+
- `optim`: adamw_torch_fused
|
| 283 |
+
- `optim_args`: None
|
| 284 |
+
- `adafactor`: False
|
| 285 |
+
- `group_by_length`: False
|
| 286 |
+
- `length_column_name`: length
|
| 287 |
+
- `ddp_find_unused_parameters`: None
|
| 288 |
+
- `ddp_bucket_cap_mb`: None
|
| 289 |
+
- `ddp_broadcast_buffers`: False
|
| 290 |
+
- `dataloader_pin_memory`: True
|
| 291 |
+
- `dataloader_persistent_workers`: False
|
| 292 |
+
- `skip_memory_metrics`: True
|
| 293 |
+
- `use_legacy_prediction_loop`: False
|
| 294 |
+
- `push_to_hub`: False
|
| 295 |
+
- `resume_from_checkpoint`: None
|
| 296 |
+
- `hub_model_id`: None
|
| 297 |
+
- `hub_strategy`: every_save
|
| 298 |
+
- `hub_private_repo`: None
|
| 299 |
+
- `hub_always_push`: False
|
| 300 |
+
- `hub_revision`: None
|
| 301 |
+
- `gradient_checkpointing`: False
|
| 302 |
+
- `gradient_checkpointing_kwargs`: None
|
| 303 |
+
- `include_inputs_for_metrics`: False
|
| 304 |
+
- `include_for_metrics`: []
|
| 305 |
+
- `eval_do_concat_batches`: True
|
| 306 |
+
- `fp16_backend`: auto
|
| 307 |
+
- `push_to_hub_model_id`: None
|
| 308 |
+
- `push_to_hub_organization`: None
|
| 309 |
+
- `mp_parameters`:
|
| 310 |
+
- `auto_find_batch_size`: False
|
| 311 |
+
- `full_determinism`: False
|
| 312 |
+
- `torchdynamo`: None
|
| 313 |
+
- `ray_scope`: last
|
| 314 |
+
- `ddp_timeout`: 1800
|
| 315 |
+
- `torch_compile`: False
|
| 316 |
+
- `torch_compile_backend`: None
|
| 317 |
+
- `torch_compile_mode`: None
|
| 318 |
+
- `include_tokens_per_second`: False
|
| 319 |
+
- `include_num_input_tokens_seen`: False
|
| 320 |
+
- `neftune_noise_alpha`: None
|
| 321 |
+
- `optim_target_modules`: None
|
| 322 |
+
- `batch_eval_metrics`: False
|
| 323 |
+
- `eval_on_start`: False
|
| 324 |
+
- `use_liger_kernel`: False
|
| 325 |
+
- `liger_kernel_config`: None
|
| 326 |
+
- `eval_use_gather_object`: False
|
| 327 |
+
- `average_tokens_across_devices`: False
|
| 328 |
+
- `prompts`: None
|
| 329 |
+
- `batch_sampler`: batch_sampler
|
| 330 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 331 |
+
- `router_mapping`: {}
|
| 332 |
+
- `learning_rate_mapping`: {}
|
| 333 |
+
|
| 334 |
+
</details>
|
| 335 |
+
|
| 336 |
+
### Training Logs
|
| 337 |
+
| Epoch | Step | Training Loss | spearman_cosine |
|
| 338 |
+
|:------:|:----:|:-------------:|:---------------:|
|
| 339 |
+
| 1.0 | 360 | - | 0.2967 |
|
| 340 |
+
| 1.3889 | 500 | 0.11 | 0.3338 |
|
| 341 |
+
| 2.0 | 720 | - | 0.3665 |
|
| 342 |
+
| 2.7778 | 1000 | 0.0857 | 0.4101 |
|
| 343 |
+
| 3.0 | 1080 | - | 0.4595 |
|
| 344 |
+
|
| 345 |
+
|
| 346 |
+
### Framework Versions
|
| 347 |
+
- Python: 3.12.11
|
| 348 |
+
- Sentence Transformers: 5.1.0
|
| 349 |
+
- Transformers: 4.56.1
|
| 350 |
+
- PyTorch: 2.8.0+cu126
|
| 351 |
+
- Accelerate: 1.10.1
|
| 352 |
+
- Datasets: 4.0.0
|
| 353 |
+
- Tokenizers: 0.22.0
|
| 354 |
+
|
| 355 |
+
## Citation
|
| 356 |
+
|
| 357 |
+
### BibTeX
|
| 358 |
+
|
| 359 |
+
#### Sentence Transformers
|
| 360 |
+
```bibtex
|
| 361 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 362 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 363 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 364 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 365 |
+
month = "11",
|
| 366 |
+
year = "2019",
|
| 367 |
+
publisher = "Association for Computational Linguistics",
|
| 368 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 369 |
+
}
|
| 370 |
+
```
|
| 371 |
|
| 372 |
+
<!--
|
| 373 |
+
## Glossary
|
| 374 |
|
| 375 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 376 |
+
-->
|
|
|
|
| 377 |
|
| 378 |
+
<!--
|
| 379 |
+
## Model Card Authors
|
| 380 |
|
| 381 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 382 |
+
-->
|
| 383 |
|
| 384 |
+
<!--
|
| 385 |
+
## Model Card Contact
|
| 386 |
|
| 387 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 388 |
+
-->
|
calibration.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"slope": 3.959556818008423,
|
| 3 |
+
"intercept": 0.20897746086120605
|
| 4 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "SentenceTransformer",
|
| 3 |
+
"__version__": {
|
| 4 |
+
"sentence_transformers": "5.1.0",
|
| 5 |
+
"transformers": "4.56.1",
|
| 6 |
+
"pytorch": "2.8.0+cu126"
|
| 7 |
+
},
|
| 8 |
+
"prompts": {
|
| 9 |
+
"query": "",
|
| 10 |
+
"document": ""
|
| 11 |
+
},
|
| 12 |
+
"default_prompt_name": null,
|
| 13 |
+
"similarity_fn_name": "cosine"
|
| 14 |
+
}
|
eval/similarity_evaluation_results.csv
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
epoch,steps,cosine_pearson,cosine_spearman
|
| 2 |
+
1.0,360,0.2903157409900097,0.2967070624119561
|
| 3 |
+
2.0,720,0.36665449013888085,0.3665099682349319
|
| 4 |
+
3.0,1080,0.4639747212598005,0.4595105448711385
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 336250336
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9788644e1e4b74fe5d93732ab6ac45bba7c0eb4a9b19066b971ed5bb2eb1772
|
| 3 |
size 336250336
|
modules.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.models.Transformer"
|
| 7 |
+
},
|
| 8 |
+
{
|
| 9 |
+
"idx": 1,
|
| 10 |
+
"name": "1",
|
| 11 |
+
"path": "1_Pooling",
|
| 12 |
+
"type": "sentence_transformers.models.Pooling"
|
| 13 |
+
}
|
| 14 |
+
]
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"max_seq_length": 2048,
|
| 3 |
+
"do_lower_case": false
|
| 4 |
+
}
|
tokenizer.json
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:216e2a79606fe879c9f17c529c71cd241338407fd5646b595ffd3c4b9ea1d503
|
| 3 |
+
size 33385262
|
tokenizer_config.json
CHANGED
|
@@ -51334,12 +51334,18 @@
|
|
| 51334 |
"image_token": "<image_soft_token>"
|
| 51335 |
},
|
| 51336 |
"image_token": "<image_soft_token>",
|
|
|
|
| 51337 |
"model_max_length": 2048,
|
|
|
|
| 51338 |
"pad_token": "<pad>",
|
|
|
|
| 51339 |
"padding_side": "right",
|
| 51340 |
"sp_model_kwargs": null,
|
| 51341 |
"spaces_between_special_tokens": false,
|
|
|
|
| 51342 |
"tokenizer_class": "GemmaTokenizer",
|
|
|
|
|
|
|
| 51343 |
"unk_token": "<unk>",
|
| 51344 |
"use_default_system_prompt": false
|
| 51345 |
}
|
|
|
|
| 51334 |
"image_token": "<image_soft_token>"
|
| 51335 |
},
|
| 51336 |
"image_token": "<image_soft_token>",
|
| 51337 |
+
"max_length": 2048,
|
| 51338 |
"model_max_length": 2048,
|
| 51339 |
+
"pad_to_multiple_of": null,
|
| 51340 |
"pad_token": "<pad>",
|
| 51341 |
+
"pad_token_type_id": 0,
|
| 51342 |
"padding_side": "right",
|
| 51343 |
"sp_model_kwargs": null,
|
| 51344 |
"spaces_between_special_tokens": false,
|
| 51345 |
+
"stride": 0,
|
| 51346 |
"tokenizer_class": "GemmaTokenizer",
|
| 51347 |
+
"truncation_side": "right",
|
| 51348 |
+
"truncation_strategy": "longest_first",
|
| 51349 |
"unk_token": "<unk>",
|
| 51350 |
"use_default_system_prompt": false
|
| 51351 |
}
|