Skip to main content

Table 4 Performance comparison of all single models with cross entropy loss or focal loss functions using macro F1 score

From: Automated classification of clinical trial eligibility criteria text based on ensemble learning and metric learning

Model

Cross entropy loss

Focal loss

BERT

0.7902

0.7973

XLNet

0.7987

0.8030

RoBERTa

0.7959

0.7992

ERNIE

0.7868

0.7952

ELECTRA

0.7804

0.7862

Back to article page

BMC Medical Informatics and Decision Making

ISSN: 1472-6947

Contact us