| { | |
| "_name_or_path": "ErnestBeckham/MulticancerViT", | |
| "architectures": ["ViTForImageClassification"], | |
| "id2label": { | |
| "0": "cervix_koc", | |
| "1": "cervix_dyk", | |
| "2": "cervix_pab", | |
| "3": "cervix_sfi", | |
| "4": "cervix_mep", | |
| "5": "colon_bnt", | |
| "6": "colon_aca", | |
| "7": "lung_aca", | |
| "8": "lung_bnt", | |
| "9": "lung_scc", | |
| "10": "oral_scc", | |
| "11": "oral_normal", | |
| "12":"kidney_tumor", | |
| "13":"kidney_normal", | |
| "14":"breast_benign", | |
| "15":"breast_malignant", | |
| "16":"lymph_fl", | |
| "17":"lymph_cll", | |
| "18":"lymph_mcl", | |
| "19":"brain_tumor", | |
| "20":"brain_glioma", | |
| "21":"brain_menin" | |
| }, | |
| "model_type": "vit", | |
| "lr":1e-4, | |
| "num_layers":12, | |
| "hidden_dim":512, | |
| "mlp_dim":12, | |
| "dropout":0.1 | |
| } | |