Table 4:
Model | AUC | F1 | Recall | Prec | P@3 |
---|---|---|---|---|---|
RNN | 0.6605 | 0.3659 | 0.3333 | 0.4054 | 0.5631 |
LSTM | 0.6792 | 0.4222 | 0.4222 | 0.4222 | 0.5931 |
GRU | 0.7140 | 0.4471 | 0.4222 | 0.4750 | 0.6323 |
TransformerRNN | 0.7598 | 0.5435 | 0.5556 | 0.5319 | 0.6671 |
TransformerLSTM | 0.7207 | 0.4396 | 0.4444 | 0.4348 | 0.5691 |
TransformerGRU | 0.7246 | 0.5376 | 0.5556 | 0.5208 | 0.6117 |
Transformer | 0.7005 | 0.4742 | 0.5111 | 0.4423 | 0.5974 |
TST | 0.7050 | 0.3810 | 0.3556 | 0.4103 | 0.5777 |
RNNAttention | 0.6416 | 0.3913 | 0.4000 | 0.3830 | 0.5111 |
LSTMAttention | 0.7015 | 0.4706 | 0.4444 | 0.5000 | 0.6477 |
GRUAttention | 0.6246 | 0.3656 | 0.3778 | 0.3542 | 0.4801 |
Random Forest | 0.7533 | 0.4878 | 0.4444 | 0.5405 | 0.6882 |