Table 3:
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.