| license: cc-by-nc-4.0 | |
| tags: | |
| - pathology | |
| - cell-classification | |
| - pytorch | |
| - torch-export | |
| # MetPredict-cell-reclassifier | |
| Per-cell MLP classifier. Maps **histoplus 768-dim cell features** to | |
| softmax probabilities over: `['background', 'tumor']`. | |
| Exported with `torch.export` (batch dim: `dynamic`, feature dim: `768`). | |
| The exported program wraps `MLP + softmax`, so the output is probabilities | |
| directly — no model class required at load time. | |
| ## Usage | |
| ```python | |
| from huggingface_hub import hf_hub_download | |
| import json, torch | |
| path = hf_hub_download("RendeiroLab/MetPredict-cell-reclassifier", "model.pt2") | |
| labels = json.loads(open(hf_hub_download("RendeiroLab/MetPredict-cell-reclassifier", "labels.json")).read())["labels"] | |
| m = torch.export.load(path).module() | |
| probs = m(torch.randn(N, 768)) # (N, 2), softmaxed | |
| pred_idx = probs.argmax(-1) | |
| pred_class = [labels[i] for i in pred_idx.tolist()] | |
| ``` | |
| ## Files | |
| - `model.pt2` — `torch.export` program (MLP + softmax) | |
| - `labels.json` — class index → name map, plus `in_dim` / `n_classes` / `batch` | |
| - `README.md` — this file | |