/RoBERTa-attention-visualization

Visualization of attention layer from protein-ligand binding model. 2D visual pictures of ligands with highlighted regions that are predicted to bind to thrombin.

Primary LanguageJupyter Notebook

Watchers