TABLE 3.
Summary of training and validation performance.
Model | Training time (minutes) | Steps | Training |
Validation |
||
Accuracy | Loss | Accuracy | Loss | |||
CNN | 46 | 97 | 0.8755 | 0.3102 | 0.7270 | 0.7094 |
VGG-16 | 75 | 57 | 0.9492 | 0.1511 | 0.8066 | 0.5263 |
ResNet-50 | 91 | 56 | 0.9164 | 0.2150 | 0.7686 | 0.5901 |