SkinCancerViT / config.json
mrs83's picture
Upload 3 files
088aafb verified
{
"age_max": 85.0,
"age_mean": 51.91346355807603,
"age_min": 0.0,
"age_std": 16.930758000593915,
"architectures": [
"SkinCancerViTModel"
],
"id2label": {
"0": "actinic_keratoses",
"1": "basal_cell_carcinoma",
"2": "benign_keratosis-like_lesions",
"3": "dermatofibroma",
"4": "melanocytic_Nevi",
"5": "melanoma",
"6": "vascular_lesions"
},
"label2id": {
"actinic_keratoses": 0,
"basal_cell_carcinoma": 1,
"benign_keratosis-like_lesions": 2,
"dermatofibroma": 3,
"melanocytic_Nevi": 4,
"melanoma": 5,
"vascular_lesions": 6
},
"localization_to_id": {
"abdomen": 0,
"acral": 1,
"back": 2,
"chest": 3,
"ear": 4,
"face": 5,
"foot": 6,
"genital": 7,
"hand": 8,
"lower extremity": 9,
"neck": 10,
"scalp": 11,
"trunk": 12,
"unknown": 13,
"upper extremity": 14
},
"model_type": "combined_multimodal_skin_cancer",
"num_dx_labels": 7,
"num_localization_features": 15,
"torch_dtype": "float32",
"total_tabular_features_dim": 16,
"transformers_version": "4.53.2",
"vision_model_checkpoint": "google/vit-base-patch16-224-in21k"
}