PIP_B2B_Tools_using_files.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "PIP - B2B Tools using files",
      "provenance": [],
      "collapsed_sections": [],
      "private_outputs": true,
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "gpuClass": "standard"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Bio2Byte/public_notebooks/blob/main/Bio2ByteTools_v3_singleseq_pypi_with_files.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "# Bio2Byte - Single Sequence Predictions\n",
        "For most of below tools, we now have the b2bTools one-stop file upload location where you can easily try them: [online b2BTools](https://bio2byte.be/b2btools/)"
      ],
      "metadata": {
        "id": "EsZPZMEGYV2d"
      }
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "YaxPffRnrSVW",
        "cellView": "form"
      },
      "outputs": [],
      "source": [
        "#@title Install dependencies\n",
        "%%capture\n",
        "\n",
        "!pip install b2bTools==3.0.5"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Import dependencies\n",
        "import os\n",
        "import json\n",
        "from google.colab import files\n",
        "from b2bTools import SingleSeq\n",
        "from matplotlib import pyplot as plt\n",
        "from matplotlib.lines import Line2D\n",
        "\n",
        "if not os.path.exists(\"/content/results\"):\n",
        "    os.mkdir('/content/results')"
      ],
      "metadata": {
        "id": "cKwsuaFgr3CT",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Input protein sequences\n",
        "#@markdown Please select the predictor tools to be executed and run this cell\n",
        "#@markdown to open the file prompt in order to upload a valid FASTA file.\n",
        "\n",
        "#@markdown **🚨⚠️ IMPORTANT:** If beta aggregation is selected, the input file must contain up to 10 sequences.\n",
        "\n",
        "input_fasta = ''\n",
        "\n",
        "#@markdown ### DynaMine predictor tool\n",
        "#@markdown >Fast predictor of protein backbone dynamics using only sequence information as input. \n",
        "#@markdown >The version here also predicts side-chain dynamics and secondary structure predictors \n",
        "#@markdown >using the same principle.\n",
        "\n",
        "#@markdown **Prediction values included**: `backbone`, `sidechain`, `helix`, `ppII`, `coil`, and `sheet`\n",
        "\n",
        "dynamics = True #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown ### DisoMine predictor tool\n",
        "#@markdown >Predicts protein disorder with recurrent neural networks not directly \n",
        "#@markdown >from the amino acid sequence, but instead from more generic predictions of key \n",
        "#@markdown >biophysical properties, here protein dynamics, secondary structure \n",
        "#@markdown >and early folding.\n",
        "\n",
        "#@markdown **Prediction values included**: `disoMine`\n",
        "\n",
        "disorder = True #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown ### EFoldMine predictor tool\n",
        "#@markdown >Predicts from the primary amino acid sequence of a protein, \n",
        "#@markdown >which amino acids are likely involved in early folding events.\n",
        "\n",
        "#@markdown **Prediction values included**: `earlyFolding`\n",
        "\n",
        "early_folding = True #@param {type:\"boolean\"}\n",
        "\n",
        "#@markdown ### AgMata predictor tool\n",
        "#@markdown >Single-sequence based predictor of protein regions that are likely \n",
        "#@markdown >to cause beta-aggregation.\n",
        "\n",
        "#@markdown **Prediction values included**: `agmata`\n",
        "\n",
        "beta_aggretation = False #@param {type:\"boolean\"}\n",
        "\n",
        "tools = []\n",
        "if dynamics:\n",
        "    tools.append(\"dynamine\")\n",
        "if disorder:\n",
        "    tools.append(\"disomine\")\n",
        "if early_folding:\n",
        "    tools.append(\"efoldmine\")\n",
        "if beta_aggretation:\n",
        "    tools.append(\"agmata\")\n",
        "\n",
        "uploaded = files.upload()\n",
        "for fn in uploaded.keys():\n",
        "    print('User uploaded file \"{name}\" with length {length} bytes'.format(\n",
        "        name=fn, length=len(uploaded[fn])))\n",
        "    input_fasta = fn"
      ],
      "metadata": {
        "id": "a4AYPR-ByvL4",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Run predictors\n",
        "%%capture\n",
        "\n",
        "single_seq = SingleSeq(input_fasta)\n",
        "single_seq.predict(tools)\n",
        "\n",
        "all_predictions = single_seq.get_all_predictions()\n",
        "keys = list(all_predictions.keys())\n",
        "\n",
        "max_seq_len = max([len(pred['seq']) for pred in all_predictions.values()])"
      ],
      "metadata": {
        "cellView": "form",
        "id": "SoDxx2BLjsSc"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Plot Single Sequence predictions\n",
        "#@markdown The predictions reflect 'emerging' properties, so what the sequence \n",
        "#@markdown is capable of, not necessarily what it will do in a particular \n",
        "#@markdown context, for example when it adopts a specific fold.\n",
        "\n",
        "#@markdown #### DynaMine backbone dynamics\n",
        "#@markdown > Values above 0.8 indicate rigid conformations, \n",
        "#@markdown > values above 1.0 membrane spanning regions, \n",
        "#@markdown > values below 0.69 flexible regions. \n",
        "#@markdown > Values between 0.69-0.80 are 'context' dependent and capable of\n",
        "#@markdown > being either rigid or flexible.\n",
        "\n",
        "#@markdown #### DynaMine sidechain dynamics\n",
        "#@markdown > Higher values mean more likely rigid. \n",
        "#@markdown > These values are highly dependent on the amino acid type \n",
        "#@markdown > (i.e. a Trp will be rigid, an Asp flexible).\n",
        "\n",
        "#@markdown #### DynaMine conformational propensities (sheet, helix, coil, ppII (polyproline II))\n",
        "#@markdown > Higher values indicate higher propensities.\n",
        "\n",
        "#@markdown #### EFoldMine earlyFolding propensity\n",
        "#@markdown > Values above 0.169 indicate residues that are likely to start \n",
        "#@markdown > the protein folding process, based on only local interactions \n",
        "#@markdown > with other amino acids.\n",
        "\n",
        "#@markdown #### Disomine disorder\n",
        "#@markdown >  Values above 0.5 indicate that this is likely a disordered residue.\n",
        "\n",
        "\n",
        "fig, axs = plt.subplots(2, 4)\n",
        "\n",
        "ax1 = axs[0, 0]\n",
        "ax2 = axs[0, 1]\n",
        "ax3 = axs[0, 2]\n",
        "ax4 = axs[0, 3]\n",
        "ax5 = axs[1, 0]\n",
        "ax6 = axs[1, 1]\n",
        "ax7 = axs[1, 2]\n",
        "ax8 = axs[1, 3]\n",
        "\n",
        "fig.set_figwidth(30)\n",
        "fig.set_figheight(10)\n",
        "\n",
        "fig.suptitle(f'Single Sequence Predictions')\n",
        "\n",
        "for seq_key in keys:\n",
        "    predictions = all_predictions[seq_key]\n",
        "    x_position = range(len(predictions['seq']))\n",
        "\n",
        "    if dynamics:\n",
        "        backbone_pred = predictions['backbone']\n",
        "        coil_pred = predictions['coil']\n",
        "        sheet_pred = predictions['sheet']\n",
        "        ppII_pred = predictions['ppII']\n",
        "        helix_pred = predictions['helix']\n",
        "        sidechain_pred = predictions['sidechain']\n",
        "\n",
        "        ax1.plot(x_position, backbone_pred, label=seq_key)\n",
        "        ax2.plot(x_position, sidechain_pred, label=seq_key)\n",
        "        ax3.plot(x_position, coil_pred, label=seq_key)\n",
        "        ax4.plot(x_position, sheet_pred, label=seq_key)\n",
        "        ax5.plot(x_position, ppII_pred, label=seq_key)\n",
        "        ax6.plot(x_position, helix_pred, label=seq_key)\n",
        "                    \n",
        "    if disorder:\n",
        "        disomine_pred = predictions['disoMine']\n",
        "        ax7.plot(x_position, disomine_pred, label=seq_key)\n",
        "\n",
        "    \n",
        "    if early_folding:\n",
        "        earlyFolding_pred = predictions['earlyFolding'] \n",
        "        ax8.plot(x_position, earlyFolding_pred, label=seq_key)\n",
        "\n",
        "    if beta_aggretation:\n",
        "        agmata_pred = predictions['agmata']\n",
        "\n",
        "ax1.set_title('DynaMine backbone dynamics')\n",
        "ax1.set_ylim([-0.2, 1.2])\n",
        "ax1.set_xlabel('residue index')\n",
        "ax1.set_ylabel('prediction values')\n",
        "ax1.axhspan(1, 1.2, alpha=0.3, color='red')\n",
        "ax1.axhspan(0.8, 1, alpha=0.5, color='pink')\n",
        "ax1.axhspan(0.69, 0.8, alpha=0.5, color='orange')\n",
        "ax1.axhspan(-0.2, 0.69, alpha=0.5, color='yellow')\n",
        "ax1.grid(axis='y')\n",
        "ax1.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax2.set_title('DynaMine sidechain dynamics')\n",
        "ax2.set_ylim([-0.2, 1.2])\n",
        "ax2.set_xlabel('residue index')\n",
        "ax2.set_ylabel('prediction values')\n",
        "ax2.grid(axis='y')\n",
        "ax2.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax3.set_title('DynaMine conformational propensities: Coil')\n",
        "ax3.set_ylim([-0.2, 1.2])\n",
        "ax3.set_xlabel('residue index')\n",
        "ax3.set_ylabel('prediction values')\n",
        "ax3.grid(axis='y')\n",
        "ax3.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax4.set_title('DynaMine conformational propensities: Sheet')\n",
        "ax4.set_ylim([-0.2, 1.2])\n",
        "ax4.set_xlabel('residue index')\n",
        "ax4.set_ylabel('prediction values')\n",
        "ax4.grid(axis='y')\n",
        "ax4.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "ax5.set_title('DynaMine conformational propensities: ppII (polyproline II)')\n",
        "ax5.set_ylim([-0.2, 1.2])\n",
        "ax5.set_xlabel('residue index')\n",
        "ax5.set_ylabel('prediction values')\n",
        "ax5.grid(axis='y')\n",
        "ax5.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax6.set_title('DynaMine conformational propensities: Helix')\n",
        "ax6.set_ylim([-0.2, 1.2])\n",
        "ax6.set_xlabel('residue index')\n",
        "ax6.set_ylabel('prediction values')\n",
        "ax6.grid(axis='y')\n",
        "ax6.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax7.set_title('Early folding (EFoldMine)')\n",
        "ax7.set_ylim([-0.2, 1.2])\n",
        "ax7.set_xlabel('residue index')\n",
        "ax7.set_ylabel('prediction values')\n",
        "ax7.axhspan(-0.2, 0.169, alpha=0.5, color='yellow')\n",
        "ax7.axhspan(0.169, 1.2, alpha=0.5, color='orange')\n",
        "ax7.grid(axis='y')\n",
        "ax7.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "\n",
        "ax8.set_title('Disorder (disoMine)')\n",
        "ax8.set_ylim([-0.2, 1.2])\n",
        "ax8.set_xlabel('residue index')\n",
        "ax8.set_ylabel('prediction values')\n",
        "ax8.axhspan(0.5, 1.2, alpha=0.5, color='orange')\n",
        "ax8.axhspan(-0.2, 0.5, alpha=0.5, color='yellow')\n",
        "ax8.grid(axis='y')\n",
        "ax8.set_xlim([0, max_seq_len - 1])\n",
        "\n",
        "if len(keys) < 10: \n",
        "    plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.3), fancybox=True, shadow=True, ncol=len(keys))\n",
        "\n",
        "plt.subplots_adjust(hspace=0.4)\n",
        "plt.savefig('/content/results/single_sequence_predictons.png')\n",
        "plt.show()"
      ],
      "metadata": {
        "id": "KOH6FzXFrzw4",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Plot AgMata aggregation propensity\n",
        "#@markdown The predictions reflect 'emerging' properties, so what the sequence \n",
        "#@markdown is capable of, not necessarily what it will do in a particular \n",
        "#@markdown context, for example when it adopts a specific fold.\n",
        "\n",
        "#@markdown > These values are divided by a factor of 20 from the original. \n",
        "#@markdown > Peaks indicate residues likely to be involved in beta-sheet \n",
        "#@markdown > aggregation.\n",
        "\n",
        "if beta_aggretation:\n",
        "\n",
        "    fig, ax = plt.subplots(1, 1)\n",
        "    fig.set_figwidth(30)\n",
        "    fig.set_figheight(5)\n",
        "    fig.suptitle('Agmata aggregation propensity')\n",
        "\n",
        "    for seq_key in keys:\n",
        "        predictions = all_predictions[seq_key]\n",
        "        agmata_pred = predictions['agmata']\n",
        "        ax.plot(range(len(agmata_pred)), agmata_pred, label=seq_key)\n",
        "\n",
        "    ax.set_xlim([0, max_seq_len - 1])\n",
        "    ax.set_xlabel('residue index')\n",
        "    ax.set_ylabel('prediction values')\n",
        "    ax.grid(axis='y')\n",
        "\n",
        "    plt.legend(loc='upper center', bbox_to_anchor=(0.5, -0.2), fancybox=True, shadow=True, ncol=len(keys))\n",
        "    plt.savefig('/content/results/agmata_predictons.png')\n",
        "    plt.show()\n",
        "\n",
        "else:\n",
        "    print(\"Beta aggregation tool was not selected\")"
      ],
      "metadata": {
        "id": "m_K9OQKnSXg3",
        "cellView": "form"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Plot Single Sequence predictions (3D projection)\n",
        "#@markdown The predictions reflect 'emerging' properties, so what the sequence \n",
        "#@markdown is capable of, not necessarily what it will do in a particular \n",
        "#@markdown context, for example when it adopts a specific fold.\n",
        "\n",
        "#@markdown #### DynaMine backbone dynamics\n",
        "#@markdown > Values above 0.8 indicate rigid conformations, \n",
        "#@markdown > values above 1.0 membrane spanning regions, \n",
        "#@markdown > values below 0.69 flexible regions. \n",
        "#@markdown > Values between 0.69-0.80 are 'context' dependent and capable of\n",
        "#@markdown > being either rigid or flexible.\n",
        "\n",
        "#@markdown #### DynaMine sidechain dynamics\n",
        "#@markdown > Higher values mean more likely rigid. \n",
        "#@markdown > These values are highly dependent on the amino acid type \n",
        "#@markdown > (i.e. a Trp will be rigid, an Asp flexible).\n",
        "\n",
        "#@markdown #### DynaMine conformational propensities (sheet, helix, coil, ppII (polyproline II))\n",
        "#@markdown > Higher values indicate higher propensities.\n",
        "\n",
        "#@markdown #### EFoldMine earlyFolding propensity\n",
        "#@markdown > Values above 0.169 indicate residues that are likely to start \n",
        "#@markdown > the protein folding process, based on only local interactions \n",
        "#@markdown > with other amino acids.\n",
        "\n",
        "#@markdown #### Disomine disorder\n",
        "#@markdown >  Values above 0.5 indicate that this is likely a disordered residue.\n",
        "\n",
        "fig = plt.figure(figsize=plt.figaspect(0.5))\n",
        "# axs = fig.add_subplot(2, 4, 1, projection='3d')\n",
        "\n",
        "ax1 = fig.add_subplot(4, 2, 1, projection='3d')\n",
        "ax2 = fig.add_subplot(4, 2, 2, projection='3d')\n",
        "ax3 = fig.add_subplot(4, 2, 3, projection='3d')\n",
        "ax4 = fig.add_subplot(4, 2, 4, projection='3d')\n",
        "ax5 = fig.add_subplot(4, 2, 5, projection='3d')\n",
        "ax6 = fig.add_subplot(4, 2, 6, projection='3d')\n",
        "ax7 = fig.add_subplot(4, 2, 7, projection='3d')\n",
        "ax8 = fig.add_subplot(4, 2, 8, projection='3d')\n",
        "\n",
        "fig.set_figwidth(30)\n",
        "fig.set_figheight(40)\n",
        "\n",
        "fig.suptitle(f'Single Sequence Predictions (3D Projection)')\n",
        "\n",
        "for index, seq_key in enumerate(keys):\n",
        "    predictions = all_predictions[seq_key]\n",
        "    x_position = range(len(predictions['seq']))\n",
        "    z_position = [index] * len(predictions['seq'])\n",
        "\n",
        "    if dynamics:\n",
        "        backbone_pred = predictions['backbone']\n",
        "        coil_pred = predictions['coil']\n",
        "        sheet_pred = predictions['sheet']\n",
        "        ppII_pred = predictions['ppII']\n",
        "        helix_pred = predictions['helix']\n",
        "        sidechain_pred = predictions['sidechain']\n",
        "\n",
        "        ax1.plot(x_position, z_position, backbone_pred, label=seq_key)\n",
        "        ax2.plot(x_position, z_position, sidechain_pred, label=seq_key)\n",
        "        ax3.plot(x_position, z_position, coil_pred, label=seq_key)\n",
        "        ax4.plot(x_position, z_position, sheet_pred, label=seq_key)\n",
        "        ax5.plot(x_position, z_position, ppII_pred, label=seq_key)\n",
        "        ax6.plot(x_position, z_position, helix_pred, label=seq_key)\n",
        "    \n",
        "    if disorder:\n",
        "        disomine_pred = predictions['disoMine']\n",
        "        ax7.plot(x_position, z_position, disomine_pred, label=seq_key)\n",
        "\n",
        "    if early_folding:\n",
        "        earlyFolding_pred = predictions['earlyFolding'] \n",
        "        ax8.plot(x_position, z_position, earlyFolding_pred, label=seq_key)\n",
        "\n",
        "ax1.set_title('DynaMine backbone dynamics')\n",
        "ax1.set_zlim([-0.1, 1.1])\n",
        "ax1.set_xlabel('residue index')\n",
        "ax1.set_zlabel('prediction values')\n",
        "ax1.set_yticks(range(len(keys)))\n",
        "ax1.set_yticklabels(keys)\n",
        "ax1.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax2.set_title('DynaMine sidechain dynamics')\n",
        "ax2.set_zlim([-0.1, 1.1])\n",
        "ax2.set_xlabel('residue index')\n",
        "ax2.set_zlabel('prediction values')\n",
        "ax2.set_yticks(range(len(keys)))\n",
        "ax2.set_yticklabels(keys)\n",
        "ax2.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax3.set_title('DynaMine conformational propensities: Coil')\n",
        "ax3.set_zlim([-0.1, 1.1])\n",
        "ax3.set_xlabel('residue index')\n",
        "ax3.set_zlabel('prediction values')\n",
        "ax3.set_yticks(range(len(keys)))\n",
        "ax3.set_yticklabels(keys)\n",
        "ax3.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax4.set_title('DynaMine conformational propensities: Sheet')\n",
        "ax4.set_zlim([-0.1, 1.1])\n",
        "ax4.set_xlabel('residue index')\n",
        "ax4.set_zlabel('prediction values')\n",
        "ax4.set_yticks(range(len(keys)))\n",
        "ax4.set_yticklabels(keys)\n",
        "ax4.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax5.set_title('DynaMine conformational propensities: ppII (polyproline II)')\n",
        "ax5.set_zlim([-0.1, 1.1])\n",
        "ax5.set_xlabel('residue index')\n",
        "ax5.set_zlabel('prediction values')\n",
        "ax5.set_yticks(range(len(keys)))\n",
        "ax5.set_yticklabels(keys)\n",
        "ax5.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax6.set_title('DynaMine conformational propensities: Helix')\n",
        "ax6.set_zlim([-0.1, 1.1])\n",
        "ax6.set_xlabel('residue index')\n",
        "ax6.set_ylabel('sequence')\n",
        "ax6.set_zlabel('prediction values')\n",
        "ax6.set_yticks(range(len(keys)))\n",
        "ax6.set_yticklabels(keys)\n",
        "ax6.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax7.set_title('Disorder (disoMine)')\n",
        "ax7.set_zlim([-0.1, 1.1])\n",
        "ax7.set_xlabel('residue index')\n",
        "ax7.set_ylabel('sequence')\n",
        "ax7.set_zlabel('prediction values')\n",
        "ax7.set_yticks(range(len(keys)))\n",
        "ax7.set_yticklabels(keys)\n",
        "ax7.view_init(60, -45)\n",
        "\n",
        "\n",
        "ax8.set_title('Early folding (EFoldMine)')\n",
        "ax8.set_zlim([-0.1, 1.1])\n",
        "ax8.set_xlabel('residue index')\n",
        "ax8.set_ylabel('sequence')\n",
        "ax8.set_zlabel('prediction values')\n",
        "ax8.set_yticks(range(len(keys)))\n",
        "ax8.set_yticklabels(keys)\n",
        "ax8.view_init(60, -45)\n",
        "\n",
        "\n",
        "plt.subplots_adjust(hspace=0.2)\n",
        "plt.savefig('/content/results/single_sequence_predictons_3d.png')\n",
        "plt.show()"
      ],
      "metadata": {
        "cellView": "form",
        "id": "JKVklmzvY9Y3"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Plot AgMata aggregation propensity (3D projection)\n",
        "#@markdown The predictions reflect 'emerging' properties, so what the sequence \n",
        "#@markdown is capable of, not necessarily what it will do in a particular \n",
        "#@markdown context, for example when it adopts a specific fold.\n",
        "\n",
        "#@markdown > These values are divided by a factor of 20 from the original. \n",
        "#@markdown > Peaks indicate residues likely to be involved in beta-sheet \n",
        "#@markdown > aggregation.\n",
        "\n",
        "if beta_aggretation:\n",
        "    fig = plt.figure(figsize=plt.figaspect(0.5))\n",
        "    ax1 = fig.add_subplot(1, 1, 1, projection='3d')\n",
        "\n",
        "    fig.set_figwidth(15)\n",
        "    fig.set_figheight(10)\n",
        "\n",
        "    fig.suptitle(f'Single Sequence Predictions (3D Projection)')\n",
        "\n",
        "    for index, seq_key in enumerate(keys):\n",
        "        predictions = all_predictions[seq_key]\n",
        "        x_position = range(len(predictions['seq']))\n",
        "        z_position = [index] * len(predictions['seq'])\n",
        "\n",
        "        agmata_pred = predictions['agmata']\n",
        "\n",
        "        ax1.plot(x_position, z_position, agmata_pred, label=seq_key)\n",
        "\n",
        "    ax1.set_title('Agmata aggregation propensity')\n",
        "    ax1.set_xlabel('residue index')\n",
        "    ax1.set_zlabel('prediction values')\n",
        "    ax1.set_yticks(range(len(keys)))\n",
        "    ax1.set_yticklabels(keys)\n",
        "    ax1.view_init(60, -45)\n",
        "\n",
        "    plt.savefig('/content/results/agmata_predictons_3d.png')\n",
        "    plt.show()\n",
        "\n",
        "else:\n",
        "    print(\"Beta aggregation was not selected\")"
      ],
      "metadata": {
        "cellView": "form",
        "id": "oRxE84upgFdn"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@title Download the results\n",
        "#@markdown After running this cell you will be able to download the plots\n",
        "#@markdown plain results in JSON format to your computer.\n",
        "\n",
        "json.dump(all_predictions, open('/content/results/b2btools_results.json', 'w'), indent=4, sort_keys=True)\n",
        "\n",
        "!zip -r /content/b2b-tools-results.zip /content/results\n",
        "\n",
        "files.download(\"/content/b2b-tools-results.zip\")"
      ],
      "metadata": {
        "cellView": "form",
        "id": "NyBFvJ4vMqBJ"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}