| { | |
| "model_type": "multimodal", | |
| "vision_backbone": "google/vit-base-patch16-224", | |
| "num_classes": 7, | |
| "num_tabular_features": 20, | |
| "class_labels": { | |
| "0": "carcinoma_in_situ", | |
| "1": "light_dysplastic", | |
| "2": "moderate_dysplastic", | |
| "3": "normal_columnar", | |
| "4": "normal_intermediate", | |
| "5": "normal_superficiel", | |
| "6": "severe_dysplastic" | |
| }, | |
| "feature_columns": [ | |
| "Kerne_A", | |
| "Cyto_A", | |
| "K/C", | |
| "Kerne_Ycol", | |
| "Cyto_Ycol", | |
| "KerneShort", | |
| "KerneLong", | |
| "KerneElong", | |
| "KerneRund", | |
| "CytoShort", | |
| "CytoLong", | |
| "CytoElong", | |
| "CytoRund", | |
| "KernePeri", | |
| "CytoPeri", | |
| "KernePos", | |
| "KerneMax", | |
| "KerneMin", | |
| "CytoMax", | |
| "CytoMin" | |
| ], | |
| "training_config": { | |
| "batch_size": 16, | |
| "learning_rate": 2e-05, | |
| "num_epochs": 50, | |
| "optimizer": "AdamW", | |
| "scheduler": "CosineAnnealingLR", | |
| "early_stopping_patience": 10 | |
| } | |
| } |