Skip to main content
. 2024 Feb 6;24:37. doi: 10.1186/s12911-024-02446-x

Table 5.

Confusion matrices of VGG, ResNet, and Efficient model

Model Name VGG-16 VGG-19 ResNet-18 ResNet-34 ResNet-50
Real Result Predicted Result
0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4
0 3638 0 0 0 0 3887 0 0 0 0 3825 0 0 0 0 3830 0 0 0 0 3864 0 0 0 0
1 363 0 0 0 0 376 0 0 0 0 371 0 0 0 0 391 0 0 0 0 365 0 0 0 0
2 763 0 0 0 0 793 0 0 0 0 806 0 0 0 0 804 0 0 0 0 806 0 0 0 0
3 131 0 0 0 0 131 0 0 0 0 152 0 0 0 0 138 0 0 0 0 116 0 0 0 0
4 113 0 0 0 0 113 0 0 0 0 114 0 0 0 0 105 0 0 0 0 117 0 0 0 0
Model Name ResNet-101 ResNet-152 EfficientNet-B0 EfficientNet-B1 EfficientNet-B2
Real Result Predicted Result
0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4
0 3848 0 0 0 0 3775 0 0 0 0 3787 0 105 2 5 3833 0 68 0 0 3761 0 120 0 0
1 383 0 0 0 0 360 0 0 0 0 321 0 20 0 0 357 0 17 0 0 346 0 29 0 0
2 783 0 0 0 0 814 0 0 0 0 532 0 228 25 1 501 0 273 0 2 487 0 307 0 0
3 140 0 0 0 0 128 0 0 0 0 27 0 67 39 2 30 0 89 0 3 24 0 86 0 0
4 103 0 0 0 0 90 0 0 0 0 28 0 47 17 15 34 0 41 0 18 17 0 91 0 0
Model Name EfficientNet-B3_30 EfficientNet-B3_60 EfficientNet-B4 EfficientNet-B5 EfficientNet-B6
Real Result Predicted Result
0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4 0 1 2 3 4
0 3868 3 20 0 0 3772 8 65 0 0 3851 0 38 0 2 3726 0 83 1 0 3809 0 59 0 2
1 326 23 34 0 0 244 31 128 0 0 348 0 3 0 0 363 0 11 0 0 335 0 13 0 0
2 247 34 501 0 0 114 9 624 5 0 605 0 194 5 8 587 0 234 21 0 515 0 269 1 10
3 10 0 69 38 0 3 0 77 46 13 42 0 71 46 2 36 0 70 33 0 26 0 87 5 18
4 16 0 29 15 37 9 0 14 6 98 44 0 58 6 16 19 0 65 21 0 33 0 66 1 21