Skip to main content
. 2022 Nov 10;12:19200. doi: 10.1038/s41598-022-21848-3

Table 6.

This table summarises the pre-processing, data augmentation, and model compilation details for the standardised framework.

Parameters Values
Pre-processing (flow_from_dataframe) target_size N × N (see Table 5)
batch_size 16
shuffle True
seed 123
class_mode categorical
Data Augmentation RandomFlip horizontal_and_vertical
RandomRotation 0.2
RandomZoom 0.2
Model Compilation Optimiser Adam Optimiser
Learning rate 0.001
Loss function Weighted Categorical Cross Entropy
Metrics Accuracy
Epochs 100