Table 8. A comparison the accuracy, precision, recall, and F1 of Criminal charge prediction using BERT and XLNET models.
| Model | Accuracy | |||
|---|---|---|---|---|
| XLNET | 98.16% | 98.17% | 98.16% | 98.16% |
| BERT | 97.53% | 97.53% | 97.53% | 97.53% |
| Two chunking BERT | 98.5% | 98.5% | 98.5% | 98.5% |
| Three chunking BERT | 98.95% | 98.95% | 98.95% | 98.95% |