Skip to main content

Table 9 Comparison of methods without BERT. We use MIMIC-III in this experiment

From: al-BERT: a semi-supervised denoising technique for disease prediction

Model

AUC-ROC

Accuracy

Precision

Recall

F1

LSTM

0.9247

0.8918

0.8754

0.9701

0.9204

RETAIN

0.9505

0.8826

0.9371

0.8846

0.8812

al (ours)

0.9621

0.9280

0.9448

0.9769

0.9304

  1. The bold case represents the best result under a single metric
Back to article page

BMC Medical Informatics and Decision Making

ISSN: 1472-6947

Contact us