Table 3 Sample training data from breast cancer dataset.
Hyperparameter | Value |
|---|---|
Input image size | 224 × 224 |
Batch size | 16 |
Optimizer | Adam |
Learning rate | 0.0001 |
Loss function | Categorical cross-entropy |
Number of epochs | 8 |
Weight initialization | ImageNet |
Data augmentation | Rescaling, Shear, Zoom, Horizontal Flip |
Preprocessing | Normalization [0,1] |
Evaluation metric | Accuracy, MAE, MCC |
