Sentence Similarity
sentence-transformers
ONNX
Safetensors
Italian
modernbert
feature-extraction
dense
Generated from Trainer
dataset_size:45181
loss:MatryoshkaLoss
loss:CachedMultipleNegativesRankingLoss
loss:MultipleNegativesRankingLoss
Eval Results (legacy)
text-embeddings-inference
Instructions to use nickprock/granite-311m-italiano-matryoshka with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use nickprock/granite-311m-italiano-matryoshka with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("nickprock/granite-311m-italiano-matryoshka") sentences = [ "In quali anni ha partecipato Carriera ai Campionati europei?", "Mokgomane è un villaggio del Botswana situato nel distretto Meridionale, sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011, conta 708 abitanti.", "['Carriera\\nCon la ha disputato i Campionati europei del 2001 e i Campionati mondiali del 2002.\\n\\nCon la ha disputato due edizioni dei Campionati europei (2003, 2005).', 'Carriera\\nCon la ha disputato i Campionati mondiali del 2002 e tre edizioni dei Campionati europei (1997, 1999, 2001).\\n\\nCon la ha disputato i Campionati europei del 2003.']", "Carriera Con la ha disputato i Giochi olimpici di Montréal 1976, i Campionati mondiali del 1975 e tre edizioni dei Campionati europei (1974, 1976, 1978)." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,91 +1,126 @@
|
|
| 1 |
-
---
|
| 2 |
-
tags:
|
| 3 |
-
- sentence-transformers
|
| 4 |
-
- sentence-similarity
|
| 5 |
-
- feature-extraction
|
| 6 |
-
- dense
|
| 7 |
-
- generated_from_trainer
|
| 8 |
-
- dataset_size:45181
|
| 9 |
-
- loss:MatryoshkaLoss
|
| 10 |
-
- loss:CachedMultipleNegativesRankingLoss
|
| 11 |
-
- loss:MultipleNegativesRankingLoss
|
| 12 |
-
base_model: ibm-granite/granite-embedding-311m-multilingual-r2
|
| 13 |
-
widget:
|
| 14 |
-
- source_sentence: In quali anni ha partecipato Carriera ai Campionati europei?
|
| 15 |
-
sentences:
|
| 16 |
-
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
di
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
# SentenceTransformer based on ibm-granite/granite-embedding-311m-multilingual-r2
|
| 91 |
|
|
@@ -176,34 +211,51 @@ You can finetune this model on your own dataset.
|
|
| 176 |
|
| 177 |
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 178 |
-->
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 179 |
|
| 180 |
-
## Evaluation
|
| 181 |
-
|
| 182 |
-
### Metrics
|
| 183 |
-
|
| 184 |
-
#### Triplet
|
| 185 |
-
|
| 186 |
-
* Dataset: `triplet-eval-it`
|
| 187 |
-
* Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.sentence_transformer.evaluation.TripletEvaluator)
|
| 188 |
-
|
| 189 |
-
| Metric | Value |
|
| 190 |
-
|:--------------------|:-----------|
|
| 191 |
-
| **cosine_accuracy** | **0.9996** |
|
| 192 |
-
|
| 193 |
-
<!--
|
| 194 |
-
## Bias, Risks and Limitations
|
| 195 |
-
|
| 196 |
-
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 197 |
-
-->
|
| 198 |
-
|
| 199 |
-
<!--
|
| 200 |
-
### Recommendations
|
| 201 |
-
|
| 202 |
-
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 203 |
-
-->
|
| 204 |
|
| 205 |
## Training Details
|
| 206 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 207 |
### Training Dataset
|
| 208 |
|
| 209 |
#### it-wiki-retrieval-synthetic-hn
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- sentence-transformers
|
| 4 |
+
- sentence-similarity
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- dense
|
| 7 |
+
- generated_from_trainer
|
| 8 |
+
- dataset_size:45181
|
| 9 |
+
- loss:MatryoshkaLoss
|
| 10 |
+
- loss:CachedMultipleNegativesRankingLoss
|
| 11 |
+
- loss:MultipleNegativesRankingLoss
|
| 12 |
+
base_model: ibm-granite/granite-embedding-311m-multilingual-r2
|
| 13 |
+
widget:
|
| 14 |
+
- source_sentence: In quali anni ha partecipato Carriera ai Campionati europei?
|
| 15 |
+
sentences:
|
| 16 |
+
- >-
|
| 17 |
+
Mokgomane è un villaggio del Botswana situato nel distretto Meridionale,
|
| 18 |
+
sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
|
| 19 |
+
conta 708 abitanti.
|
| 20 |
+
- >-
|
| 21 |
+
['Carriera\nCon la ha disputato i Campionati europei del 2001 e i Campionati
|
| 22 |
+
mondiali del 2002.\n\nCon la ha disputato due edizioni dei Campionati
|
| 23 |
+
europei (2003, 2005).', 'Carriera\nCon la ha disputato i Campionati mondiali
|
| 24 |
+
del 2002 e tre edizioni dei Campionati europei (1997, 1999, 2001).\n\nCon la
|
| 25 |
+
ha disputato i Campionati europei del 2003.']
|
| 26 |
+
- >-
|
| 27 |
+
Carriera Con la ha disputato i Giochi olimpici di Montréal 1976, i
|
| 28 |
+
Campionati mondiali del 1975 e tre edizioni dei Campionati europei (1974,
|
| 29 |
+
1976, 1978).
|
| 30 |
+
- source_sentence: >-
|
| 31 |
+
Quali sono le caratteristiche geografiche e demografiche della prefettura di
|
| 32 |
+
Avé in Togo?
|
| 33 |
+
sentences:
|
| 34 |
+
- >-
|
| 35 |
+
La prefettura di Avé è una prefettura del Togo situato nella regione
|
| 36 |
+
Marittima con 97.830 abitanti al censimento 2010. Il capoluogo è la città di
|
| 37 |
+
Kévé.
|
| 38 |
+
- >-
|
| 39 |
+
['La prefettura di Haho è una prefettura del Togo situato nella regione
|
| 40 |
+
degli Altopiani con 247.817 abitanti al censimento 2010. Il capoluogo è la
|
| 41 |
+
città di Notsé.', 'La prefettura di Lacs è una prefettura del Togo situato
|
| 42 |
+
nella regione Marittima con 172.148 abitanti al censimento 2010. Il
|
| 43 |
+
capoluogo è la città di Aneho.']
|
| 44 |
+
- >-
|
| 45 |
+
Mmathethe è un villaggio del Botswana situato nel distretto Meridionale,
|
| 46 |
+
sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
|
| 47 |
+
conta 5.078 abitanti.
|
| 48 |
+
- source_sentence: Quali sono i nomi alternativi per il comune Těrlicko?
|
| 49 |
+
sentences:
|
| 50 |
+
- >-
|
| 51 |
+
Malotwana Siding è un villaggio del Botswana situato nel distretto di
|
| 52 |
+
Kgatleng, sottodistretto di Kgatleng. Il villaggio, secondo il censimento
|
| 53 |
+
del 2011, conta 608 abitanti.
|
| 54 |
+
- >-
|
| 55 |
+
['Il toponimo Rohoznice può riferirsi a diverse località della Repubblica
|
| 56 |
+
Ceca:\n\nRohoznice, comune del distretto di Jičín\nRohoznice, comune del
|
| 57 |
+
distretto di Pardubice', "Biskupice (in tedesco Biskupitz) è un comune della
|
| 58 |
+
Repubblica Ceca facente parte del distretto di Zlín, nella regione di
|
| 59 |
+
Zlín.\n\nQua è nato l'ex calciatore ed allenatore Milan Máčala."]
|
| 60 |
+
- >-
|
| 61 |
+
Těrlicko (in polacco Cierlicko, in tedesco Tierlitzko) è un comune della
|
| 62 |
+
Repubblica Ceca facente parte del distretto di Karviná, nella regione della
|
| 63 |
+
Moravia-Slesia.
|
| 64 |
+
- source_sentence: Quante persone habita nel villaggio Tswaaneng al censimento del 2011?
|
| 65 |
+
sentences:
|
| 66 |
+
- >-
|
| 67 |
+
Mogoditshane è un villaggio del Botswana situato nel distretto di Kweneng,
|
| 68 |
+
sottodistretto di Kweneng East. Il villaggio, secondo il censimento del
|
| 69 |
+
2011, conta 58.079 abitanti.
|
| 70 |
+
- >-
|
| 71 |
+
['Tswaane è un villaggio del Botswana situato nel distretto di Kweneng,
|
| 72 |
+
sottodistretto di Kweneng West. Il villaggio, secondo il censimento del
|
| 73 |
+
2011, conta 603 abitanti.', 'Tshwaane è un villaggio del Botswana situato
|
| 74 |
+
nel distretto Meridionale, sottodistretto di Ngwaketse. Il villaggio,
|
| 75 |
+
secondo il censimento del 2011, conta 299 abitanti.']
|
| 76 |
+
- >-
|
| 77 |
+
Tswaaneng è un villaggio del Botswana situato nel distretto Meridionale,
|
| 78 |
+
sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
|
| 79 |
+
conta 679 abitanti.
|
| 80 |
+
- source_sentence: In quali gare fu eliminato l'atleta nelle Olimpiadi di Parigi?
|
| 81 |
+
sentences:
|
| 82 |
+
- >-
|
| 83 |
+
Partecipò ai Giochi della II Olimpiade di Parigi nella gara di spada
|
| 84 |
+
individuale e sciabola individuale. In entrambe le gare fu eliminato alle
|
| 85 |
+
semifinali.
|
| 86 |
+
- >-
|
| 87 |
+
Partecipò ai Giochi della II Olimpiade di Parigi nelle gare di fioretto per
|
| 88 |
+
maestri, dove giunse al nono posto, e di spada per maestri, dove fu
|
| 89 |
+
eliminato in semifinale.
|
| 90 |
+
- >-
|
| 91 |
+
['Partecipò ai Giochi della II Olimpiade di Parigi alla gara di fioretto
|
| 92 |
+
individuale, in cui fu eliminato ai quarti, e alla gara di spada
|
| 93 |
+
individuale, in cui fu eliminato al primo turno.', 'Partecipò ai Giochi
|
| 94 |
+
della II Olimpiade di Parigi nelle gare di fioretto per maestri, in cui fu
|
| 95 |
+
eliminato ai ripescaggi, e di spada per maestri, in cui fu eliminato al
|
| 96 |
+
primo turno.']
|
| 97 |
+
pipeline_tag: sentence-similarity
|
| 98 |
+
library_name: sentence-transformers
|
| 99 |
+
metrics:
|
| 100 |
+
- cosine_accuracy
|
| 101 |
+
model-index:
|
| 102 |
+
- name: >-
|
| 103 |
+
SentenceTransformer based on
|
| 104 |
+
ibm-granite/granite-embedding-311m-multilingual-r2
|
| 105 |
+
results:
|
| 106 |
+
- task:
|
| 107 |
+
type: triplet
|
| 108 |
+
name: Triplet
|
| 109 |
+
dataset:
|
| 110 |
+
name: triplet eval it
|
| 111 |
+
type: triplet-eval-it
|
| 112 |
+
metrics:
|
| 113 |
+
- type: cosine_accuracy
|
| 114 |
+
value: 0.9995794892311096
|
| 115 |
+
name: Cosine Accuracy
|
| 116 |
+
license: apache-2.0
|
| 117 |
+
datasets:
|
| 118 |
+
- nickprock/it-wiki-retrieval-synthetic-hn
|
| 119 |
+
- mteb/stsb_multi_mt
|
| 120 |
+
- crux82/squad_it
|
| 121 |
+
language:
|
| 122 |
+
- it
|
| 123 |
+
---
|
| 124 |
|
| 125 |
# SentenceTransformer based on ibm-granite/granite-embedding-311m-multilingual-r2
|
| 126 |
|
|
|
|
| 211 |
|
| 212 |
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 213 |
-->
|
| 214 |
+
## Evaluation & Benchmarks
|
| 215 |
+
|
| 216 |
+
To ensure full transparency, the model was evaluated across two distinct benchmarks: **Information Retrieval / RAG** (SQuAD-it) and **Semantic Textual Similarity** (STSb-it), compared against widely used Italian embedding models.
|
| 217 |
+
|
| 218 |
+
### 1. Information Retrieval & RAG Benchmark (SQuAD-it)
|
| 219 |
+
* **Dataset:** `crux82/squad_it` (Test Split - 1,000 queries matched against a unique passage corpus)
|
| 220 |
+
* **Objective:** Evaluate how effectively the model retrieves the exact paragraph containing the answer among 1,000 candidate passages.
|
| 221 |
+
|
| 222 |
+
| Modello | Dim. Vettore | Hit@1 / Precision (%) | MRR@10 | NDCG@10 | Latency (1k Queries) |
|
| 223 |
+
| :--- | :---: | :---: | :---: | :---: | :---: |
|
| 224 |
+
| `nickprock/sentence-bert-base-italian-uncased` | 768 | 36.40% | 0.4980 | 0.5723 | 2.55s |
|
| 225 |
+
| `nickprock/sentence-bert-base-italian-xxl-uncased` | 768 | 40.60% | 0.5347 | 0.6026 | 2.17s |
|
| 226 |
+
| `nickprock/multi-sentence-BERTino` | 768 | 39.30% | 0.4933 | 0.5465 | 1.11s |
|
| 227 |
+
| `DeepMount00/Ita-Search` | 1024 | 43.50% | 0.5465 | 0.6087 | 17.51s |
|
| 228 |
+
| **`granite-311m-italiano-matryoshka (Full)`** | **768** | **65.40%** | **0.7560** | **0.8009** | **1.71s** |
|
| 229 |
+
| **`granite-311m-italiano-matryoshka (Truncated)`** | **256** | **65.00%** | **0.7500** | **0.7957** | **1.61s** |
|
| 230 |
+
|
| 231 |
+
---
|
| 232 |
+
|
| 233 |
+
### 2. Semantic Textual Similarity Benchmark (STSb-it)
|
| 234 |
+
* **Dataset:** `stsb_multi_mt` (Italian Test Split - 1,379 sentence pairs with human scores from 0 to 5)
|
| 235 |
+
* **Objective:** Measure correlation between model cosine similarity and human judgment on sentence equivalence.
|
| 236 |
+
|
| 237 |
+
| Modello | Dim. Vettore | Spearman Cosine | Pearson Cosine | Throughput (Pairs/Sec) |
|
| 238 |
+
| :--- | :---: | :---: | :---: | :---: |
|
| 239 |
+
| `nickprock/sentence-bert-base-italian-uncased` | 768 | 0.8097 | 0.8098 | 596.9 |
|
| 240 |
+
| `nickprock/sentence-bert-base-italian-xxl-uncased` | 768 | 0.8205 | 0.8224 | 684.1 |
|
| 241 |
+
| `nickprock/multi-sentence-BERTino` | 768 | 0.8137 | 0.8163 | 1206.1 |
|
| 242 |
+
| `DeepMount00/Ita-Search` | 1024 | 0.7608 | 0.7702 | 120.2 |
|
| 243 |
+
| **`granite-311m-italiano-matryoshka (Full)`** | **768** | **0.7612** | **0.7574** | **588.2** |
|
| 244 |
+
| **`granite-311m-italiano-matryoshka (Truncated)`** | **256** | **0.7627** | **0.7599** | **572.0** |
|
| 245 |
+
|
| 246 |
+
---
|
| 247 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 248 |
|
| 249 |
## Training Details
|
| 250 |
|
| 251 |
+
### Training Setup
|
| 252 |
+
- **Dataset:** `nickprock/it-wiki-retrieval-synthetic-hn` (45,181 samples)
|
| 253 |
+
- **Epochs:** 1
|
| 254 |
+
- **Batch Size:** 16 (per device) x 16 (Gradient Accumulation) = 256 effective batch size
|
| 255 |
+
- **Optimizer:** `adamw_8bit`
|
| 256 |
+
- **Learning Rate:** 2e-5 with linear warmup (0.1)
|
| 257 |
+
- **Precision:** `bf16`
|
| 258 |
+
|
| 259 |
### Training Dataset
|
| 260 |
|
| 261 |
#### it-wiki-retrieval-synthetic-hn
|