Skip to main content
. 2020 Nov 17;51(5):2850–2863. doi: 10.1007/s10489-020-02055-x

Table 4.

Parameters setting details in our method

Experimental parameters Setting
Image size 150 × 150
Batch size 12
Epoch 60
Optimizer Adam
Learning rate (LR) 0.0001
Loss Categorical cross entropy
Validation spit 0.2
Shear_range 0.25
Zoom_range 0.1
Channel_shift_range 20
Horizontal_flip True
Vertical_flip True
Rotation_range 50
Rescale 1/255
Height_shift_range 0.2
Width_shift_range 0.2