Skip to main content
. Author manuscript; available in PMC: 2018 Aug 23.
Published in final edited form as: Proc Conf. 2018 Jun;2018:2081–2091. doi: 10.18653/v1/N18-1189

Table 3:

Results for the MIMIC III dataset.

F1 AUC (PR) AUC (ROC) P@k R@k
P R Micro Macro Micro Macro Micro Macro 8 40 8 40
Logistic (Vani et al., 2017)* 0.711 0.242 0.361 0.026 0.419 0.147 0.961 0.751 0.554 0.211 0.414 0.686
CNN (Baumel et al., 2018)* 0.726 0.246 0.367 0.021 0.376 0.095 0.942 0.697 0.534 0.196 0.395 0.636
Matching Network* 0.248 0.237 0.242 0.008 0.183 0.028 0.823 0.554 0.310 0.128 0.231 0.431
Match-CNN (Ours) 0.466 0.447 0.456 0.041 0.421 0.119 0.963 0.726 0.557 0.206 0.413 0.670
Match-CNN Ens. (Ours) 0.488 0.449 0.468 0.043 0.441 0.129 0.965 0.760 0.570 0.211 0.422 0.683

Models marked with * represent our custom implementations.