Table 1. Performance of train and test data for predicting weekly norovirus warning, using the 6 machine learning algorithms.
Estimator | Training cases (n = 424) | Test cases (n = 106) | ||||
---|---|---|---|---|---|---|
Accuracy | F1-score | AUC | Accuracy | F1-score | AUC | |
SVM | 0.931 | 0.938 | 0.935 | 0.887 | 0.915 | 0.901 |
MLP | 0.92 | 0.928 | 0.92 | 0.896 | 0.923 | 0.908 |
RF | 0.965 | 0.968 | 0.966 | 0.915 | 0.94 | 0.902 |
GB | 0.965 | 0.969 | 0.965 | 0.915 | 0.94 | 0.902 |
LSTM | 0.972 | 0.974 | 0.974 | 0.925 | 0.947 | 0.899 |
GRU | 0.974 | 0.978 | 0.974 | 0.9 | 0.923 | 0.899 |