nsaroiu commited on
Commit
ade41fa
·
verified ·
1 Parent(s): ee65391

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +16 -9
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
- Full code, architecture writeup, and training history:
22
- [github.com/nsaroiu/mtg-card-embeddings](https://github.com/nsaroiu/mtg-card-embeddings).
 
 
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 currently
89
- requires the training code from the GitHub repo above:
 
90
 
91
  ```bash
92
- git clone https://github.com/nsaroiu/mtg-card-embeddings
93
- cd mtg-card-embeddings
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
- See `training/complete_deck.py` for the CLI, or `ui/` for a full web UI
125
- built on this model.
 
 
 
 
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