Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -18,8 +18,10 @@ Given a Commander (and optionally a partial decklist), suggests which other
|
|
| 18 |
cards belong in the deck. Trained on 122,571 real public Commander decklists
|
| 19 |
from [Moxfield](https://moxfield.com) ([dataset](https://huggingface.co/datasets/nsaroiu/moxfield-dump)).
|
| 20 |
|
| 21 |
-
|
| 22 |
-
|
|
|
|
|
|
|
| 23 |
|
| 24 |
## Architecture
|
| 25 |
|
|
@@ -81,16 +83,17 @@ generalization via card content rather than memorized co-occurrence):
|
|
| 81 |
Recall@50 0.27–0.41 across the same ratio range — meaningfully above chance,
|
| 82 |
confirming the hybrid content embeddings carry real signal for unfamiliar
|
| 83 |
commanders. Full breakdown by card-popularity tier, plus the synergy/pruning
|
| 84 |
-
checkpoints' own numbers, in the GitHub repo.
|
| 85 |
|
| 86 |
## Usage
|
| 87 |
|
| 88 |
-
There's no standalone pip package yet — loading these checkpoints
|
| 89 |
-
|
|
|
|
| 90 |
|
| 91 |
```bash
|
| 92 |
-
git clone https://github.com/nsaroiu/mtg-
|
| 93 |
-
cd mtg-
|
| 94 |
pip install -r requirements.txt
|
| 95 |
```
|
| 96 |
|
|
@@ -121,8 +124,12 @@ for name, score, source in results:
|
|
| 121 |
print(f"{score:.3f} [{source:7}] {name}")
|
| 122 |
```
|
| 123 |
|
| 124 |
-
|
| 125 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
|
| 127 |
## Limitations
|
| 128 |
|
|
|
|
| 18 |
cards belong in the deck. Trained on 122,571 real public Commander decklists
|
| 19 |
from [Moxfield](https://moxfield.com) ([dataset](https://huggingface.co/datasets/nsaroiu/moxfield-dump)).
|
| 20 |
|
| 21 |
+
Training/evaluation code: [github.com/nsaroiu/mtg-deck-completion](https://github.com/nsaroiu/mtg-deck-completion)
|
| 22 |
+
(the tokenizer + model + training/eval pipeline that produced these weights — not the
|
| 23 |
+
Moxfield data-collection pipeline or the web UI built on top of this model, which live
|
| 24 |
+
elsewhere and aren't public).
|
| 25 |
|
| 26 |
## Architecture
|
| 27 |
|
|
|
|
| 83 |
Recall@50 0.27–0.41 across the same ratio range — meaningfully above chance,
|
| 84 |
confirming the hybrid content embeddings carry real signal for unfamiliar
|
| 85 |
commanders. Full breakdown by card-popularity tier, plus the synergy/pruning
|
| 86 |
+
checkpoints' own numbers, in the GitHub repo's eval output.
|
| 87 |
|
| 88 |
## Usage
|
| 89 |
|
| 90 |
+
There's no standalone pip package yet — loading these checkpoints requires
|
| 91 |
+
the tokenizer + model code from the GitHub repo above (`tokenizer/` and
|
| 92 |
+
`training/`; no other part of that repo is needed to run inference):
|
| 93 |
|
| 94 |
```bash
|
| 95 |
+
git clone https://github.com/nsaroiu/mtg-deck-completion
|
| 96 |
+
cd mtg-deck-completion
|
| 97 |
pip install -r requirements.txt
|
| 98 |
```
|
| 99 |
|
|
|
|
| 124 |
print(f"{score:.3f} [{source:7}] {name}")
|
| 125 |
```
|
| 126 |
|
| 127 |
+
`training/ensemble.py` is also a CLI covering the same flow shown above
|
| 128 |
+
(`--commander` — repeatable, for partner commanders — `--card`,
|
| 129 |
+
`--chunk-size`/`--temperature` for the iterative synergy phase,
|
| 130 |
+
`--prune-checkpoint` to enable the pruning pass). `training/complete_deck.py`
|
| 131 |
+
exposes just one specialist checkpoint at a time, without the ensemble
|
| 132 |
+
fusion, if that's all you need.
|
| 133 |
|
| 134 |
## Limitations
|
| 135 |
|