Table 3. Hyperparameters used in the proposed individual transfer learning models and an ensemble model.
Hyperparameters | Value |
---|---|
Image size | 224 × 224 |
Optimizer | 0.005 |
Maximum Habitat probability | SGD with momentum |
Learning rate | 1e−6 |
Batch size | 16 |
Number of epochs | 10 |
Dropout | 0.3 |
Loss function | Cross Entropy |