Table 5.
Normalized confusion matrix of networks trained with the balanced dataset and the imbalanced dataset.
Predicted Class | ||||||
---|---|---|---|---|---|---|
Actual class | Balanced dataset | Imbalanced dataset | ||||
Normal | Drowsy | Normal | Drowsy | |||
Normal | 0.940 | 0.060 | 0.998 | 0.002 | ||
Drowsy | 0.072 | 0.928 | 0.116 | 0.884 |