End of training script.
Browse files- README.md +63 -18
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -5,9 +5,35 @@ tags:
|
|
| 5 |
- feature-extraction
|
| 6 |
- dense
|
| 7 |
- generated_from_trainer
|
| 8 |
-
- dataset_size:
|
| 9 |
- loss:ContrastiveLoss
|
| 10 |
base_model: sentence-transformers/clip-ViT-B-32
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
pipeline_tag: sentence-similarity
|
| 12 |
library_name: sentence-transformers
|
| 13 |
---
|
|
@@ -60,9 +86,9 @@ from sentence_transformers import SentenceTransformer
|
|
| 60 |
model = SentenceTransformer("philipp-zettl/MTGEmb-small")
|
| 61 |
# Run inference
|
| 62 |
sentences = [
|
| 63 |
-
'
|
| 64 |
-
|
| 65 |
-
'
|
| 66 |
]
|
| 67 |
embeddings = model.encode(sentences)
|
| 68 |
print(embeddings.shape)
|
|
@@ -71,9 +97,9 @@ print(embeddings.shape)
|
|
| 71 |
# Get the similarity scores for the embeddings
|
| 72 |
similarities = model.similarity(embeddings, embeddings)
|
| 73 |
print(similarities)
|
| 74 |
-
# tensor([[1.0000, 0.
|
| 75 |
-
# [0.
|
| 76 |
-
# [0.
|
| 77 |
```
|
| 78 |
|
| 79 |
<!--
|
|
@@ -118,19 +144,19 @@ You can finetune this model on your own dataset.
|
|
| 118 |
|
| 119 |
#### Unnamed Dataset
|
| 120 |
|
| 121 |
-
* Size:
|
| 122 |
* Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
|
| 123 |
-
* Approximate statistics based on the first
|
| 124 |
-
| | sentence_0
|
| 125 |
-
|
| 126 |
-
| type | string
|
| 127 |
-
| details | <ul><li>min: 3 tokens</li><li>mean: 16
|
| 128 |
* Samples:
|
| 129 |
-
| sentence_0
|
| 130 |
-
|
| 131 |
-
| <code>
|
| 132 |
-
| <code>
|
| 133 |
-
| <code>
|
| 134 |
* Loss: [<code>ContrastiveLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#contrastiveloss) with these parameters:
|
| 135 |
```json
|
| 136 |
{
|
|
@@ -270,6 +296,25 @@ You can finetune this model on your own dataset.
|
|
| 270 |
|
| 271 |
</details>
|
| 272 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 273 |
### Framework Versions
|
| 274 |
- Python: 3.13.7
|
| 275 |
- Sentence Transformers: 5.1.2
|
|
|
|
| 5 |
- feature-extraction
|
| 6 |
- dense
|
| 7 |
- generated_from_trainer
|
| 8 |
+
- dataset_size:149460
|
| 9 |
- loss:ContrastiveLoss
|
| 10 |
base_model: sentence-transformers/clip-ViT-B-32
|
| 11 |
+
widget:
|
| 12 |
+
- source_sentence: Meltdown
|
| 13 |
+
sentences:
|
| 14 |
+
- Ancient Imperiosaur
|
| 15 |
+
- https://cards.scryfall.io/normal/front/1/9/192ccc7f-ffb1-4f78-8cf0-a220df612be7.jpg?1682536817
|
| 16 |
+
- https://cards.scryfall.io/normal/front/5/6/56301392-3496-48d0-8d91-6b82e1164c98.jpg?1721427942
|
| 17 |
+
- source_sentence: Etali, Primal Storm
|
| 18 |
+
sentences:
|
| 19 |
+
- https://cards.scryfall.io/normal/front/4/8/4874388e-0227-4b89-a986-d86c14482c81.jpg?1594065427
|
| 20 |
+
- Battle of Wits
|
| 21 |
+
- https://cards.scryfall.io/normal/front/1/d/1d3d8bb4-0430-45bb-930d-5d6db6521945.jpg?1587309687
|
| 22 |
+
- source_sentence: Chrome Prowler
|
| 23 |
+
sentences:
|
| 24 |
+
- https://cards.scryfall.io/normal/front/a/2/a263f594-621e-46af-8561-f7eee565a19a.jpg?1562643297
|
| 25 |
+
- https://cards.scryfall.io/normal/front/3/d/3dff363d-7e9f-4764-a9ee-ec2f23239df6.jpg?1562907900
|
| 26 |
+
- https://cards.scryfall.io/normal/front/2/1/21121857-85b8-4ba8-9363-beafdb1005c2.jpg?1730486782
|
| 27 |
+
- source_sentence: Beastbreaker of Bala Ged
|
| 28 |
+
sentences:
|
| 29 |
+
- https://cards.scryfall.io/normal/front/2/8/287ca034-9cea-4b84-98ba-76c24f038edb.jpg?1599709496
|
| 30 |
+
- https://cards.scryfall.io/normal/front/5/4/547f2641-bcd6-4536-ba5a-f46170dd2803.jpg?1573513110
|
| 31 |
+
- https://cards.scryfall.io/normal/front/4/c/4c29f6a1-42a5-433f-9c09-c160b096f8e1.jpg?1562542378
|
| 32 |
+
- source_sentence: Against All Odds
|
| 33 |
+
sentences:
|
| 34 |
+
- https://cards.scryfall.io/normal/front/4/a/4ab2f81a-fcbe-44d1-8281-04dd78bb9ea3.jpg?1593274931
|
| 35 |
+
- https://cards.scryfall.io/normal/front/3/c/3cd8dd4e-6892-49d7-8fae-97d04f9f6c84.jpg?1675956885
|
| 36 |
+
- Sheltering Prayers
|
| 37 |
pipeline_tag: sentence-similarity
|
| 38 |
library_name: sentence-transformers
|
| 39 |
---
|
|
|
|
| 86 |
model = SentenceTransformer("philipp-zettl/MTGEmb-small")
|
| 87 |
# Run inference
|
| 88 |
sentences = [
|
| 89 |
+
'Against All Odds',
|
| 90 |
+
'https://cards.scryfall.io/normal/front/3/c/3cd8dd4e-6892-49d7-8fae-97d04f9f6c84.jpg?1675956885',
|
| 91 |
+
'https://cards.scryfall.io/normal/front/4/a/4ab2f81a-fcbe-44d1-8281-04dd78bb9ea3.jpg?1593274931',
|
| 92 |
]
|
| 93 |
embeddings = model.encode(sentences)
|
| 94 |
print(embeddings.shape)
|
|
|
|
| 97 |
# Get the similarity scores for the embeddings
|
| 98 |
similarities = model.similarity(embeddings, embeddings)
|
| 99 |
print(similarities)
|
| 100 |
+
# tensor([[1.0000, 0.9248, 0.6695],
|
| 101 |
+
# [0.9248, 1.0000, 0.6947],
|
| 102 |
+
# [0.6695, 0.6947, 1.0000]])
|
| 103 |
```
|
| 104 |
|
| 105 |
<!--
|
|
|
|
| 144 |
|
| 145 |
#### Unnamed Dataset
|
| 146 |
|
| 147 |
+
* Size: 149,460 training samples
|
| 148 |
* Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
|
| 149 |
+
* Approximate statistics based on the first 1000 samples:
|
| 150 |
+
| | sentence_0 | sentence_1 | label |
|
| 151 |
+
|:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:------------------------------------------------|
|
| 152 |
+
| type | string | string | int |
|
| 153 |
+
| details | <ul><li>min: 3 tokens</li><li>mean: 17.16 tokens</li><li>max: 69 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 54.64 tokens</li><li>max: 69 tokens</li></ul> | <ul><li>0: ~57.60%</li><li>1: ~42.40%</li></ul> |
|
| 154 |
* Samples:
|
| 155 |
+
| sentence_0 | sentence_1 | label |
|
| 156 |
+
|:----------------------------------|:------------------------------------------------------------------------------------------------------------|:---------------|
|
| 157 |
+
| <code>Comparative Analysis</code> | <code>https://cards.scryfall.io/normal/front/d/d/dd83129b-7e8c-4cc5-a7b3-e0ae221d7ad4.jpg?1562939549</code> | <code>1</code> |
|
| 158 |
+
| <code>Breathkeeper Seraph</code> | <code>https://cards.scryfall.io/normal/front/1/b/1bdd3ecb-8c11-4a4c-a503-bc29f79a9dcb.jpg?1682204691</code> | <code>0</code> |
|
| 159 |
+
| <code>Wei Infantry</code> | <code>https://cards.scryfall.io/normal/front/7/2/72c6465f-3144-4faf-b248-a9fb941dc002.jpg?1562257016</code> | <code>1</code> |
|
| 160 |
* Loss: [<code>ContrastiveLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#contrastiveloss) with these parameters:
|
| 161 |
```json
|
| 162 |
{
|
|
|
|
| 296 |
|
| 297 |
</details>
|
| 298 |
|
| 299 |
+
### Training Logs
|
| 300 |
+
| Epoch | Step | Training Loss |
|
| 301 |
+
|:------:|:----:|:-------------:|
|
| 302 |
+
| 0.2140 | 500 | 0.0342 |
|
| 303 |
+
| 0.4281 | 1000 | 0.0311 |
|
| 304 |
+
| 0.6421 | 1500 | 0.0306 |
|
| 305 |
+
| 0.8562 | 2000 | 0.0302 |
|
| 306 |
+
| 1.0702 | 2500 | 0.0287 |
|
| 307 |
+
| 1.2842 | 3000 | 0.0262 |
|
| 308 |
+
| 1.4983 | 3500 | 0.025 |
|
| 309 |
+
| 1.7123 | 4000 | 0.0236 |
|
| 310 |
+
| 1.9264 | 4500 | 0.022 |
|
| 311 |
+
| 2.1404 | 5000 | 0.016 |
|
| 312 |
+
| 2.3545 | 5500 | 0.0128 |
|
| 313 |
+
| 2.5685 | 6000 | 0.0119 |
|
| 314 |
+
| 2.7825 | 6500 | 0.0108 |
|
| 315 |
+
| 2.9966 | 7000 | 0.0103 |
|
| 316 |
+
|
| 317 |
+
|
| 318 |
### Framework Versions
|
| 319 |
- Python: 3.13.7
|
| 320 |
- Sentence Transformers: 5.1.2
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 605156676
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:736ecbd11008ba396ab46fc621960e2b4e5208e9d7f029b61b72bfa64c918b10
|
| 3 |
size 605156676
|