Table 1.
Results of diagnosis prediction performance. Numbers with bold font indicate the best performance.
| Models | Precision | Recall | F1 score |
|---|---|---|---|
| BERT-base-Chinese-256-single | 0.6001 | 0.6148 | 0.6077 |
| BERT-base-Chinese-256-concatenate | 0.6081 | 0.67 | 0.6376 |
| BERT-base-Chinese-256-average | 0.6215 | 0.6626 | 0.6414 |
| BERT-base-Chinese-512-concatenate | 0.6739 | 0.7056 | 0.6893 |
| BERT-base-Chinese-512-average | 0.6562 | 0.7062 | 0.6803 |
| MacBERT-base-Chinese-256-single | 0.6017 | 0.6285 | 0.6148 |
| MacBERT-base-Chinese-256-concatenate | 0.6119 | 0.6678 | 0.6387 |
| MacBERT-base-Chinese-256-average | 0.6017 | 0.679 | 0.6381 |
| MacBERT-base-Chinese-512-concatenate | 0.667 | 0.7157 | 0.6905 |
| MacBERT-base-Chinese-512-average | 0.6562 | 0.7267 | 0.6897 |