Figure 2.

Pojected heatmap of self-attention weights on the 4 main proteins of SARS-CoV-2. This figure shows the interpretability of our model, which can give us the binding site that has the most probability of binding to the ligand.