End of training script.

README.md

@@ -5,59 +5,11 @@ tags:
 - feature-extraction
 - dense
 - generated_from_trainer
-- dataset_size:294252
+- dataset_size:5
 - loss:ContrastiveLoss
 base_model: sentence-transformers/clip-ViT-B-32
-widget:
-- source_sentence: Ranger's Guile
-  sentences:
-  - >-
-    https://cards.scryfall.io/normal/front/0/1/01388e24-d66f-40a8-af41-0f84dd8ed3fe.jpg?1562781768
-  - >-
-    https://cards.scryfall.io/normal/front/2/0/20995638-e217-49f1-bccc-d1a344694fd8.jpg?1592711262
-  - >-
-    https://cards.scryfall.io/normal/front/1/3/13680953-cf05-4e38-a3cf-22900c02fab7.jpg?1706240764
-- source_sentence: |-
-    {2}, {T}: Choose one that hasn't been chosen —
-    • This artifact deals 2 damage to target creature.
-    • Tap target creature.
-    • Sacrifice this artifact. You gain 3 life.
-  sentences:
-  - Haunted Ridge
-  - >-
-    https://cards.scryfall.io/normal/front/1/e/1ebb7ade-ebbb-49fb-b64f-fa247f9a9af6.jpg?1562544721
-  - Three Bowls of Porridge
-- source_sentence: Soratami Mindsweeper
-  sentences:
-  - >-
-    https://cards.scryfall.io/normal/front/c/1/c13eacff-0fd1-4e2e-8213-5bf5fff973c8.jpg?1593095724
-  - >-
-    https://cards.scryfall.io/normal/front/6/7/67f55ef7-8479-4d56-9801-cb4fd5526e73.jpg?1675644776
-  - >-
-    https://cards.scryfall.io/normal/front/0/e/0ef3b83a-777c-45b6-97a1-25be237df79b.jpg?1561932561
-- source_sentence: |-
-    Flying
-    Ward—Sacrifice a creature.
-    Whenever a creature dies, target opponent loses 2 life and you gain 2 life.
-  sentences:
-  - >-
-    https://cards.scryfall.io/normal/front/e/4/e4c7919a-de62-46df-a937-593647473ef5.jpg?1654567436
-  - >-
-    https://cards.scryfall.io/normal/front/9/7/9712ecaa-4059-44ba-98b7-07bfe7411b5b.jpg?1562447340
-  - >-
-    https://cards.scryfall.io/normal/front/0/7/078933b3-6d82-45f2-94e8-addf54cf1704.jpg?1706241798
-- source_sentence: Glimpse of Freedom
-  sentences:
-  - >-
-    https://cards.scryfall.io/normal/front/c/1/c1e9c025-1cdb-4da8-8d28-14ad5efb512d.jpg?1581479361
-  - >-
-    https://cards.scryfall.io/normal/front/0/7/0770eb6a-4f01-4677-a401-14c1b30692c9.jpg?1673306796
-  - >-
-    https://cards.scryfall.io/normal/front/3/3/333039ef-b328-4082-85ae-162caed5612e.jpg?1738358111
 pipeline_tag: sentence-similarity
 library_name: sentence-transformers
-datasets:
-- philipp-zettl/mtg_cards-2025-04-04
 ---
 
 # SentenceTransformer based on sentence-transformers/clip-ViT-B-32
@@ -108,9 +60,9 @@ from sentence_transformers import SentenceTransformer
 model = SentenceTransformer("philipp-zettl/MTGEmb-small")
 # Run inference
 sentences = [
-    'Glimpse of Freedom',
-    'https://cards.scryfall.io/normal/front/c/1/c1e9c025-1cdb-4da8-8d28-14ad5efb512d.jpg?1581479361',
-    'https://cards.scryfall.io/normal/front/3/3/333039ef-b328-4082-85ae-162caed5612e.jpg?1738358111',
+    'The weather is lovely today.',
+    "It's so sunny outside!",
+    'He drove to the stadium.',
 ]
 embeddings = model.encode(sentences)
 print(embeddings.shape)
@@ -119,9 +71,9 @@ print(embeddings.shape)
 # Get the similarity scores for the embeddings
 similarities = model.similarity(embeddings, embeddings)
 print(similarities)
-# tensor([[1.0000, 0.7515, 0.7515],
-#         [0.7515, 1.0000, 1.0000],
-#         [0.7515, 1.0000, 1.0000]])
+# tensor([[1.0000, 0.9425, 0.8177],
+#         [0.9425, 1.0000, 0.8015],
+#         [0.8177, 0.8015, 1.0000]])
 ```
 
 <!--
@@ -166,19 +118,19 @@ You can finetune this model on your own dataset.
 
 #### Unnamed Dataset
 
-* Size: 294,252 training samples
+* Size: 5 training samples
 * Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
-* Approximate statistics based on the first 1000 samples:
-  |         | sentence_0                                                                        | sentence_1                                                                        | label                                           |
-  |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:------------------------------------------------|
-  | type    | string                                                                            | string                                                                            | int                                             |
-  | details | <ul><li>min: 3 tokens</li><li>mean: 25.44 tokens</li><li>max: 77 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 54.71 tokens</li><li>max: 69 tokens</li></ul> | <ul><li>0: ~38.90%</li><li>1: ~61.10%</li></ul> |
+* Approximate statistics based on the first 5 samples:
+  |         | sentence_0                                                                       | sentence_1                                                                       | label                                           |
+  |:--------|:---------------------------------------------------------------------------------|:---------------------------------------------------------------------------------|:------------------------------------------------|
+  | type    | string                                                                           | string                                                                           | int                                             |
+  | details | <ul><li>min: 3 tokens</li><li>mean: 16.8 tokens</li><li>max: 66 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 53.6 tokens</li><li>max: 66 tokens</li></ul> | <ul><li>0: ~60.00%</li><li>1: ~40.00%</li></ul> |
 * Samples:
-  | sentence_0                                                                                                                                                                                                                                                                                                                                          | sentence_1                                                                                                  | label          |
-  |:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------|:---------------|
-  | <code>Flying<br>Ascend (If you control ten or more permanents, you get the city's blessing for the rest of the game.)<br>At the beginning of your upkeep, if you have the city's blessing, reveal the top card of your library and put it into your hand. Each opponent loses X life and you gain X life, where X is that card's mana value.</code> | <code>Twilight Prophet</code>                                                                               | <code>1</code> |
-  | <code>Target creature gains deathtouch until end of turn and must be blocked this turn if able.<br>Flashback {G} (You may cast this card from your graveyard for its flashback cost. Then exile it.)</code>                                                                                                                                         | <code>https://cards.scryfall.io/normal/front/f/2/f2c97632-3cf1-4b79-9d18-d8991654dcca.jpg?1562796910</code> | <code>0</code> |
-  | <code>Pay 1 life, Sacrifice another creature: Create a Treasure token. (It's an artifact with "{T}, Sacrifice this token: Add one mana of any color.")</code>                                                                                                                                                                                       | <code>https://cards.scryfall.io/normal/front/7/7/77c0929f-bee6-416d-8571-6540ae4f6e4f.jpg?1562879392</code> | <code>0</code> |
+  | sentence_0                   | sentence_1                                                                                                  | label          |
+  |:-----------------------------|:------------------------------------------------------------------------------------------------------------|:---------------|
+  | <code>Veteran Armorer</code> | <code>https://cards.scryfall.io/normal/front/0/0/0000419b-0bba-4488-8f7a-6194544ce91e.jpg?1721427487</code> | <code>0</code> |
+  | <code>Forest</code>          | <code>https://cards.scryfall.io/normal/front/0/0/0000419b-0bba-4488-8f7a-6194544ce91e.jpg?1721427487</code> | <code>1</code> |
+  | <code>Forest</code>          | <code>Veteran Armorer</code>                                                                                | <code>0</code> |
 * Loss: [<code>ContrastiveLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#contrastiveloss) with these parameters:
   ```json
   {
@@ -191,9 +143,8 @@ You can finetune this model on your own dataset.
 ### Training Hyperparameters
 #### Non-Default Hyperparameters
 
-- `per_device_train_batch_size`: 56
-- `per_device_eval_batch_size`: 56
-- `num_train_epochs`: 5
+- `per_device_train_batch_size`: 64
+- `per_device_eval_batch_size`: 64
 - `multi_dataset_batch_sampler`: round_robin
 
 #### All Hyperparameters
@@ -203,8 +154,8 @@ You can finetune this model on your own dataset.
 - `do_predict`: False
 - `eval_strategy`: no
 - `prediction_loss_only`: True
-- `per_device_train_batch_size`: 56
-- `per_device_eval_batch_size`: 56
+- `per_device_train_batch_size`: 64
+- `per_device_eval_batch_size`: 64
 - `per_gpu_train_batch_size`: None
 - `per_gpu_eval_batch_size`: None
 - `gradient_accumulation_steps`: 1
@@ -216,7 +167,7 @@ You can finetune this model on your own dataset.
 - `adam_beta2`: 0.999
 - `adam_epsilon`: 1e-08
 - `max_grad_norm`: 1
-- `num_train_epochs`: 5
+- `num_train_epochs`: 3
 - `max_steps`: -1
 - `lr_scheduler_type`: linear
 - `lr_scheduler_kwargs`: {}
@@ -319,63 +270,6 @@ You can finetune this model on your own dataset.
 
 </details>
 
-### Training Logs
-| Epoch  | Step  | Training Loss |
-|:------:|:-----:|:-------------:|
-| 0.0951 | 500   | 0.0269        |
-| 0.1903 | 1000  | 0.0253        |
-| 0.2854 | 1500  | 0.0253        |
-| 0.3806 | 2000  | 0.0252        |
-| 0.4757 | 2500  | 0.0252        |
-| 0.5709 | 3000  | 0.0253        |
-| 0.6660 | 3500  | 0.0252        |
-| 0.7612 | 4000  | 0.0251        |
-| 0.8563 | 4500  | 0.0251        |
-| 0.9515 | 5000  | 0.0251        |
-| 1.0466 | 5500  | 0.0251        |
-| 1.1418 | 6000  | 0.0251        |
-| 1.2369 | 6500  | 0.0252        |
-| 1.3321 | 7000  | 0.0251        |
-| 1.4272 | 7500  | 0.0252        |
-| 1.5224 | 8000  | 0.0252        |
-| 1.6175 | 8500  | 0.0251        |
-| 1.7127 | 9000  | 0.0249        |
-| 1.8078 | 9500  | 0.0252        |
-| 1.9029 | 10000 | 0.025         |
-| 1.9981 | 10500 | 0.0252        |
-| 2.0932 | 11000 | 0.0251        |
-| 2.1884 | 11500 | 0.0251        |
-| 2.2835 | 12000 | 0.0252        |
-| 2.3787 | 12500 | 0.0252        |
-| 2.4738 | 13000 | 0.0251        |
-| 2.5690 | 13500 | 0.025         |
-| 2.6641 | 14000 | 0.025         |
-| 2.7593 | 14500 | 0.0251        |
-| 2.8544 | 15000 | 0.0251        |
-| 2.9496 | 15500 | 0.0251        |
-| 3.0447 | 16000 | 0.0251        |
-| 3.1399 | 16500 | 0.0252        |
-| 3.2350 | 17000 | 0.0251        |
-| 3.3302 | 17500 | 0.025         |
-| 3.4253 | 18000 | 0.0252        |
-| 3.5205 | 18500 | 0.0251        |
-| 3.6156 | 19000 | 0.025         |
-| 3.7108 | 19500 | 0.0251        |
-| 3.8059 | 20000 | 0.0252        |
-| 3.9010 | 20500 | 0.025         |
-| 3.9962 | 21000 | 0.025         |
-| 4.0913 | 21500 | 0.0251        |
-| 4.1865 | 22000 | 0.0251        |
-| 4.2816 | 22500 | 0.025         |
-| 4.3768 | 23000 | 0.025         |
-| 4.4719 | 23500 | 0.025         |
-| 4.5671 | 24000 | 0.0252        |
-| 4.6622 | 24500 | 0.025         |
-| 4.7574 | 25000 | 0.0252        |
-| 4.8525 | 25500 | 0.025         |
-| 4.9477 | 26000 | 0.025         |
-
-
 ### Framework Versions
 - Python: 3.13.7
 - Sentence Transformers: 5.1.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:341fc80e886d58cc50059de8b9b063bd1b9a39dc246dd10f13db826617cc9d66
+oid sha256:3b41f0699e0483f5792e303ee4230710736646a005a402e19017010acb4607ad
 size 605156676