Table 1 The prediction performance of all the models (mean ± std. (standard deviation))
From: Interpretable clinical prediction via attention-based neural network
Models | Accuracy | Precision | Recall | F1 | AUC |
|---|---|---|---|---|---|
MLP_attention | 0.667 ± 0.030 | 0.710 ± 0.020 | 0.795 ± 0.059 | 0.749 ± 0.029 | 0.691 ± 0.047 |
MLP | 0.651 ± 0.028 | 0.692 ± 0.022 | 0.799 ± 0.062 | 0.741 ± 0.030 | 0.683 ± 0.041 |
LR | 0.655 ± 0.027 | 0.700 ± 0.019 | 0.792 ± 0.043 | 0.743 ± 0.024 | 0.684 ± 0.039 |
SDAE | 0.623 ± 0.025 | 0.670 ± 0.018 | 0.782 ± 0.038 | 0.722 ± 0.022 | 0.658 ± 0.033 |