Sentence Similarity
sentence-transformers
Safetensors
English
static-embedding
chess
retrieval
exploratory
Instructions to use oneryalcin/static-embedding-chess with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use oneryalcin/static-embedding-chess with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("oneryalcin/static-embedding-chess") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Training in progress, step 100
Browse files- README.md +26 -16
- eval/Information-Retrieval_evaluation_chess-ir_results.csv +1 -0
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -67,28 +67,28 @@ model-index:
|
|
| 67 |
value: 0.015
|
| 68 |
name: Cosine Accuracy@1
|
| 69 |
- type: cosine_accuracy@10
|
| 70 |
-
value: 0.
|
| 71 |
name: Cosine Accuracy@10
|
| 72 |
- type: cosine_precision@1
|
| 73 |
value: 0.015
|
| 74 |
name: Cosine Precision@1
|
| 75 |
- type: cosine_precision@10
|
| 76 |
-
value: 0.
|
| 77 |
name: Cosine Precision@10
|
| 78 |
- type: cosine_recall@1
|
| 79 |
value: 0.005
|
| 80 |
name: Cosine Recall@1
|
| 81 |
- type: cosine_recall@10
|
| 82 |
-
value: 0.
|
| 83 |
name: Cosine Recall@10
|
| 84 |
- type: cosine_ndcg@10
|
| 85 |
-
value: 0.
|
| 86 |
name: Cosine Ndcg@10
|
| 87 |
- type: cosine_mrr@10
|
| 88 |
-
value: 0.
|
| 89 |
name: Cosine Mrr@10
|
| 90 |
- type: cosine_map@100
|
| 91 |
-
value: 0.
|
| 92 |
name: Cosine Map@100
|
| 93 |
---
|
| 94 |
|
|
@@ -155,7 +155,7 @@ print(query_embeddings.shape, document_embeddings.shape)
|
|
| 155 |
# Get the similarity scores for the embeddings
|
| 156 |
similarities = model.similarity(query_embeddings, document_embeddings)
|
| 157 |
print(similarities)
|
| 158 |
-
# tensor([[ 0.
|
| 159 |
```
|
| 160 |
<!--
|
| 161 |
### Direct Usage (Transformers)
|
|
@@ -193,14 +193,14 @@ You can finetune this model on your own dataset.
|
|
| 193 |
| Metric | Value |
|
| 194 |
|:--------------------|:-----------|
|
| 195 |
| cosine_accuracy@1 | 0.015 |
|
| 196 |
-
| cosine_accuracy@10 | 0.
|
| 197 |
| cosine_precision@1 | 0.015 |
|
| 198 |
-
| cosine_precision@10 | 0.
|
| 199 |
| cosine_recall@1 | 0.005 |
|
| 200 |
-
| cosine_recall@10 | 0.
|
| 201 |
-
| **cosine_ndcg@10** | **0.
|
| 202 |
-
| cosine_mrr@10 | 0.
|
| 203 |
-
| cosine_map@100 | 0.
|
| 204 |
|
| 205 |
<!--
|
| 206 |
## Bias, Risks and Limitations
|
|
@@ -385,12 +385,22 @@ You can finetune this model on your own dataset.
|
|
| 385 |
| 0.0140 | 40 | 1.4872 | - |
|
| 386 |
| 0.0158 | 45 | 1.4555 | - |
|
| 387 |
| 0.0176 | 50 | 1.4493 | 0.0277 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 388 |
|
| 389 |
|
| 390 |
### Training Time
|
| 391 |
-
- **Training**:
|
| 392 |
-
- **Evaluation**: 0.
|
| 393 |
-
- **Total**:
|
| 394 |
|
| 395 |
### Framework Versions
|
| 396 |
- Python: 3.12.10
|
|
|
|
| 67 |
value: 0.015
|
| 68 |
name: Cosine Accuracy@1
|
| 69 |
- type: cosine_accuracy@10
|
| 70 |
+
value: 0.135
|
| 71 |
name: Cosine Accuracy@10
|
| 72 |
- type: cosine_precision@1
|
| 73 |
value: 0.015
|
| 74 |
name: Cosine Precision@1
|
| 75 |
- type: cosine_precision@10
|
| 76 |
+
value: 0.016
|
| 77 |
name: Cosine Precision@10
|
| 78 |
- type: cosine_recall@1
|
| 79 |
value: 0.005
|
| 80 |
name: Cosine Recall@1
|
| 81 |
- type: cosine_recall@10
|
| 82 |
+
value: 0.05333333333333333
|
| 83 |
name: Cosine Recall@10
|
| 84 |
- type: cosine_ndcg@10
|
| 85 |
+
value: 0.03352606053277749
|
| 86 |
name: Cosine Ndcg@10
|
| 87 |
- type: cosine_mrr@10
|
| 88 |
+
value: 0.04136111111111111
|
| 89 |
name: Cosine Mrr@10
|
| 90 |
- type: cosine_map@100
|
| 91 |
+
value: 0.025214543549657912
|
| 92 |
name: Cosine Map@100
|
| 93 |
---
|
| 94 |
|
|
|
|
| 155 |
# Get the similarity scores for the embeddings
|
| 156 |
similarities = model.similarity(query_embeddings, document_embeddings)
|
| 157 |
print(similarities)
|
| 158 |
+
# tensor([[ 0.8804, -0.1477, 0.3899]])
|
| 159 |
```
|
| 160 |
<!--
|
| 161 |
### Direct Usage (Transformers)
|
|
|
|
| 193 |
| Metric | Value |
|
| 194 |
|:--------------------|:-----------|
|
| 195 |
| cosine_accuracy@1 | 0.015 |
|
| 196 |
+
| cosine_accuracy@10 | 0.135 |
|
| 197 |
| cosine_precision@1 | 0.015 |
|
| 198 |
+
| cosine_precision@10 | 0.016 |
|
| 199 |
| cosine_recall@1 | 0.005 |
|
| 200 |
+
| cosine_recall@10 | 0.0533 |
|
| 201 |
+
| **cosine_ndcg@10** | **0.0335** |
|
| 202 |
+
| cosine_mrr@10 | 0.0414 |
|
| 203 |
+
| cosine_map@100 | 0.0252 |
|
| 204 |
|
| 205 |
<!--
|
| 206 |
## Bias, Risks and Limitations
|
|
|
|
| 385 |
| 0.0140 | 40 | 1.4872 | - |
|
| 386 |
| 0.0158 | 45 | 1.4555 | - |
|
| 387 |
| 0.0176 | 50 | 1.4493 | 0.0277 |
|
| 388 |
+
| 0.0193 | 55 | 1.4075 | - |
|
| 389 |
+
| 0.0211 | 60 | 1.4012 | - |
|
| 390 |
+
| 0.0228 | 65 | 1.4055 | - |
|
| 391 |
+
| 0.0246 | 70 | 1.3977 | - |
|
| 392 |
+
| 0.0263 | 75 | 1.3597 | - |
|
| 393 |
+
| 0.0281 | 80 | 1.3765 | - |
|
| 394 |
+
| 0.0298 | 85 | 1.3657 | - |
|
| 395 |
+
| 0.0316 | 90 | 1.3138 | - |
|
| 396 |
+
| 0.0334 | 95 | 1.3596 | - |
|
| 397 |
+
| 0.0351 | 100 | 1.3428 | 0.0335 |
|
| 398 |
|
| 399 |
|
| 400 |
### Training Time
|
| 401 |
+
- **Training**: 9.2 seconds
|
| 402 |
+
- **Evaluation**: 0.1 seconds
|
| 403 |
+
- **Total**: 9.3 seconds
|
| 404 |
|
| 405 |
### Framework Versions
|
| 406 |
- Python: 3.12.10
|
eval/Information-Retrieval_evaluation_chess-ir_results.csv
CHANGED
|
@@ -1,2 +1,3 @@
|
|
| 1 |
epoch,steps,cosine-Accuracy@1,cosine-Accuracy@10,cosine-Precision@1,cosine-Recall@1,cosine-Precision@10,cosine-Recall@10,cosine-MRR@10,cosine-NDCG@10,cosine-MAP@100
|
| 2 |
0.0175561797752809,50,0.015,0.115,0.015,0.005,0.013000000000000001,0.04333333333333333,0.03541269841269841,0.02770564804107805,0.021195015342589062
|
|
|
|
|
|
| 1 |
epoch,steps,cosine-Accuracy@1,cosine-Accuracy@10,cosine-Precision@1,cosine-Recall@1,cosine-Precision@10,cosine-Recall@10,cosine-MRR@10,cosine-NDCG@10,cosine-MAP@100
|
| 2 |
0.0175561797752809,50,0.015,0.115,0.015,0.005,0.013000000000000001,0.04333333333333333,0.03541269841269841,0.02770564804107805,0.021195015342589062
|
| 3 |
+
0.0351123595505618,100,0.015,0.135,0.015,0.005,0.016,0.05333333333333333,0.04136111111111111,0.03352606053277749,0.025214543549657912
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 4343904
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5d4ef3bb5b44abe3bfb9417a3efa9ca89a41ae764f01f72b81aee073532c9c5f
|
| 3 |
size 4343904
|