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 | ||