Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "3ef5ffbc6400370ff2e1546550a6bad3ac61b079"
Commit 91b43223 authored by Rich Evans's avatar Rich Evans Committed by Copybara-Service
Browse files

Fix PAE visualization for single chain predictions. 

PiperOrigin-RevId: 407773747
Change-Id: I8635dd77ecf366f4f5a362993b964adffc3ff186
parent c128d1aa
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
notebooks/AlphaFold.ipynb
View file @ 91b43223
...@@ -648,8 +648,9 @@ ...@@ -648,8 +648,9 @@
" total_num_res = best_unrelaxed_prot.residue_index.shape[-1]\n", " total_num_res = best_unrelaxed_prot.residue_index.shape[-1]\n",
" chain_ids = best_unrelaxed_prot.chain_index\n", " chain_ids = best_unrelaxed_prot.chain_index\n",
" for chain_boundary in np.nonzero(chain_ids[:-1] - chain_ids[1:]):\n", " for chain_boundary in np.nonzero(chain_ids[:-1] - chain_ids[1:]):\n",
" plt.plot([0, total_num_res], [chain_boundary, chain_boundary], color='red')\n", " if chain_boundary.size:\n",
" plt.plot([chain_boundary, chain_boundary], [0, total_num_res], color='red')\n", " plt.plot([0, total_num_res], [chain_boundary, chain_boundary], color='red')\n",
" plt.plot([chain_boundary, chain_boundary], [0, total_num_res], color='red')\n",
"\n", "\n",
" plt.title('Predicted Aligned Error')\n", " plt.title('Predicted Aligned Error')\n",
" plt.xlabel('Scored residue')\n", " plt.xlabel('Scored residue')\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment