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 250
Browse files- README.md +31 -21
- eval/Information-Retrieval_evaluation_chess-ir_results.csv +1 -0
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -64,31 +64,31 @@ model-index:
|
|
| 64 |
type: chess-ir
|
| 65 |
metrics:
|
| 66 |
- type: cosine_accuracy@1
|
| 67 |
-
value: 0.
|
| 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.
|
| 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.
|
| 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)
|
|
@@ -192,15 +192,15 @@ You can finetune this model on your own dataset.
|
|
| 192 |
|
| 193 |
| Metric | Value |
|
| 194 |
|:--------------------|:-----------|
|
| 195 |
-
| cosine_accuracy@1 | 0.
|
| 196 |
-
| cosine_accuracy@10 | 0.
|
| 197 |
-
| cosine_precision@1 | 0.
|
| 198 |
-
| cosine_precision@10 | 0.
|
| 199 |
-
| cosine_recall@1 | 0.
|
| 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
|
|
@@ -415,12 +415,22 @@ You can finetune this model on your own dataset.
|
|
| 415 |
| 0.0667 | 190 | 1.3124 | - |
|
| 416 |
| 0.0685 | 195 | 1.3237 | - |
|
| 417 |
| 0.0702 | 200 | 1.3051 | 0.0451 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 418 |
|
| 419 |
|
| 420 |
### Training Time
|
| 421 |
-
- **Training**:
|
| 422 |
- **Evaluation**: 0.2 seconds
|
| 423 |
-
- **Total**:
|
| 424 |
|
| 425 |
### Framework Versions
|
| 426 |
- Python: 3.12.10
|
|
|
|
| 64 |
type: chess-ir
|
| 65 |
metrics:
|
| 66 |
- type: cosine_accuracy@1
|
| 67 |
+
value: 0.025
|
| 68 |
name: Cosine Accuracy@1
|
| 69 |
- type: cosine_accuracy@10
|
| 70 |
+
value: 0.14
|
| 71 |
name: Cosine Accuracy@10
|
| 72 |
- type: cosine_precision@1
|
| 73 |
+
value: 0.025
|
| 74 |
name: Cosine Precision@1
|
| 75 |
- type: cosine_precision@10
|
| 76 |
+
value: 0.017
|
| 77 |
name: Cosine Precision@10
|
| 78 |
- type: cosine_recall@1
|
| 79 |
+
value: 0.008333333333333333
|
| 80 |
name: Cosine Recall@1
|
| 81 |
- type: cosine_recall@10
|
| 82 |
+
value: 0.056666666666666664
|
| 83 |
name: Cosine Recall@10
|
| 84 |
- type: cosine_ndcg@10
|
| 85 |
+
value: 0.037406426241984
|
| 86 |
name: Cosine Ndcg@10
|
| 87 |
- type: cosine_mrr@10
|
| 88 |
+
value: 0.049240079365079355
|
| 89 |
name: Cosine Mrr@10
|
| 90 |
- type: cosine_map@100
|
| 91 |
+
value: 0.02874627448743367
|
| 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.9411, -0.1930, 0.3964]])
|
| 159 |
```
|
| 160 |
<!--
|
| 161 |
### Direct Usage (Transformers)
|
|
|
|
| 192 |
|
| 193 |
| Metric | Value |
|
| 194 |
|:--------------------|:-----------|
|
| 195 |
+
| cosine_accuracy@1 | 0.025 |
|
| 196 |
+
| cosine_accuracy@10 | 0.14 |
|
| 197 |
+
| cosine_precision@1 | 0.025 |
|
| 198 |
+
| cosine_precision@10 | 0.017 |
|
| 199 |
+
| cosine_recall@1 | 0.0083 |
|
| 200 |
+
| cosine_recall@10 | 0.0567 |
|
| 201 |
+
| **cosine_ndcg@10** | **0.0374** |
|
| 202 |
+
| cosine_mrr@10 | 0.0492 |
|
| 203 |
+
| cosine_map@100 | 0.0287 |
|
| 204 |
|
| 205 |
<!--
|
| 206 |
## Bias, Risks and Limitations
|
|
|
|
| 415 |
| 0.0667 | 190 | 1.3124 | - |
|
| 416 |
| 0.0685 | 195 | 1.3237 | - |
|
| 417 |
| 0.0702 | 200 | 1.3051 | 0.0451 |
|
| 418 |
+
| 0.0720 | 205 | 1.2801 | - |
|
| 419 |
+
| 0.0737 | 210 | 1.3404 | - |
|
| 420 |
+
| 0.0755 | 215 | 1.2916 | - |
|
| 421 |
+
| 0.0772 | 220 | 1.2981 | - |
|
| 422 |
+
| 0.0790 | 225 | 1.3321 | - |
|
| 423 |
+
| 0.0808 | 230 | 1.3369 | - |
|
| 424 |
+
| 0.0825 | 235 | 1.3059 | - |
|
| 425 |
+
| 0.0843 | 240 | 1.3213 | - |
|
| 426 |
+
| 0.0860 | 245 | 1.3127 | - |
|
| 427 |
+
| 0.0878 | 250 | 1.2801 | 0.0374 |
|
| 428 |
|
| 429 |
|
| 430 |
### Training Time
|
| 431 |
+
- **Training**: 21.2 seconds
|
| 432 |
- **Evaluation**: 0.2 seconds
|
| 433 |
+
- **Total**: 21.5 seconds
|
| 434 |
|
| 435 |
### Framework Versions
|
| 436 |
- Python: 3.12.10
|
eval/Information-Retrieval_evaluation_chess-ir_results.csv
CHANGED
|
@@ -3,3 +3,4 @@ epoch,steps,cosine-Accuracy@1,cosine-Accuracy@10,cosine-Precision@1,cosine-Recal
|
|
| 3 |
0.0351123595505618,100,0.015,0.135,0.015,0.005,0.016,0.05333333333333333,0.04136111111111111,0.03352606053277749,0.025214543549657912
|
| 4 |
0.05266853932584269,150,0.02,0.12,0.02,0.006666666666666666,0.0155,0.051666666666666666,0.04391468253968253,0.034539315152376744,0.02851338765635309
|
| 5 |
0.0702247191011236,200,0.03,0.16,0.03,0.009999999999999998,0.02,0.06666666666666667,0.05857142857142858,0.045080933582823335,0.033163497941181515
|
|
|
|
|
|
| 3 |
0.0351123595505618,100,0.015,0.135,0.015,0.005,0.016,0.05333333333333333,0.04136111111111111,0.03352606053277749,0.025214543549657912
|
| 4 |
0.05266853932584269,150,0.02,0.12,0.02,0.006666666666666666,0.0155,0.051666666666666666,0.04391468253968253,0.034539315152376744,0.02851338765635309
|
| 5 |
0.0702247191011236,200,0.03,0.16,0.03,0.009999999999999998,0.02,0.06666666666666667,0.05857142857142858,0.045080933582823335,0.033163497941181515
|
| 6 |
+
0.0877808988764045,250,0.025,0.14,0.025,0.008333333333333333,0.017,0.056666666666666664,0.049240079365079355,0.037406426241984,0.02874627448743367
|
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:6414613fb87664a043eac994032609e6a48c5affbe464c8bb279f06170c4aeb2
|
| 3 |
size 4343904
|