Extended Data Fig. 5: Ablation study of BERT-DS.

Ablation study F1 scores on predicting non-hits, low-hits, and high-hits from the ‘HER2Affmat’ dataset for BERT-DS (with and without pretraining on OAS), BERT-DS ablation models, a multi-layered perceptron (MLP) neural network, an MLP trained on a soft binary classification target, logistic regression, linear support vector machine and random forest models. We report F1 scores on the a) train and b) test set splits. Numerical values for F1, precision and recall for each model and train/test set can be found in Supplementary Tables 5–14.