Table 14. Configuration of optimal model based on hyper-parameters.
Configuration | Value |
---|---|
Input shape | 224 × 224 |
Epochs | 200 |
Optimization function | Adam |
Learning rate | 0.001 |
Batch size | 32 |
Weight decay | 0.0001 |
Loss function | Binary Crossentropy |
A hidden layer activation function | ReLU |
Output layer Activation function | Softmax |
Dropout | 0.5 |
Momentum | 0.9 |