pyg2neo/Movie_recommendations.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "Movie-recommendations.ipynb",
      "provenance": [],
      "authorship_tag": "ABX9TyMOoqQubsR6SVgeKrpLENAb",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU",
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "6e42a7a80bcc4f18853085bb4e9f1602": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_4fd077d31fb944d39674308ead3f79ec",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_484630f8782d470e81cc17f24a285dd6",
              "IPY_MODEL_67c2c79a590f4de884f7a61e93d35df6",
              "IPY_MODEL_b85481e270a5460a86268f63b58fc028"
            ]
          }
        },
        "4fd077d31fb944d39674308ead3f79ec": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "484630f8782d470e81cc17f24a285dd6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_52efc84bf9b34b468177dcd86d426d01",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_49f974d6b1ec42c99eaf5ac1c0af6582"
          }
        },
        "67c2c79a590f4de884f7a61e93d35df6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_93bffa61924546e59a3b427e6ef61fba",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 1175,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 1175,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_d6b5c4130eef47549badcb8883b9faf2"
          }
        },
        "b85481e270a5460a86268f63b58fc028": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_01d56aee12c84b5bac09486e4a5c291a",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 1.18k/1.18k [00:00&lt;00:00, 25.2kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_1d976d0faa87429587d8e629c61bc790"
          }
        },
        "52efc84bf9b34b468177dcd86d426d01": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "49f974d6b1ec42c99eaf5ac1c0af6582": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "93bffa61924546e59a3b427e6ef61fba": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "d6b5c4130eef47549badcb8883b9faf2": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "01d56aee12c84b5bac09486e4a5c291a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "1d976d0faa87429587d8e629c61bc790": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "7ee93e9261f1426b8e22eda01fa4afaa": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_a2a2827235624287812522be4550e4e3",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_73457a3e28ff4003bcd86897829a3f52",
              "IPY_MODEL_267895bda38347edb26bd6c76d87a67e",
              "IPY_MODEL_ed171a21a44b4cb2bc9323a5cea1dcc7"
            ]
          }
        },
        "a2a2827235624287812522be4550e4e3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "73457a3e28ff4003bcd86897829a3f52": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_2e8dcefe6e0b4c3aa9354aad5e25c9ea",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_667e3bbff63a407e9a0ca8814d66a11f"
          }
        },
        "267895bda38347edb26bd6c76d87a67e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_4dda9c1ef5f643ada30c1ea5574836a1",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 10177,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 10177,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_4798de8b355c4251928bde9109af76a1"
          }
        },
        "ed171a21a44b4cb2bc9323a5cea1dcc7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_e636b4a360a44df7902b17032a50b014",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 10.2k/10.2k [00:00&lt;00:00, 232kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_95a6df8d509a443a9560f973742e8d7c"
          }
        },
        "2e8dcefe6e0b4c3aa9354aad5e25c9ea": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "667e3bbff63a407e9a0ca8814d66a11f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "4dda9c1ef5f643ada30c1ea5574836a1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "4798de8b355c4251928bde9109af76a1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "e636b4a360a44df7902b17032a50b014": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "95a6df8d509a443a9560f973742e8d7c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "7fdfc2ae6c254b4ea3e1bfa7bd6bf1d9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_d2f720de95954d2b8742ea91989c39b2",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_1189c090eb224630b1dc92b94fe8e71d",
              "IPY_MODEL_26070f66567445419eb360b47270cf15",
              "IPY_MODEL_1dfa4b09793c42929aab78eea6dbee10"
            ]
          }
        },
        "d2f720de95954d2b8742ea91989c39b2": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "1189c090eb224630b1dc92b94fe8e71d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_f5ac3ef85af0407b9abd5fabc6681824",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_976e4173c34249cf83e1fcea2fd1c568"
          }
        },
        "26070f66567445419eb360b47270cf15": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_49e268eb8c5444fca4db6450ce302852",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 612,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 612,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_347b4df225b44f1f98a181d10620b5fd"
          }
        },
        "1dfa4b09793c42929aab78eea6dbee10": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_8c0ede7d55fd460798c6c93048ff4412",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 612/612 [00:00&lt;00:00, 16.4kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_6757017d028049c2ad4fb9ccebaa9110"
          }
        },
        "f5ac3ef85af0407b9abd5fabc6681824": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "976e4173c34249cf83e1fcea2fd1c568": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "49e268eb8c5444fca4db6450ce302852": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "347b4df225b44f1f98a181d10620b5fd": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "8c0ede7d55fd460798c6c93048ff4412": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "6757017d028049c2ad4fb9ccebaa9110": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "58acdcf5f0d540f2a21beec2a65e83de": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_5e6c6f3d122b42a88bc2d4470c1cf31e",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_712c0cb08ed84590866012ac64b3bb8d",
              "IPY_MODEL_1a52022b38844932a157faab32792519",
              "IPY_MODEL_7793533cdd314f36b96ae534f8a2d503"
            ]
          }
        },
        "5e6c6f3d122b42a88bc2d4470c1cf31e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "712c0cb08ed84590866012ac64b3bb8d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_f13b16fec0f34959b69e1da99a9cad63",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_a304eac3795142e79ad03965def10eb9"
          }
        },
        "1a52022b38844932a157faab32792519": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_284728ec737f41d58260c8e8c00af9e3",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 116,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 116,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_28721fe261aa4db3b9f9f22daf376f66"
          }
        },
        "7793533cdd314f36b96ae534f8a2d503": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_b8799b5a37c64a5b9c797ad3b1461a37",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 116/116 [00:00&lt;00:00, 3.21kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_17b89e0ca1274e70932fd7240107191c"
          }
        },
        "f13b16fec0f34959b69e1da99a9cad63": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "a304eac3795142e79ad03965def10eb9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "284728ec737f41d58260c8e8c00af9e3": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "28721fe261aa4db3b9f9f22daf376f66": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b8799b5a37c64a5b9c797ad3b1461a37": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "17b89e0ca1274e70932fd7240107191c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "93cc8fee07c44ca0afc406fe8d4613c9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_6dc05c71373b4f76992ce4a6c906df6e",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_1ddd930ef90c489f81ae2a26dc10acf0",
              "IPY_MODEL_d9589531cfd941d2b36a60d85f6b36ac",
              "IPY_MODEL_25998ca65b4d4861aed99f2240c535de"
            ]
          }
        },
        "6dc05c71373b4f76992ce4a6c906df6e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "1ddd930ef90c489f81ae2a26dc10acf0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_88b442811b99493588f553a8772fa5c6",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_9dd6f123fba048e69291fa45f7ad531f"
          }
        },
        "d9589531cfd941d2b36a60d85f6b36ac": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_63d58abaf4e043c0946d74925eaf7b16",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 39265,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 39265,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_d88898dcc0674603a85fa03e37a99742"
          }
        },
        "25998ca65b4d4861aed99f2240c535de": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_9333c6f9617b4b43a74e294a6712ee38",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 39.3k/39.3k [00:00&lt;00:00, 863kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_6bc0f40661fc4694a1a8dd4027fe335d"
          }
        },
        "88b442811b99493588f553a8772fa5c6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "9dd6f123fba048e69291fa45f7ad531f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "63d58abaf4e043c0946d74925eaf7b16": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "d88898dcc0674603a85fa03e37a99742": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "9333c6f9617b4b43a74e294a6712ee38": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "6bc0f40661fc4694a1a8dd4027fe335d": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "fadde8c04f5d4a5ebfc2a364feaac196": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_078c67724a3544f68386dea213a238ad",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_a75d669ce8ed4c5f8c61d14865d93e14",
              "IPY_MODEL_a4dfcbaa8167439e90bbae46692d8dd1",
              "IPY_MODEL_792a35155f7d400196d1f11db5083c5a"
            ]
          }
        },
        "078c67724a3544f68386dea213a238ad": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "a75d669ce8ed4c5f8c61d14865d93e14": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_b7d818c04a494e918d902f3564abdb5b",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_6e7d48547e01479f989c2709fef2fdb3"
          }
        },
        "a4dfcbaa8167439e90bbae46692d8dd1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_a2835377c4434f389e509430607b7f1b",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 349,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 349,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_0fd715eb636f4be0998ce846bc0f786b"
          }
        },
        "792a35155f7d400196d1f11db5083c5a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_020553f4826146d0a8f5a5fa335bb293",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 349/349 [00:00&lt;00:00, 6.75kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_26e2dd17fee3460d9ee5babe07335ebe"
          }
        },
        "b7d818c04a494e918d902f3564abdb5b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "6e7d48547e01479f989c2709fef2fdb3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "a2835377c4434f389e509430607b7f1b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "0fd715eb636f4be0998ce846bc0f786b": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "020553f4826146d0a8f5a5fa335bb293": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "26e2dd17fee3460d9ee5babe07335ebe": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "fa7985ea84864b9bb947efbeff09ff1f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_1a3a2ac407bf44b48894e020449a25e5",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_d59ca158a4834c1187cf0852f7858600",
              "IPY_MODEL_9ab0b65c3ab047c6b2e2ddbdc7f01370",
              "IPY_MODEL_5cd3e5aa48c74173beabc1ad5c75a9a0"
            ]
          }
        },
        "1a3a2ac407bf44b48894e020449a25e5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "d59ca158a4834c1187cf0852f7858600": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_10b48d350dfc467f9e956af4f75dff44",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_44e0113c67f0486cb574fce896bfe327"
          }
        },
        "9ab0b65c3ab047c6b2e2ddbdc7f01370": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_0d40462afae047c9bc3fa88044e2e273",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 90888945,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 90888945,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_b46cbab2e77d489c8bbc73b5d7b99b82"
          }
        },
        "5cd3e5aa48c74173beabc1ad5c75a9a0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_a060ee2943014ce2b82b57fa7f6e9168",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 90.9M/90.9M [00:02&lt;00:00, 35.0MB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_0eed01ccde454953bf708cc344c84dbd"
          }
        },
        "10b48d350dfc467f9e956af4f75dff44": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "44e0113c67f0486cb574fce896bfe327": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "0d40462afae047c9bc3fa88044e2e273": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "b46cbab2e77d489c8bbc73b5d7b99b82": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "a060ee2943014ce2b82b57fa7f6e9168": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "0eed01ccde454953bf708cc344c84dbd": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "8589805155334893a469738b2e66972f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_80d20d44b7164ca19cbe5acab089192c",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_b6776d7d130647dba5f89f4c9b58ebc4",
              "IPY_MODEL_42a8702228e74f678e71212e7f347a8a",
              "IPY_MODEL_09cacc55d1914adcb229f3850b8b0f3a"
            ]
          }
        },
        "80d20d44b7164ca19cbe5acab089192c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b6776d7d130647dba5f89f4c9b58ebc4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_415d9472866c400f989631b8ef73d38e",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_f91329ba17e84958b05e519e71a9260a"
          }
        },
        "42a8702228e74f678e71212e7f347a8a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_cfd8cca64ca2444b960a40b16d75280f",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 53,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 53,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_4d0f09a64c1d4fc7a253ce2b9cce7b79"
          }
        },
        "09cacc55d1914adcb229f3850b8b0f3a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_36626cc4d6ec4605b891b5ebd939ac6a",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 53.0/53.0 [00:00&lt;00:00, 1.17kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_39cb1fec3984463a8996e7deef526327"
          }
        },
        "415d9472866c400f989631b8ef73d38e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "f91329ba17e84958b05e519e71a9260a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "cfd8cca64ca2444b960a40b16d75280f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "4d0f09a64c1d4fc7a253ce2b9cce7b79": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "36626cc4d6ec4605b891b5ebd939ac6a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "39cb1fec3984463a8996e7deef526327": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "d66b2963841e4d6e8ca70d59c915f758": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_2ef788647f764e67a94e9d62ad19ecd8",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_3b635a517fdf40349a8e371a7c988302",
              "IPY_MODEL_0b473696b2b84a6ca493ec60e378bb3d",
              "IPY_MODEL_6202f1e33ebe431c8b7da6fcbb8b39ae"
            ]
          }
        },
        "2ef788647f764e67a94e9d62ad19ecd8": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "3b635a517fdf40349a8e371a7c988302": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_3beb1e4523394f52a9af817d85bcbd9d",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_b337960b719a48c481df5f862cd906be"
          }
        },
        "0b473696b2b84a6ca493ec60e378bb3d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_e345a268eac549e082022e61f1785960",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 112,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 112,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_a5c90f34a9dd4804bf45436529af50a9"
          }
        },
        "6202f1e33ebe431c8b7da6fcbb8b39ae": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_9c4a20d40fa6432ca9df7a597f845cb6",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 112/112 [00:00&lt;00:00, 2.09kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_0df73cc5352b4ca69fc05c1dfa09294a"
          }
        },
        "3beb1e4523394f52a9af817d85bcbd9d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "b337960b719a48c481df5f862cd906be": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "e345a268eac549e082022e61f1785960": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "a5c90f34a9dd4804bf45436529af50a9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "9c4a20d40fa6432ca9df7a597f845cb6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "0df73cc5352b4ca69fc05c1dfa09294a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "770b9decb42045ea8da680823b0135dd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_2cdbdd7db2654f96ba70250fab87e038",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_c0dd89c0349c4df1b18d1f7fd3da665d",
              "IPY_MODEL_e4b36b50ad974b27977def287e91fc62",
              "IPY_MODEL_1138df11c56d495cb5a7c0e53470120d"
            ]
          }
        },
        "2cdbdd7db2654f96ba70250fab87e038": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "c0dd89c0349c4df1b18d1f7fd3da665d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_e7cda2225d7441068409fd0511e16c57",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_03d1bbb774904159acce8633302f235c"
          }
        },
        "e4b36b50ad974b27977def287e91fc62": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_b4f13a61a66b42e5aea7054af9b46dc5",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 466247,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 466247,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_45e1f0145f9a41738b2ca84b4ef07799"
          }
        },
        "1138df11c56d495cb5a7c0e53470120d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_17e8f46b67cc4ac683d83c99b82ec8be",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 466k/466k [00:00&lt;00:00, 1.02MB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_b6b725950b364bf19f47999cdd6afbaf"
          }
        },
        "e7cda2225d7441068409fd0511e16c57": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "03d1bbb774904159acce8633302f235c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b4f13a61a66b42e5aea7054af9b46dc5": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "45e1f0145f9a41738b2ca84b4ef07799": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "17e8f46b67cc4ac683d83c99b82ec8be": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "b6b725950b364bf19f47999cdd6afbaf": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "a8fafcd1a99a45efadf4dde0ba1d41df": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_445cfe326830407d89a03872e12060de",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_b5c7c1390eae4e9fa497641962207a58",
              "IPY_MODEL_59a463f8e1854cd2992feb7549d101d4",
              "IPY_MODEL_de6ae923e60b447ebaeeec7d5a6b8303"
            ]
          }
        },
        "445cfe326830407d89a03872e12060de": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b5c7c1390eae4e9fa497641962207a58": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_348d87b38a7d433da0199c9493b0a88f",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_d79d549ccd6f45bfbe30223c02d9f556"
          }
        },
        "59a463f8e1854cd2992feb7549d101d4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_903c403ed2344fd19e053d1669d4cd44",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 350,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 350,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_f8b04579ac1c4b7aa5b3b773a500d876"
          }
        },
        "de6ae923e60b447ebaeeec7d5a6b8303": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_b239033e712f4b928657d9b4c55a80a1",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 350/350 [00:00&lt;00:00, 7.86kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_dac17cd757b94066b12afd3879da7835"
          }
        },
        "348d87b38a7d433da0199c9493b0a88f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "d79d549ccd6f45bfbe30223c02d9f556": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "903c403ed2344fd19e053d1669d4cd44": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "f8b04579ac1c4b7aa5b3b773a500d876": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b239033e712f4b928657d9b4c55a80a1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "dac17cd757b94066b12afd3879da7835": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b1131fa026bd4af3945105d5ed2475b8": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_d74c2af975df483abb4d20d9a34c2933",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_89fa67324b344e91b72a06ecf5c3348f",
              "IPY_MODEL_2fc7361eef4548a5927a0f8de71c20fc",
              "IPY_MODEL_9906cf9e0fa74d949caf0299959801c0"
            ]
          }
        },
        "d74c2af975df483abb4d20d9a34c2933": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "89fa67324b344e91b72a06ecf5c3348f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_a3abddf18a4c48bdbc2ef95f05844ffa",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_efca4d3e7ff54f7b98e4801f215b7213"
          }
        },
        "2fc7361eef4548a5927a0f8de71c20fc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_c6223d4a6a3f4f0cbba298e8444139c8",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 13156,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 13156,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_acc419a409b342d6b9d2848590241403"
          }
        },
        "9906cf9e0fa74d949caf0299959801c0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_6ab7c0d2725048fd8907ba943a4f5af6",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 13.2k/13.2k [00:00&lt;00:00, 338kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_a33ee87994984b188a9d461728500a28"
          }
        },
        "a3abddf18a4c48bdbc2ef95f05844ffa": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "efca4d3e7ff54f7b98e4801f215b7213": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "c6223d4a6a3f4f0cbba298e8444139c8": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "acc419a409b342d6b9d2848590241403": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "6ab7c0d2725048fd8907ba943a4f5af6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "a33ee87994984b188a9d461728500a28": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "a4f54f46d4ce44479d4d96b31bea574c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_43537a3e006b4edd8ec2b391ab6f7c8d",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_15577a3d8a224a00b067d87a4ec1dad2",
              "IPY_MODEL_33100b8bb41f4768933e9747e1e746ed",
              "IPY_MODEL_56cbf7ffc64a4ef6aa4cd840d475b2e7"
            ]
          }
        },
        "43537a3e006b4edd8ec2b391ab6f7c8d": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "15577a3d8a224a00b067d87a4ec1dad2": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_4f6c4721297b45a89d08fd0c300f2290",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_ce95d840b5444dce92bef29290433df3"
          }
        },
        "33100b8bb41f4768933e9747e1e746ed": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_8b6967ba48754103a2003ff7d2987025",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 231508,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 231508,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_0513677b926640ffa79ec52d4f034499"
          }
        },
        "56cbf7ffc64a4ef6aa4cd840d475b2e7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_ed48348145b547cbb9531fe64cae781b",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 232k/232k [00:00&lt;00:00, 967kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_e502bd16650c47aa936214a87f0359ec"
          }
        },
        "4f6c4721297b45a89d08fd0c300f2290": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "ce95d840b5444dce92bef29290433df3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "8b6967ba48754103a2003ff7d2987025": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "0513677b926640ffa79ec52d4f034499": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "ed48348145b547cbb9531fe64cae781b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "e502bd16650c47aa936214a87f0359ec": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "e3cac84d8b8a4a4c81546dd4947b3d5b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_c68141e5464e41039b2a884f78aeed6b",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_420cb85096064891a3feb56796dfdcc1",
              "IPY_MODEL_459d4965a37745199980a2f7741a16a3",
              "IPY_MODEL_77cbfbea1be94d89a788aaad076040b1"
            ]
          }
        },
        "c68141e5464e41039b2a884f78aeed6b": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "420cb85096064891a3feb56796dfdcc1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_7145a8e992d94e4399b713951a11f99f",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Downloading: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_b5fc5407f83e422d87903143950330d1"
          }
        },
        "459d4965a37745199980a2f7741a16a3": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_8a252a9ed8ec4a3c840bdb8cb488aa6f",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 190,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 190,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_a943dae332c748eca292e9e718ebbd04"
          }
        },
        "77cbfbea1be94d89a788aaad076040b1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_aad9be6a157344bc8136e7c2f8aa77a4",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 190/190 [00:00&lt;00:00, 4.69kB/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_47ac61b9bcf34c72a4e267bbd4292b8a"
          }
        },
        "7145a8e992d94e4399b713951a11f99f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "b5fc5407f83e422d87903143950330d1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "8a252a9ed8ec4a3c840bdb8cb488aa6f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "a943dae332c748eca292e9e718ebbd04": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "aad9be6a157344bc8136e7c2f8aa77a4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "47ac61b9bcf34c72a4e267bbd4292b8a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "775a77a7c94149b39998efdf03ccab7b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HBoxView",
            "_dom_classes": [],
            "_model_name": "HBoxModel",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "box_style": "",
            "layout": "IPY_MODEL_8089acca3d4b4ba189430f8b8dab3fb9",
            "_model_module": "@jupyter-widgets/controls",
            "children": [
              "IPY_MODEL_df29b9181a8e407c8a3d52b1da00df41",
              "IPY_MODEL_588de86f415746da8498a92913630c31",
              "IPY_MODEL_bb03eb3788004bbe8926d242be0896a6"
            ]
          }
        },
        "8089acca3d4b4ba189430f8b8dab3fb9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "df29b9181a8e407c8a3d52b1da00df41": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_2caca3cd88954a649d9143a3c32a54ec",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": "Batches: 100%",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_b93a40c37a8f47cd9a0e19be2d67772d"
          }
        },
        "588de86f415746da8498a92913630c31": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "ProgressView",
            "style": "IPY_MODEL_b468a29dfecf419bbc38a4deb5659e37",
            "_dom_classes": [],
            "description": "",
            "_model_name": "FloatProgressModel",
            "bar_style": "success",
            "max": 286,
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": 286,
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "orientation": "horizontal",
            "min": 0,
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_4beaf164920647aaad6496a116ce7147"
          }
        },
        "bb03eb3788004bbe8926d242be0896a6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "HTMLView",
            "style": "IPY_MODEL_49a23b684b4045bfb227f5065829825e",
            "_dom_classes": [],
            "description": "",
            "_model_name": "HTMLModel",
            "placeholder": "​",
            "_view_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "value": " 286/286 [00:03&lt;00:00, 105.50it/s]",
            "_view_count": null,
            "_view_module_version": "1.5.0",
            "description_tooltip": null,
            "_model_module": "@jupyter-widgets/controls",
            "layout": "IPY_MODEL_9d46484080f14552a3bf5fe8fa29b8ea"
          }
        },
        "2caca3cd88954a649d9143a3c32a54ec": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "b93a40c37a8f47cd9a0e19be2d67772d": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "b468a29dfecf419bbc38a4deb5659e37": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "ProgressStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "bar_color": null,
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "4beaf164920647aaad6496a116ce7147": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        },
        "49a23b684b4045bfb227f5065829825e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_view_name": "StyleView",
            "_model_name": "DescriptionStyleModel",
            "description_width": "",
            "_view_module": "@jupyter-widgets/base",
            "_model_module_version": "1.5.0",
            "_view_count": null,
            "_view_module_version": "1.2.0",
            "_model_module": "@jupyter-widgets/controls"
          }
        },
        "9d46484080f14552a3bf5fe8fa29b8ea": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_view_name": "LayoutView",
            "grid_template_rows": null,
            "right": null,
            "justify_content": null,
            "_view_module": "@jupyter-widgets/base",
            "overflow": null,
            "_model_module_version": "1.2.0",
            "_view_count": null,
            "flex_flow": null,
            "width": null,
            "min_width": null,
            "border": null,
            "align_items": null,
            "bottom": null,
            "_model_module": "@jupyter-widgets/base",
            "top": null,
            "grid_column": null,
            "overflow_y": null,
            "overflow_x": null,
            "grid_auto_flow": null,
            "grid_area": null,
            "grid_template_columns": null,
            "flex": null,
            "_model_name": "LayoutModel",
            "justify_items": null,
            "grid_row": null,
            "max_height": null,
            "align_content": null,
            "visibility": null,
            "align_self": null,
            "height": null,
            "min_height": null,
            "padding": null,
            "grid_auto_rows": null,
            "grid_gap": null,
            "max_width": null,
            "order": null,
            "_view_module_version": "1.2.0",
            "grid_template_areas": null,
            "object_position": null,
            "object_fit": null,
            "grid_auto_columns": null,
            "margin": null,
            "display": null,
            "left": null
          }
        }
      }
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/tomasonjo/blogs/blob/master/pyg2neo/Movie_recommendations.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "v_F-Vr4Yz2A9",
        "outputId": "d1bc83c0-1c15-4664-d7f1-54655659b635"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Collecting sentence_transformers\n",
            "  Downloading sentence-transformers-2.1.0.tar.gz (78 kB)\n",
            "\u001b[K     |████████████████████████████████| 78 kB 953 kB/s \n",
            "\u001b[?25hCollecting neo4j\n",
            "  Downloading neo4j-4.4.1.tar.gz (89 kB)\n",
            "\u001b[K     |████████████████████████████████| 89 kB 6.9 MB/s \n",
            "\u001b[?25hCollecting transformers<5.0.0,>=4.6.0\n",
            "  Downloading transformers-4.15.0-py3-none-any.whl (3.4 MB)\n",
            "\u001b[K     |████████████████████████████████| 3.4 MB 45.9 MB/s \n",
            "\u001b[?25hCollecting tokenizers>=0.10.3\n",
            "  Downloading tokenizers-0.11.4-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (6.8 MB)\n",
            "\u001b[K     |████████████████████████████████| 6.8 MB 20.3 MB/s \n",
            "\u001b[?25hRequirement already satisfied: tqdm in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (4.62.3)\n",
            "Requirement already satisfied: torch>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (1.10.0+cu111)\n",
            "Requirement already satisfied: torchvision in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (0.11.1+cu111)\n",
            "Requirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (1.19.5)\n",
            "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (1.0.2)\n",
            "Requirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (1.4.1)\n",
            "Requirement already satisfied: nltk in /usr/local/lib/python3.7/dist-packages (from sentence_transformers) (3.2.5)\n",
            "Collecting sentencepiece\n",
            "  Downloading sentencepiece-0.1.96-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB)\n",
            "\u001b[K     |████████████████████████████████| 1.2 MB 43.5 MB/s \n",
            "\u001b[?25hCollecting huggingface-hub\n",
            "  Downloading huggingface_hub-0.4.0-py3-none-any.whl (67 kB)\n",
            "\u001b[K     |████████████████████████████████| 67 kB 5.4 MB/s \n",
            "\u001b[?25hRequirement already satisfied: typing-extensions in /usr/local/lib/python3.7/dist-packages (from torch>=1.6.0->sentence_transformers) (3.10.0.2)\n",
            "Collecting sacremoses\n",
            "  Downloading sacremoses-0.0.47-py2.py3-none-any.whl (895 kB)\n",
            "\u001b[K     |████████████████████████████████| 895 kB 48.4 MB/s \n",
            "\u001b[?25hRequirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.7/dist-packages (from transformers<5.0.0,>=4.6.0->sentence_transformers) (2019.12.20)\n",
            "Collecting tokenizers>=0.10.3\n",
            "  Downloading tokenizers-0.10.3-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (3.3 MB)\n",
            "\u001b[K     |████████████████████████████████| 3.3 MB 45.2 MB/s \n",
            "\u001b[?25hRequirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.7/dist-packages (from transformers<5.0.0,>=4.6.0->sentence_transformers) (21.3)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from transformers<5.0.0,>=4.6.0->sentence_transformers) (2.23.0)\n",
            "Collecting pyyaml>=5.1\n",
            "  Downloading PyYAML-6.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (596 kB)\n",
            "\u001b[K     |████████████████████████████████| 596 kB 46.6 MB/s \n",
            "\u001b[?25hRequirement already satisfied: filelock in /usr/local/lib/python3.7/dist-packages (from transformers<5.0.0,>=4.6.0->sentence_transformers) (3.4.2)\n",
            "Requirement already satisfied: importlib-metadata in /usr/local/lib/python3.7/dist-packages (from transformers<5.0.0,>=4.6.0->sentence_transformers) (4.10.0)\n",
            "Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.7/dist-packages (from packaging>=20.0->transformers<5.0.0,>=4.6.0->sentence_transformers) (3.0.6)\n",
            "Requirement already satisfied: pytz in /usr/local/lib/python3.7/dist-packages (from neo4j) (2018.9)\n",
            "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->transformers<5.0.0,>=4.6.0->sentence_transformers) (3.7.0)\n",
            "Requirement already satisfied: six in /usr/local/lib/python3.7/dist-packages (from nltk->sentence_transformers) (1.15.0)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->transformers<5.0.0,>=4.6.0->sentence_transformers) (2021.10.8)\n",
            "Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests->transformers<5.0.0,>=4.6.0->sentence_transformers) (1.24.3)\n",
            "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->transformers<5.0.0,>=4.6.0->sentence_transformers) (2.10)\n",
            "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->transformers<5.0.0,>=4.6.0->sentence_transformers) (3.0.4)\n",
            "Requirement already satisfied: click in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers<5.0.0,>=4.6.0->sentence_transformers) (7.1.2)\n",
            "Requirement already satisfied: joblib in /usr/local/lib/python3.7/dist-packages (from sacremoses->transformers<5.0.0,>=4.6.0->sentence_transformers) (1.1.0)\n",
            "Requirement already satisfied: threadpoolctl>=2.0.0 in /usr/local/lib/python3.7/dist-packages (from scikit-learn->sentence_transformers) (3.0.0)\n",
            "Requirement already satisfied: pillow!=8.3.0,>=5.3.0 in /usr/local/lib/python3.7/dist-packages (from torchvision->sentence_transformers) (7.1.2)\n",
            "Building wheels for collected packages: sentence-transformers, neo4j\n",
            "  Building wheel for sentence-transformers (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for sentence-transformers: filename=sentence_transformers-2.1.0-py3-none-any.whl size=120999 sha256=53ec346297795728a6d25f0f15442480e68c1bb184737154560f8bfdc2f49c3b\n",
            "  Stored in directory: /root/.cache/pip/wheels/90/f0/bb/ed1add84da70092ea526466eadc2bfb197c4bcb8d4fa5f7bad\n",
            "  Building wheel for neo4j (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for neo4j: filename=neo4j-4.4.1-py3-none-any.whl size=114783 sha256=5eddfc90cd69ad752e1135197e11765e703d298b8cc02eb14d41e383b7a97954\n",
            "  Stored in directory: /root/.cache/pip/wheels/fd/15/02/8379f87426cd1b74fd5891f49df9fb978423feb97650cd639b\n",
            "Successfully built sentence-transformers neo4j\n",
            "Installing collected packages: pyyaml, tokenizers, sacremoses, huggingface-hub, transformers, sentencepiece, sentence-transformers, neo4j\n",
            "  Attempting uninstall: pyyaml\n",
            "    Found existing installation: PyYAML 3.13\n",
            "    Uninstalling PyYAML-3.13:\n",
            "      Successfully uninstalled PyYAML-3.13\n",
            "Successfully installed huggingface-hub-0.4.0 neo4j-4.4.1 pyyaml-6.0 sacremoses-0.0.47 sentence-transformers-2.1.0 sentencepiece-0.1.96 tokenizers-0.10.3 transformers-4.15.0\n"
          ]
        }
      ],
      "source": [
        "!pip install sentence_transformers neo4j"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "!pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-1.10.0+cpu.html"
      ],
      "metadata": {
        "id": "3QyplUOxF-73",
        "outputId": "4d14f289-cb50-43e2-cbe1-88f5e3815be8",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Looking in links: https://data.pyg.org/whl/torch-1.10.0+cpu.html\n",
            "Collecting torch-scatter\n",
            "  Downloading https://data.pyg.org/whl/torch-1.10.0%2Bcpu/torch_scatter-2.0.9-cp37-cp37m-linux_x86_64.whl (291 kB)\n",
            "\u001b[K     |████████████████████████████████| 291 kB 4.9 MB/s \n",
            "\u001b[?25hCollecting torch-sparse\n",
            "  Downloading https://data.pyg.org/whl/torch-1.10.0%2Bcpu/torch_sparse-0.6.12-cp37-cp37m-linux_x86_64.whl (639 kB)\n",
            "\u001b[K     |████████████████████████████████| 639 kB 43.1 MB/s \n",
            "\u001b[?25hCollecting torch-cluster\n",
            "  Downloading https://data.pyg.org/whl/torch-1.10.0%2Bcpu/torch_cluster-1.5.9-cp37-cp37m-linux_x86_64.whl (328 kB)\n",
            "\u001b[K     |████████████████████████████████| 328 kB 37.2 MB/s \n",
            "\u001b[?25hCollecting torch-geometric\n",
            "  Downloading torch_geometric-2.0.3.tar.gz (370 kB)\n",
            "\u001b[K     |████████████████████████████████| 370 kB 5.3 MB/s \n",
            "\u001b[?25hRequirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from torch-sparse) (1.4.1)\n",
            "Requirement already satisfied: numpy in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (1.19.5)\n",
            "Requirement already satisfied: tqdm in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (4.62.3)\n",
            "Requirement already satisfied: networkx in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (2.6.3)\n",
            "Requirement already satisfied: scikit-learn in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (1.0.2)\n",
            "Requirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (2.23.0)\n",
            "Requirement already satisfied: pandas in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (1.1.5)\n",
            "Collecting rdflib\n",
            "  Downloading rdflib-6.1.1-py3-none-any.whl (482 kB)\n",
            "\u001b[K     |████████████████████████████████| 482 kB 49.4 MB/s \n",
            "\u001b[?25hRequirement already satisfied: googledrivedownloader in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (0.4)\n",
            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (2.11.3)\n",
            "Requirement already satisfied: pyparsing in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (3.0.6)\n",
            "Collecting yacs\n",
            "  Downloading yacs-0.1.8-py3-none-any.whl (14 kB)\n",
            "Requirement already satisfied: PyYAML in /usr/local/lib/python3.7/dist-packages (from torch-geometric) (6.0)\n",
            "Requirement already satisfied: MarkupSafe>=0.23 in /usr/local/lib/python3.7/dist-packages (from jinja2->torch-geometric) (2.0.1)\n",
            "Requirement already satisfied: python-dateutil>=2.7.3 in /usr/local/lib/python3.7/dist-packages (from pandas->torch-geometric) (2.8.2)\n",
            "Requirement already satisfied: pytz>=2017.2 in /usr/local/lib/python3.7/dist-packages (from pandas->torch-geometric) (2018.9)\n",
            "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.7/dist-packages (from python-dateutil>=2.7.3->pandas->torch-geometric) (1.15.0)\n",
            "Collecting isodate\n",
            "  Downloading isodate-0.6.1-py2.py3-none-any.whl (41 kB)\n",
            "\u001b[K     |████████████████████████████████| 41 kB 586 kB/s \n",
            "\u001b[?25hRequirement already satisfied: setuptools in /usr/local/lib/python3.7/dist-packages (from rdflib->torch-geometric) (57.4.0)\n",
            "Requirement already satisfied: importlib-metadata in /usr/local/lib/python3.7/dist-packages (from rdflib->torch-geometric) (4.10.0)\n",
            "Requirement already satisfied: typing-extensions>=3.6.4 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->rdflib->torch-geometric) (3.10.0.2)\n",
            "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->rdflib->torch-geometric) (3.7.0)\n",
            "Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests->torch-geometric) (1.24.3)\n",
            "Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->torch-geometric) (2.10)\n",
            "Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->torch-geometric) (3.0.4)\n",
            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->torch-geometric) (2021.10.8)\n",
            "Requirement already satisfied: threadpoolctl>=2.0.0 in /usr/local/lib/python3.7/dist-packages (from scikit-learn->torch-geometric) (3.0.0)\n",
            "Requirement already satisfied: joblib>=0.11 in /usr/local/lib/python3.7/dist-packages (from scikit-learn->torch-geometric) (1.1.0)\n",
            "Building wheels for collected packages: torch-geometric\n",
            "  Building wheel for torch-geometric (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
            "  Created wheel for torch-geometric: filename=torch_geometric-2.0.3-py3-none-any.whl size=581968 sha256=7214241153814c2bfba7d03e5c88b4b89c66ac8c7d73764ac839eee0ebd4e308\n",
            "  Stored in directory: /root/.cache/pip/wheels/c3/2a/58/87ce0508964d4def1aafb92750c4f3ac77038efd1b9a89dcf5\n",
            "Successfully built torch-geometric\n",
            "Installing collected packages: isodate, yacs, rdflib, torch-sparse, torch-scatter, torch-geometric, torch-cluster\n",
            "Successfully installed isodate-0.6.1 rdflib-6.1.1 torch-cluster-1.5.9 torch-geometric-2.0.3 torch-scatter-2.0.9 torch-sparse-0.6.12 yacs-0.1.8\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import torch\n",
        "import pandas as pd\n",
        "from torch.nn import Linear\n",
        "import torch.nn.functional as F\n",
        "from sentence_transformers import SentenceTransformer\n",
        "\n",
        "import torch_geometric.transforms as T\n",
        "from torch_geometric.nn import SAGEConv, to_hetero\n",
        "\n",
        "from torch_geometric.data import HeteroData\n",
        "from torch_geometric.transforms import ToUndirected, RandomLinkSplit\n",
        "\n",
        "device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')"
      ],
      "metadata": {
        "id": "RlHc8p58z6b1"
      },
      "execution_count": 3,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "print(device)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "Qahop_-4FmQu",
        "outputId": "552a6054-9576-4ab9-a248-9e7bfde901f4"
      },
      "execution_count": 4,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "cpu\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "from neo4j import GraphDatabase\n",
        "\n",
        "url= 'bolt://3.86.43.255:7687'\n",
        "user = 'neo4j'\n",
        "password = 'company-science-journals'\n",
        "\n",
        "driver = GraphDatabase.driver(url, auth=(user, password))\n",
        "\n",
        "def fetch_data(query):\n",
        "  with driver.session() as session:\n",
        "    result = session.run(query)\n",
        "    return pd.DataFrame([r.values() for r in result], columns=result.keys())\n"
      ],
      "metadata": {
        "id": "4ZKcA2wW5K_J"
      },
      "execution_count": 5,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def load_node(cypher, index_col, encoders=None, **kwargs):\n",
        "    df = fetch_data(cypher)\n",
        "    df.set_index(index_col, inplace=True)\n",
        "    mapping = {index: i for i, index in enumerate(df.index.unique())}\n",
        "\n",
        "    x = None\n",
        "    if encoders is not None:\n",
        "        xs = [encoder(df[col]) for col, encoder in encoders.items()]\n",
        "        x = torch.cat(xs, dim=-1)\n",
        "\n",
        "    return x, mapping"
      ],
      "metadata": {
        "id": "bX-Dlugy0Fyu"
      },
      "execution_count": 6,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def load_edge(cypher, src_index_col, src_mapping, dst_index_col, dst_mapping,\n",
        "                  encoders=None, **kwargs):\n",
        "    df = fetch_data(cypher)\n",
        "\n",
        "    src = [src_mapping[index] for index in df[src_index_col]]\n",
        "    dst = [dst_mapping[index] for index in df[dst_index_col]]\n",
        "    edge_index = torch.tensor([src, dst])\n",
        "\n",
        "    edge_attr = None\n",
        "    if encoders is not None:\n",
        "        edge_attrs = [encoder(df[col]) for col, encoder in encoders.items()]\n",
        "        edge_attr = torch.cat(edge_attrs, dim=-1)\n",
        "\n",
        "    return edge_index, edge_attr"
      ],
      "metadata": {
        "id": "b-QNMfvN3tk9"
      },
      "execution_count": 7,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "class SequenceEncoder(object):\n",
        "    # The 'SequenceEncoder' encodes raw column strings into embeddings.\n",
        "    def __init__(self, model_name='all-MiniLM-L6-v2', device=None):\n",
        "        self.device = device\n",
        "        self.model = SentenceTransformer(model_name, device=device)\n",
        "\n",
        "    @torch.no_grad()\n",
        "    def __call__(self, df):\n",
        "        x = self.model.encode(df.values, show_progress_bar=True,\n",
        "                              convert_to_tensor=True, device=self.device)\n",
        "        return x.cpu()"
      ],
      "metadata": {
        "id": "8xF_x6SZ3wIy"
      },
      "execution_count": 8,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "class GenresEncoder(object):\n",
        "    # The 'GenreEncoder' splits the raw column strings by 'sep' and converts\n",
        "    # individual elements to categorical labels.\n",
        "    def __init__(self, sep='|'):\n",
        "        self.sep = sep\n",
        "\n",
        "    def __call__(self, df):\n",
        "        genres = set(g for col in df.values for g in col.split(self.sep))\n",
        "        mapping = {genre: i for i, genre in enumerate(genres)}\n",
        "\n",
        "        x = torch.zeros(len(df), len(mapping))\n",
        "        for i, col in enumerate(df.values):\n",
        "            for genre in col.split(self.sep):\n",
        "                x[i, mapping[genre]] = 1\n",
        "        return x"
      ],
      "metadata": {
        "id": "B3MrEGO933bA"
      },
      "execution_count": 9,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "class IdentityEncoder(object):\n",
        "    # The 'IdentityEncoder' takes the raw column values and converts them to\n",
        "    # PyTorch tensors.\n",
        "    def __init__(self, dtype=None):\n",
        "        self.dtype = dtype\n",
        "\n",
        "    def __call__(self, df):\n",
        "        return torch.from_numpy(df.values).to(self.dtype)"
      ],
      "metadata": {
        "id": "n1QiEzKZ348J"
      },
      "execution_count": 33,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "user_x, user_mapping = load_node(\"MATCH (u:User) RETURN u.userId AS userId\" , index_col='userId')\n"
      ],
      "metadata": {
        "id": "9a3plIYf74tX"
      },
      "execution_count": 34,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "movie_query = \"\"\"\n",
        "MATCH (m:Movie)-[:IN_GENRE]->(genre:Genre)\n",
        "WITH m, collect(genre.name) AS genres_list\n",
        "RETURN m.movieId AS movieId, m.title AS title, apoc.text.join(genres_list, '|') AS genres\n",
        "\"\"\"\n",
        "\n",
        "movie_x, movie_mapping = load_node(\n",
        "    movie_query, \n",
        "    index_col='movieId', encoders={\n",
        "        'title': SequenceEncoder(),\n",
        "        'genres': GenresEncoder()\n",
        "    })"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 497,
          "referenced_widgets": [
            "6e42a7a80bcc4f18853085bb4e9f1602",
            "4fd077d31fb944d39674308ead3f79ec",
            "484630f8782d470e81cc17f24a285dd6",
            "67c2c79a590f4de884f7a61e93d35df6",
            "b85481e270a5460a86268f63b58fc028",
            "52efc84bf9b34b468177dcd86d426d01",
            "49f974d6b1ec42c99eaf5ac1c0af6582",
            "93bffa61924546e59a3b427e6ef61fba",
            "d6b5c4130eef47549badcb8883b9faf2",
            "01d56aee12c84b5bac09486e4a5c291a",
            "1d976d0faa87429587d8e629c61bc790",
            "7ee93e9261f1426b8e22eda01fa4afaa",
            "a2a2827235624287812522be4550e4e3",
            "73457a3e28ff4003bcd86897829a3f52",
            "267895bda38347edb26bd6c76d87a67e",
            "ed171a21a44b4cb2bc9323a5cea1dcc7",
            "2e8dcefe6e0b4c3aa9354aad5e25c9ea",
            "667e3bbff63a407e9a0ca8814d66a11f",
            "4dda9c1ef5f643ada30c1ea5574836a1",
            "4798de8b355c4251928bde9109af76a1",
            "e636b4a360a44df7902b17032a50b014",
            "95a6df8d509a443a9560f973742e8d7c",
            "7fdfc2ae6c254b4ea3e1bfa7bd6bf1d9",
            "d2f720de95954d2b8742ea91989c39b2",
            "1189c090eb224630b1dc92b94fe8e71d",
            "26070f66567445419eb360b47270cf15",
            "1dfa4b09793c42929aab78eea6dbee10",
            "f5ac3ef85af0407b9abd5fabc6681824",
            "976e4173c34249cf83e1fcea2fd1c568",
            "49e268eb8c5444fca4db6450ce302852",
            "347b4df225b44f1f98a181d10620b5fd",
            "8c0ede7d55fd460798c6c93048ff4412",
            "6757017d028049c2ad4fb9ccebaa9110",
            "58acdcf5f0d540f2a21beec2a65e83de",
            "5e6c6f3d122b42a88bc2d4470c1cf31e",
            "712c0cb08ed84590866012ac64b3bb8d",
            "1a52022b38844932a157faab32792519",
            "7793533cdd314f36b96ae534f8a2d503",
            "f13b16fec0f34959b69e1da99a9cad63",
            "a304eac3795142e79ad03965def10eb9",
            "284728ec737f41d58260c8e8c00af9e3",
            "28721fe261aa4db3b9f9f22daf376f66",
            "b8799b5a37c64a5b9c797ad3b1461a37",
            "17b89e0ca1274e70932fd7240107191c",
            "93cc8fee07c44ca0afc406fe8d4613c9",
            "6dc05c71373b4f76992ce4a6c906df6e",
            "1ddd930ef90c489f81ae2a26dc10acf0",
            "d9589531cfd941d2b36a60d85f6b36ac",
            "25998ca65b4d4861aed99f2240c535de",
            "88b442811b99493588f553a8772fa5c6",
            "9dd6f123fba048e69291fa45f7ad531f",
            "63d58abaf4e043c0946d74925eaf7b16",
            "d88898dcc0674603a85fa03e37a99742",
            "9333c6f9617b4b43a74e294a6712ee38",
            "6bc0f40661fc4694a1a8dd4027fe335d",
            "fadde8c04f5d4a5ebfc2a364feaac196",
            "078c67724a3544f68386dea213a238ad",
            "a75d669ce8ed4c5f8c61d14865d93e14",
            "a4dfcbaa8167439e90bbae46692d8dd1",
            "792a35155f7d400196d1f11db5083c5a",
            "b7d818c04a494e918d902f3564abdb5b",
            "6e7d48547e01479f989c2709fef2fdb3",
            "a2835377c4434f389e509430607b7f1b",
            "0fd715eb636f4be0998ce846bc0f786b",
            "020553f4826146d0a8f5a5fa335bb293",
            "26e2dd17fee3460d9ee5babe07335ebe",
            "fa7985ea84864b9bb947efbeff09ff1f",
            "1a3a2ac407bf44b48894e020449a25e5",
            "d59ca158a4834c1187cf0852f7858600",
            "9ab0b65c3ab047c6b2e2ddbdc7f01370",
            "5cd3e5aa48c74173beabc1ad5c75a9a0",
            "10b48d350dfc467f9e956af4f75dff44",
            "44e0113c67f0486cb574fce896bfe327",
            "0d40462afae047c9bc3fa88044e2e273",
            "b46cbab2e77d489c8bbc73b5d7b99b82",
            "a060ee2943014ce2b82b57fa7f6e9168",
            "0eed01ccde454953bf708cc344c84dbd",
            "8589805155334893a469738b2e66972f",
            "80d20d44b7164ca19cbe5acab089192c",
            "b6776d7d130647dba5f89f4c9b58ebc4",
            "42a8702228e74f678e71212e7f347a8a",
            "09cacc55d1914adcb229f3850b8b0f3a",
            "415d9472866c400f989631b8ef73d38e",
            "f91329ba17e84958b05e519e71a9260a",
            "cfd8cca64ca2444b960a40b16d75280f",
            "4d0f09a64c1d4fc7a253ce2b9cce7b79",
            "36626cc4d6ec4605b891b5ebd939ac6a",
            "39cb1fec3984463a8996e7deef526327",
            "d66b2963841e4d6e8ca70d59c915f758",
            "2ef788647f764e67a94e9d62ad19ecd8",
            "3b635a517fdf40349a8e371a7c988302",
            "0b473696b2b84a6ca493ec60e378bb3d",
            "6202f1e33ebe431c8b7da6fcbb8b39ae",
            "3beb1e4523394f52a9af817d85bcbd9d",
            "b337960b719a48c481df5f862cd906be",
            "e345a268eac549e082022e61f1785960",
            "a5c90f34a9dd4804bf45436529af50a9",
            "9c4a20d40fa6432ca9df7a597f845cb6",
            "0df73cc5352b4ca69fc05c1dfa09294a",
            "770b9decb42045ea8da680823b0135dd",
            "2cdbdd7db2654f96ba70250fab87e038",
            "c0dd89c0349c4df1b18d1f7fd3da665d",
            "e4b36b50ad974b27977def287e91fc62",
            "1138df11c56d495cb5a7c0e53470120d",
            "e7cda2225d7441068409fd0511e16c57",
            "03d1bbb774904159acce8633302f235c",
            "b4f13a61a66b42e5aea7054af9b46dc5",
            "45e1f0145f9a41738b2ca84b4ef07799",
            "17e8f46b67cc4ac683d83c99b82ec8be",
            "b6b725950b364bf19f47999cdd6afbaf",
            "a8fafcd1a99a45efadf4dde0ba1d41df",
            "445cfe326830407d89a03872e12060de",
            "b5c7c1390eae4e9fa497641962207a58",
            "59a463f8e1854cd2992feb7549d101d4",
            "de6ae923e60b447ebaeeec7d5a6b8303",
            "348d87b38a7d433da0199c9493b0a88f",
            "d79d549ccd6f45bfbe30223c02d9f556",
            "903c403ed2344fd19e053d1669d4cd44",
            "f8b04579ac1c4b7aa5b3b773a500d876",
            "b239033e712f4b928657d9b4c55a80a1",
            "dac17cd757b94066b12afd3879da7835",
            "b1131fa026bd4af3945105d5ed2475b8",
            "d74c2af975df483abb4d20d9a34c2933",
            "89fa67324b344e91b72a06ecf5c3348f",
            "2fc7361eef4548a5927a0f8de71c20fc",
            "9906cf9e0fa74d949caf0299959801c0",
            "a3abddf18a4c48bdbc2ef95f05844ffa",
            "efca4d3e7ff54f7b98e4801f215b7213",
            "c6223d4a6a3f4f0cbba298e8444139c8",
            "acc419a409b342d6b9d2848590241403",
            "6ab7c0d2725048fd8907ba943a4f5af6",
            "a33ee87994984b188a9d461728500a28",
            "a4f54f46d4ce44479d4d96b31bea574c",
            "43537a3e006b4edd8ec2b391ab6f7c8d",
            "15577a3d8a224a00b067d87a4ec1dad2",
            "33100b8bb41f4768933e9747e1e746ed",
            "56cbf7ffc64a4ef6aa4cd840d475b2e7",
            "4f6c4721297b45a89d08fd0c300f2290",
            "ce95d840b5444dce92bef29290433df3",
            "8b6967ba48754103a2003ff7d2987025",
            "0513677b926640ffa79ec52d4f034499",
            "ed48348145b547cbb9531fe64cae781b",
            "e502bd16650c47aa936214a87f0359ec",
            "e3cac84d8b8a4a4c81546dd4947b3d5b",
            "c68141e5464e41039b2a884f78aeed6b",
            "420cb85096064891a3feb56796dfdcc1",
            "459d4965a37745199980a2f7741a16a3",
            "77cbfbea1be94d89a788aaad076040b1",
            "7145a8e992d94e4399b713951a11f99f",
            "b5fc5407f83e422d87903143950330d1",
            "8a252a9ed8ec4a3c840bdb8cb488aa6f",
            "a943dae332c748eca292e9e718ebbd04",
            "aad9be6a157344bc8136e7c2f8aa77a4",
            "47ac61b9bcf34c72a4e267bbd4292b8a",
            "775a77a7c94149b39998efdf03ccab7b",
            "8089acca3d4b4ba189430f8b8dab3fb9",
            "df29b9181a8e407c8a3d52b1da00df41",
            "588de86f415746da8498a92913630c31",
            "bb03eb3788004bbe8926d242be0896a6",
            "2caca3cd88954a649d9143a3c32a54ec",
            "b93a40c37a8f47cd9a0e19be2d67772d",
            "b468a29dfecf419bbc38a4deb5659e37",
            "4beaf164920647aaad6496a116ce7147",
            "49a23b684b4045bfb227f5065829825e",
            "9d46484080f14552a3bf5fe8fa29b8ea"
          ]
        },
        "id": "ORpJYlwD89K6",
        "outputId": "24f09cf7-bf31-43a4-992c-3b64f9b20c4b"
      },
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "6e42a7a80bcc4f18853085bb4e9f1602",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/1.18k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "7ee93e9261f1426b8e22eda01fa4afaa",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/10.2k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "7fdfc2ae6c254b4ea3e1bfa7bd6bf1d9",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/612 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "58acdcf5f0d540f2a21beec2a65e83de",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/116 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "93cc8fee07c44ca0afc406fe8d4613c9",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/39.3k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "fadde8c04f5d4a5ebfc2a364feaac196",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/349 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "fa7985ea84864b9bb947efbeff09ff1f",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/90.9M [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "8589805155334893a469738b2e66972f",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/53.0 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "d66b2963841e4d6e8ca70d59c915f758",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/112 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "770b9decb42045ea8da680823b0135dd",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/466k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "a8fafcd1a99a45efadf4dde0ba1d41df",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/350 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "b1131fa026bd4af3945105d5ed2475b8",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/13.2k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "a4f54f46d4ce44479d4d96b31bea574c",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/232k [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "e3cac84d8b8a4a4c81546dd4947b3d5b",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Downloading:   0%|          | 0.00/190 [00:00<?, ?B/s]"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "775a77a7c94149b39998efdf03ccab7b",
              "version_minor": 0,
              "version_major": 2
            },
            "text/plain": [
              "Batches:   0%|          | 0/286 [00:00<?, ?it/s]"
            ]
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "rating_query = \"\"\"\n",
        "MATCH (u:User)-[r:RATED]->(m:Movie) \n",
        "RETURN u.userId AS userId, m.movieId AS movieId, r.rating AS rating\n",
        "\"\"\"\n",
        "\n",
        "edge_index, edge_label = load_edge(\n",
        "    rating_query,\n",
        "    src_index_col='userId',\n",
        "    src_mapping=user_mapping,\n",
        "    dst_index_col='movieId',\n",
        "    dst_mapping=movie_mapping,\n",
        "    encoders={'rating': IdentityEncoder(dtype=torch.long)},\n",
        ")"
      ],
      "metadata": {
        "id": "tR4B7auh9_bu"
      },
      "execution_count": 35,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "data = HeteroData()\n",
        "data['user'].num_nodes = len(user_mapping)  # Users do not have any features.\n",
        "# Add user node features for message passing:\n",
        "data['user'].x = torch.eye(data['user'].num_nodes, device=device)\n",
        "del data['user'].num_nodes\n",
        "\n",
        "data['movie'].x = movie_x\n",
        "data['user', 'rates', 'movie'].edge_index = edge_index\n",
        "data['user', 'rates', 'movie'].edge_label = edge_label\n",
        "data.to(device)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "v5WEVdiz_Xai",
        "outputId": "21a6a8b4-b8f5-4cb9-dc1c-b54e64a2a04d"
      },
      "execution_count": 36,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "HeteroData(\n",
              "  \u001b[1muser\u001b[0m={ x=[671, 671] },\n",
              "  \u001b[1mmovie\u001b[0m={ x=[9125, 404] },\n",
              "  \u001b[1m(user, rates, movie)\u001b[0m={\n",
              "    edge_index=[2, 100004],\n",
              "    edge_label=[100004]\n",
              "  }\n",
              ")"
            ]
          },
          "metadata": {},
          "execution_count": 36
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import os.path as osp\n",
        "from torch_geometric.datasets import MovieLens\n",
        "dataset = 'MyLens'\n",
        "path = osp.join('.', 'data', dataset)\n",
        "dataset = MovieLens(path, model_name='all-MiniLM-L6-v2')\n",
        "data_nice = dataset[0].to(device)"
      ],
      "metadata": {
        "id": "jQ8oyKKkX818"
      },
      "execution_count": 31,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "data = ToUndirected()(data)\n",
        "del data['movie', 'rev_rates', 'user'].edge_label  # Remove \"reverse\" label.\n",
        "\n",
        "# 2. Perform a link-level split into training, validation, and test edges.\n",
        "transform = RandomLinkSplit(\n",
        "    num_val=0.1,\n",
        "    num_test=0.1,\n",
        "    neg_sampling_ratio=0.0,\n",
        "    edge_types=[('user', 'rates', 'movie')],\n",
        "    rev_edge_types=[('movie', 'rev_rates', 'user')],\n",
        ")\n",
        "train_data, val_data, test_data = transform(data)"
      ],
      "metadata": {
        "id": "wWPosXvc_nYk"
      },
      "execution_count": 37,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "class GNNEncoder(torch.nn.Module):\n",
        "    def __init__(self, hidden_channels, out_channels):\n",
        "        super().__init__()\n",
        "        self.conv1 = SAGEConv((-1, -1), hidden_channels)\n",
        "        self.conv2 = SAGEConv((-1, -1), out_channels)\n",
        "\n",
        "    def forward(self, x, edge_index):\n",
        "        x = self.conv1(x, edge_index).relu()\n",
        "        x = self.conv2(x, edge_index)\n",
        "        return x\n",
        "\n",
        "\n",
        "class EdgeDecoder(torch.nn.Module):\n",
        "    def __init__(self, hidden_channels):\n",
        "        super().__init__()\n",
        "        self.lin1 = Linear(2 * hidden_channels, hidden_channels)\n",
        "        self.lin2 = Linear(hidden_channels, 1)\n",
        "\n",
        "    def forward(self, z_dict, edge_label_index):\n",
        "        row, col = edge_label_index\n",
        "        z = torch.cat([z_dict['user'][row], z_dict['movie'][col]], dim=-1)\n",
        "\n",
        "        z = self.lin1(z).relu()\n",
        "        z = self.lin2(z)\n",
        "        return z.view(-1)"
      ],
      "metadata": {
        "id": "_9p-mP6wAQ9-"
      },
      "execution_count": 38,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "class Model(torch.nn.Module):\n",
        "    def __init__(self, hidden_channels):\n",
        "        super().__init__()\n",
        "        self.encoder = GNNEncoder(hidden_channels, hidden_channels)\n",
        "        self.encoder = to_hetero(self.encoder, data.metadata(), aggr='sum')\n",
        "        self.decoder = EdgeDecoder(hidden_channels)\n",
        "\n",
        "    def forward(self, x_dict, edge_index_dict, edge_label_index):\n",
        "        z_dict = self.encoder(x_dict, edge_index_dict)\n",
        "        return self.decoder(z_dict, edge_label_index)"
      ],
      "metadata": {
        "id": "HNT9FX3eAjwt"
      },
      "execution_count": 39,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "model = Model(hidden_channels=32).to(device)"
      ],
      "metadata": {
        "id": "RgPHXNpSAmJ2"
      },
      "execution_count": 40,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "weight = torch.bincount(train_data['user', 'movie'].edge_label)\n",
        "weight = weight.max() / weight\n",
        "\n",
        "def weighted_mse_loss(pred, target, weight=None):\n",
        "    weight = 1. if weight is None else weight[target].to(pred.dtype)\n",
        "    return (weight * (pred - target.to(pred.dtype)).pow(2)).mean()"
      ],
      "metadata": {
        "id": "AYlJTCV8Cfz-"
      },
      "execution_count": 42,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# Due to lazy initialization, we need to run one model step so the number\n",
        "# of parameters can be inferred:\n",
        "with torch.no_grad():\n",
        "    model.encoder(train_data.x_dict, train_data.edge_index_dict)\n",
        "\n",
        "optimizer = torch.optim.Adam(model.parameters(), lr=0.01)"
      ],
      "metadata": {
        "id": "JLlVrSU8AyxT"
      },
      "execution_count": 43,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def train():\n",
        "    model.train()\n",
        "    optimizer.zero_grad()\n",
        "    pred = model(train_data.x_dict, train_data.edge_index_dict,\n",
        "                 train_data['user', 'movie'].edge_label_index)\n",
        "    target = train_data['user', 'movie'].edge_label\n",
        "    loss = weighted_mse_loss(pred, target, weight)\n",
        "    loss.backward()\n",
        "    optimizer.step()\n",
        "    return float(loss)"
      ],
      "metadata": {
        "id": "4dJtEPSuBTDI"
      },
      "execution_count": 44,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "@torch.no_grad()\n",
        "def test(data):\n",
        "    model.eval()\n",
        "    pred = model(data.x_dict, data.edge_index_dict,\n",
        "                 data['user', 'movie'].edge_label_index)\n",
        "    pred = pred.clamp(min=0, max=5)\n",
        "    target = data['user', 'movie'].edge_label.float()\n",
        "    rmse = F.mse_loss(pred, target).sqrt()\n",
        "    return float(rmse)"
      ],
      "metadata": {
        "id": "Rew4KdIGarkO"
      },
      "execution_count": 46,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "for epoch in range(1, 300):\n",
        "    loss = train()\n",
        "    train_rmse = test(train_data)\n",
        "    val_rmse = test(val_data)\n",
        "    test_rmse = test(test_data)\n",
        "    print(f'Epoch: {epoch:03d}, Loss: {loss:.4f}, Train: {train_rmse:.4f}, '\n",
        "          f'Val: {val_rmse:.4f}, Test: {test_rmse:.4f}')"
      ],
      "metadata": {
        "id": "gjndb8tPCPRO",
        "outputId": "3034b09b-6c51-4771-e424-4f9a1201332e",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "execution_count": 47,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Epoch: 001, Loss: 18.1004, Train: 3.2379, Val: 3.2330, Test: 3.2382\n",
            "Epoch: 002, Loss: 16.2975, Train: 2.9616, Val: 2.9584, Test: 2.9636\n",
            "Epoch: 003, Loss: 13.7349, Train: 2.4912, Val: 2.4913, Test: 2.4967\n",
            "Epoch: 004, Loss: 10.1835, Train: 1.7573, Val: 1.7625, Test: 1.7685\n",
            "Epoch: 005, Loss: 6.8211, Train: 1.1065, Val: 1.1074, Test: 1.1145\n",
            "Epoch: 006, Loss: 8.0671, Train: 1.1376, Val: 1.1337, Test: 1.1404\n",
            "Epoch: 007, Loss: 9.3780, Train: 1.1089, Val: 1.1123, Test: 1.1197\n",
            "Epoch: 008, Loss: 7.4036, Train: 1.3432, Val: 1.3500, Test: 1.3569\n",
            "Epoch: 009, Loss: 6.2444, Train: 1.6741, Val: 1.6793, Test: 1.6857\n",
            "Epoch: 010, Loss: 6.5007, Train: 1.8877, Val: 1.8915, Test: 1.8975\n",
            "Epoch: 011, Loss: 7.0878, Train: 1.9598, Val: 1.9631, Test: 1.9691\n",
            "Epoch: 012, Loss: 7.3351, Train: 1.9107, Val: 1.9145, Test: 1.9206\n",
            "Epoch: 013, Loss: 7.1261, Train: 1.7668, Val: 1.7719, Test: 1.7783\n",
            "Epoch: 014, Loss: 6.6227, Train: 1.5553, Val: 1.5622, Test: 1.5692\n",
            "Epoch: 015, Loss: 6.1015, Train: 1.3245, Val: 1.3332, Test: 1.3409\n",
            "Epoch: 016, Loss: 5.8870, Train: 1.1537, Val: 1.1630, Test: 1.1714\n",
            "Epoch: 017, Loss: 6.1142, Train: 1.0895, Val: 1.0984, Test: 1.1072\n",
            "Epoch: 018, Loss: 6.3685, Train: 1.0971, Val: 1.1064, Test: 1.1153\n",
            "Epoch: 019, Loss: 6.2088, Train: 1.1672, Val: 1.1769, Test: 1.1857\n",
            "Epoch: 020, Loss: 5.8298, Train: 1.2915, Val: 1.3007, Test: 1.3092\n",
            "Epoch: 021, Loss: 5.6009, Train: 1.4244, Val: 1.4325, Test: 1.4407\n",
            "Epoch: 022, Loss: 5.6041, Train: 1.5214, Val: 1.5286, Test: 1.5366\n",
            "Epoch: 023, Loss: 5.6980, Train: 1.5617, Val: 1.5683, Test: 1.5764\n",
            "Epoch: 024, Loss: 5.7324, Train: 1.5420, Val: 1.5487, Test: 1.5570\n",
            "Epoch: 025, Loss: 5.6437, Train: 1.4698, Val: 1.4770, Test: 1.4857\n",
            "Epoch: 026, Loss: 5.4567, Train: 1.3616, Val: 1.3696, Test: 1.3789\n",
            "Epoch: 027, Loss: 5.2565, Train: 1.2442, Val: 1.2530, Test: 1.2630\n",
            "Epoch: 028, Loss: 5.1386, Train: 1.1497, Val: 1.1590, Test: 1.1697\n",
            "Epoch: 029, Loss: 5.1304, Train: 1.0993, Val: 1.1086, Test: 1.1197\n",
            "Epoch: 030, Loss: 5.1416, Train: 1.0936, Val: 1.1027, Test: 1.1141\n",
            "Epoch: 031, Loss: 5.0564, Train: 1.1272, Val: 1.1358, Test: 1.1474\n",
            "Epoch: 032, Loss: 4.8843, Train: 1.1913, Val: 1.1992, Test: 1.2108\n",
            "Epoch: 033, Loss: 4.7314, Train: 1.2646, Val: 1.2715, Test: 1.2831\n",
            "Epoch: 034, Loss: 4.6593, Train: 1.3196, Val: 1.3257, Test: 1.3374\n",
            "Epoch: 035, Loss: 4.6315, Train: 1.3371, Val: 1.3425, Test: 1.3545\n",
            "Epoch: 036, Loss: 4.5790, Train: 1.3121, Val: 1.3171, Test: 1.3295\n",
            "Epoch: 037, Loss: 4.4688, Train: 1.2541, Val: 1.2588, Test: 1.2718\n",
            "Epoch: 038, Loss: 4.3296, Train: 1.1853, Val: 1.1894, Test: 1.2029\n",
            "Epoch: 039, Loss: 4.2240, Train: 1.1339, Val: 1.1371, Test: 1.1508\n",
            "Epoch: 040, Loss: 4.1850, Train: 1.1191, Val: 1.1214, Test: 1.1349\n",
            "Epoch: 041, Loss: 4.1579, Train: 1.1371, Val: 1.1385, Test: 1.1519\n",
            "Epoch: 042, Loss: 4.0867, Train: 1.1789, Val: 1.1797, Test: 1.1932\n",
            "Epoch: 043, Loss: 4.0068, Train: 1.2307, Val: 1.2310, Test: 1.2446\n",
            "Epoch: 044, Loss: 3.9733, Train: 1.2727, Val: 1.2726, Test: 1.2864\n",
            "Epoch: 045, Loss: 3.9741, Train: 1.2903, Val: 1.2898, Test: 1.3037\n",
            "Epoch: 046, Loss: 3.9670, Train: 1.2819, Val: 1.2811, Test: 1.2947\n",
            "Epoch: 047, Loss: 3.9363, Train: 1.2589, Val: 1.2577, Test: 1.2707\n",
            "Epoch: 048, Loss: 3.9061, Train: 1.2375, Val: 1.2362, Test: 1.2485\n",
            "Epoch: 049, Loss: 3.9004, Train: 1.2280, Val: 1.2267, Test: 1.2385\n",
            "Epoch: 050, Loss: 3.9022, Train: 1.2326, Val: 1.2317, Test: 1.2432\n",
            "Epoch: 051, Loss: 3.8800, Train: 1.2496, Val: 1.2495, Test: 1.2608\n",
            "Epoch: 052, Loss: 3.8425, Train: 1.2722, Val: 1.2730, Test: 1.2842\n",
            "Epoch: 053, Loss: 3.8201, Train: 1.2862, Val: 1.2878, Test: 1.2988\n",
            "Epoch: 054, Loss: 3.8086, Train: 1.2798, Val: 1.2824, Test: 1.2927\n",
            "Epoch: 055, Loss: 3.7849, Train: 1.2544, Val: 1.2578, Test: 1.2673\n",
            "Epoch: 056, Loss: 3.7507, Train: 1.2232, Val: 1.2276, Test: 1.2359\n",
            "Epoch: 057, Loss: 3.7264, Train: 1.2008, Val: 1.2063, Test: 1.2136\n",
            "Epoch: 058, Loss: 3.7116, Train: 1.1947, Val: 1.2015, Test: 1.2079\n",
            "Epoch: 059, Loss: 3.6876, Train: 1.2021, Val: 1.2100, Test: 1.2158\n",
            "Epoch: 060, Loss: 3.6600, Train: 1.2134, Val: 1.2223, Test: 1.2276\n",
            "Epoch: 061, Loss: 3.6431, Train: 1.2162, Val: 1.2262, Test: 1.2306\n",
            "Epoch: 062, Loss: 3.6304, Train: 1.2049, Val: 1.2158, Test: 1.2193\n",
            "Epoch: 063, Loss: 3.6118, Train: 1.1835, Val: 1.1955, Test: 1.1978\n",
            "Epoch: 064, Loss: 3.5930, Train: 1.1639, Val: 1.1768, Test: 1.1780\n",
            "Epoch: 065, Loss: 3.5816, Train: 1.1571, Val: 1.1710, Test: 1.1713\n",
            "Epoch: 066, Loss: 3.5688, Train: 1.1630, Val: 1.1776, Test: 1.1775\n",
            "Epoch: 067, Loss: 3.5507, Train: 1.1740, Val: 1.1893, Test: 1.1888\n",
            "Epoch: 068, Loss: 3.5356, Train: 1.1806, Val: 1.1966, Test: 1.1958\n",
            "Epoch: 069, Loss: 3.5227, Train: 1.1767, Val: 1.1935, Test: 1.1922\n",
            "Epoch: 070, Loss: 3.5061, Train: 1.1645, Val: 1.1821, Test: 1.1802\n",
            "Epoch: 071, Loss: 3.4880, Train: 1.1519, Val: 1.1704, Test: 1.1677\n",
            "Epoch: 072, Loss: 3.4735, Train: 1.1474, Val: 1.1668, Test: 1.1635\n",
            "Epoch: 073, Loss: 3.4586, Train: 1.1530, Val: 1.1732, Test: 1.1696\n",
            "Epoch: 074, Loss: 3.4400, Train: 1.1621, Val: 1.1831, Test: 1.1793\n",
            "Epoch: 075, Loss: 3.4232, Train: 1.1664, Val: 1.1883, Test: 1.1841\n",
            "Epoch: 076, Loss: 3.4079, Train: 1.1622, Val: 1.1853, Test: 1.1804\n",
            "Epoch: 077, Loss: 3.3905, Train: 1.1529, Val: 1.1773, Test: 1.1715\n",
            "Epoch: 078, Loss: 3.3736, Train: 1.1467, Val: 1.1725, Test: 1.1658\n",
            "Epoch: 079, Loss: 3.3590, Train: 1.1491, Val: 1.1763, Test: 1.1687\n",
            "Epoch: 080, Loss: 3.3428, Train: 1.1570, Val: 1.1852, Test: 1.1771\n",
            "Epoch: 081, Loss: 3.3274, Train: 1.1612, Val: 1.1906, Test: 1.1819\n",
            "Epoch: 082, Loss: 3.3139, Train: 1.1578, Val: 1.1885, Test: 1.1790\n",
            "Epoch: 083, Loss: 3.2993, Train: 1.1505, Val: 1.1826, Test: 1.1724\n",
            "Epoch: 084, Loss: 3.2864, Train: 1.1483, Val: 1.1816, Test: 1.1708\n",
            "Epoch: 085, Loss: 3.2736, Train: 1.1516, Val: 1.1859, Test: 1.1747\n",
            "Epoch: 086, Loss: 3.2604, Train: 1.1545, Val: 1.1898, Test: 1.1782\n",
            "Epoch: 087, Loss: 3.2485, Train: 1.1519, Val: 1.1883, Test: 1.1763\n",
            "Epoch: 088, Loss: 3.2360, Train: 1.1459, Val: 1.1835, Test: 1.1711\n",
            "Epoch: 089, Loss: 3.2241, Train: 1.1428, Val: 1.1816, Test: 1.1689\n",
            "Epoch: 090, Loss: 3.2129, Train: 1.1457, Val: 1.1853, Test: 1.1726\n",
            "Epoch: 091, Loss: 3.2012, Train: 1.1479, Val: 1.1885, Test: 1.1757\n",
            "Epoch: 092, Loss: 3.1907, Train: 1.1455, Val: 1.1871, Test: 1.1741\n",
            "Epoch: 093, Loss: 3.1803, Train: 1.1400, Val: 1.1825, Test: 1.1695\n",
            "Epoch: 094, Loss: 3.1703, Train: 1.1378, Val: 1.1810, Test: 1.1681\n",
            "Epoch: 095, Loss: 3.1610, Train: 1.1407, Val: 1.1845, Test: 1.1717\n",
            "Epoch: 096, Loss: 3.1516, Train: 1.1414, Val: 1.1857, Test: 1.1730\n",
            "Epoch: 097, Loss: 3.1432, Train: 1.1374, Val: 1.1822, Test: 1.1696\n",
            "Epoch: 098, Loss: 3.1344, Train: 1.1342, Val: 1.1795, Test: 1.1671\n",
            "Epoch: 099, Loss: 3.1262, Train: 1.1364, Val: 1.1820, Test: 1.1696\n",
            "Epoch: 100, Loss: 3.1173, Train: 1.1377, Val: 1.1837, Test: 1.1710\n",
            "Epoch: 101, Loss: 3.1079, Train: 1.1376, Val: 1.1839, Test: 1.1709\n",
            "Epoch: 102, Loss: 3.0998, Train: 1.1393, Val: 1.1857, Test: 1.1725\n",
            "Epoch: 103, Loss: 3.0926, Train: 1.1368, Val: 1.1835, Test: 1.1703\n",
            "Epoch: 104, Loss: 3.0850, Train: 1.1306, Val: 1.1776, Test: 1.1648\n",
            "Epoch: 105, Loss: 3.0779, Train: 1.1310, Val: 1.1783, Test: 1.1656\n",
            "Epoch: 106, Loss: 3.0710, Train: 1.1341, Val: 1.1815, Test: 1.1690\n",
            "Epoch: 107, Loss: 3.0645, Train: 1.1339, Val: 1.1814, Test: 1.1689\n",
            "Epoch: 108, Loss: 3.0582, Train: 1.1331, Val: 1.1806, Test: 1.1681\n",
            "Epoch: 109, Loss: 3.0517, Train: 1.1303, Val: 1.1779, Test: 1.1653\n",
            "Epoch: 110, Loss: 3.0458, Train: 1.1271, Val: 1.1745, Test: 1.1620\n",
            "Epoch: 111, Loss: 3.0406, Train: 1.1297, Val: 1.1768, Test: 1.1643\n",
            "Epoch: 112, Loss: 3.0355, Train: 1.1285, Val: 1.1757, Test: 1.1634\n",
            "Epoch: 113, Loss: 3.0305, Train: 1.1287, Val: 1.1759, Test: 1.1638\n",
            "Epoch: 114, Loss: 3.0260, Train: 1.1300, Val: 1.1771, Test: 1.1652\n",
            "Epoch: 115, Loss: 3.0215, Train: 1.1279, Val: 1.1749, Test: 1.1631\n",
            "Epoch: 116, Loss: 3.0167, Train: 1.1278, Val: 1.1747, Test: 1.1628\n",
            "Epoch: 117, Loss: 3.0123, Train: 1.1240, Val: 1.1710, Test: 1.1592\n",
            "Epoch: 118, Loss: 3.0079, Train: 1.1277, Val: 1.1747, Test: 1.1630\n",
            "Epoch: 119, Loss: 3.0032, Train: 1.1265, Val: 1.1738, Test: 1.1620\n",
            "Epoch: 120, Loss: 2.9986, Train: 1.1252, Val: 1.1729, Test: 1.1610\n",
            "Epoch: 121, Loss: 2.9944, Train: 1.1269, Val: 1.1748, Test: 1.1628\n",
            "Epoch: 122, Loss: 2.9904, Train: 1.1224, Val: 1.1706, Test: 1.1587\n",
            "Epoch: 123, Loss: 2.9861, Train: 1.1254, Val: 1.1738, Test: 1.1616\n",
            "Epoch: 124, Loss: 2.9815, Train: 1.1252, Val: 1.1737, Test: 1.1617\n",
            "Epoch: 125, Loss: 2.9770, Train: 1.1227, Val: 1.1717, Test: 1.1595\n",
            "Epoch: 126, Loss: 2.9724, Train: 1.1226, Val: 1.1719, Test: 1.1596\n",
            "Epoch: 127, Loss: 2.9681, Train: 1.1245, Val: 1.1739, Test: 1.1616\n",
            "Epoch: 128, Loss: 2.9630, Train: 1.1253, Val: 1.1749, Test: 1.1625\n",
            "Epoch: 129, Loss: 2.9584, Train: 1.1205, Val: 1.1705, Test: 1.1582\n",
            "Epoch: 130, Loss: 2.9533, Train: 1.1184, Val: 1.1688, Test: 1.1565\n",
            "Epoch: 131, Loss: 2.9484, Train: 1.1244, Val: 1.1744, Test: 1.1624\n",
            "Epoch: 132, Loss: 2.9436, Train: 1.1223, Val: 1.1726, Test: 1.1608\n",
            "Epoch: 133, Loss: 2.9381, Train: 1.1145, Val: 1.1656, Test: 1.1537\n",
            "Epoch: 134, Loss: 2.9340, Train: 1.1249, Val: 1.1753, Test: 1.1638\n",
            "Epoch: 135, Loss: 2.9298, Train: 1.1159, Val: 1.1671, Test: 1.1554\n",
            "Epoch: 136, Loss: 2.9239, Train: 1.1172, Val: 1.1686, Test: 1.1570\n",
            "Epoch: 137, Loss: 2.9182, Train: 1.1183, Val: 1.1698, Test: 1.1585\n",
            "Epoch: 138, Loss: 2.9135, Train: 1.1157, Val: 1.1676, Test: 1.1564\n",
            "Epoch: 139, Loss: 2.9091, Train: 1.1182, Val: 1.1703, Test: 1.1591\n",
            "Epoch: 140, Loss: 2.9041, Train: 1.1146, Val: 1.1671, Test: 1.1562\n",
            "Epoch: 141, Loss: 2.8994, Train: 1.1145, Val: 1.1671, Test: 1.1563\n",
            "Epoch: 142, Loss: 2.8951, Train: 1.1138, Val: 1.1670, Test: 1.1558\n",
            "Epoch: 143, Loss: 2.8906, Train: 1.1130, Val: 1.1665, Test: 1.1556\n",
            "Epoch: 144, Loss: 2.8861, Train: 1.1178, Val: 1.1711, Test: 1.1605\n",
            "Epoch: 145, Loss: 2.8823, Train: 1.1059, Val: 1.1605, Test: 1.1498\n",
            "Epoch: 146, Loss: 2.8797, Train: 1.1229, Val: 1.1760, Test: 1.1661\n",
            "Epoch: 147, Loss: 2.8770, Train: 1.1054, Val: 1.1600, Test: 1.1500\n",
            "Epoch: 148, Loss: 2.8705, Train: 1.1071, Val: 1.1621, Test: 1.1520\n",
            "Epoch: 149, Loss: 2.8656, Train: 1.1203, Val: 1.1745, Test: 1.1649\n",
            "Epoch: 150, Loss: 2.8633, Train: 1.1039, Val: 1.1597, Test: 1.1501\n",
            "Epoch: 151, Loss: 2.8591, Train: 1.1136, Val: 1.1685, Test: 1.1597\n",
            "Epoch: 152, Loss: 2.8540, Train: 1.1105, Val: 1.1658, Test: 1.1571\n",
            "Epoch: 153, Loss: 2.8495, Train: 1.1023, Val: 1.1584, Test: 1.1500\n",
            "Epoch: 154, Loss: 2.8469, Train: 1.1160, Val: 1.1709, Test: 1.1631\n",
            "Epoch: 155, Loss: 2.8436, Train: 1.1028, Val: 1.1594, Test: 1.1513\n",
            "Epoch: 156, Loss: 2.8383, Train: 1.1058, Val: 1.1626, Test: 1.1546\n",
            "Epoch: 157, Loss: 2.8340, Train: 1.1167, Val: 1.1727, Test: 1.1653\n",
            "Epoch: 158, Loss: 2.8319, Train: 1.0983, Val: 1.1560, Test: 1.1486\n",
            "Epoch: 159, Loss: 2.8294, Train: 1.1133, Val: 1.1699, Test: 1.1627\n",
            "Epoch: 160, Loss: 2.8255, Train: 1.1023, Val: 1.1601, Test: 1.1528\n",
            "Epoch: 161, Loss: 2.8196, Train: 1.0996, Val: 1.1579, Test: 1.1508\n",
            "Epoch: 162, Loss: 2.8171, Train: 1.1171, Val: 1.1741, Test: 1.1673\n",
            "Epoch: 163, Loss: 2.8155, Train: 1.0953, Val: 1.1546, Test: 1.1475\n",
            "Epoch: 164, Loss: 2.8115, Train: 1.1072, Val: 1.1655, Test: 1.1588\n",
            "Epoch: 165, Loss: 2.8062, Train: 1.1058, Val: 1.1643, Test: 1.1579\n",
            "Epoch: 166, Loss: 2.8022, Train: 1.0969, Val: 1.1564, Test: 1.1498\n",
            "Epoch: 167, Loss: 2.8002, Train: 1.1110, Val: 1.1694, Test: 1.1629\n",
            "Epoch: 168, Loss: 2.7980, Train: 1.0938, Val: 1.1540, Test: 1.1474\n",
            "Epoch: 169, Loss: 2.7934, Train: 1.1014, Val: 1.1610, Test: 1.1548\n",
            "Epoch: 170, Loss: 2.7887, Train: 1.1105, Val: 1.1699, Test: 1.1635\n",
            "Epoch: 171, Loss: 2.7869, Train: 1.0898, Val: 1.1514, Test: 1.1448\n",
            "Epoch: 172, Loss: 2.7853, Train: 1.1099, Val: 1.1692, Test: 1.1632\n",
            "Epoch: 173, Loss: 2.7833, Train: 1.0943, Val: 1.1553, Test: 1.1489\n",
            "Epoch: 174, Loss: 2.7768, Train: 1.0950, Val: 1.1568, Test: 1.1501\n",
            "Epoch: 175, Loss: 2.7734, Train: 1.1090, Val: 1.1697, Test: 1.1634\n",
            "Epoch: 176, Loss: 2.7715, Train: 1.0899, Val: 1.1522, Test: 1.1460\n",
            "Epoch: 177, Loss: 2.7686, Train: 1.1030, Val: 1.1645, Test: 1.1582\n",
            "Epoch: 178, Loss: 2.7635, Train: 1.0989, Val: 1.1612, Test: 1.1547\n",
            "Epoch: 179, Loss: 2.7594, Train: 1.0893, Val: 1.1524, Test: 1.1461\n",
            "Epoch: 180, Loss: 2.7575, Train: 1.1045, Val: 1.1662, Test: 1.1602\n",
            "Epoch: 181, Loss: 2.7553, Train: 1.0882, Val: 1.1517, Test: 1.1455\n",
            "Epoch: 182, Loss: 2.7512, Train: 1.0975, Val: 1.1609, Test: 1.1546\n",
            "Epoch: 183, Loss: 2.7465, Train: 1.0997, Val: 1.1631, Test: 1.1570\n",
            "Epoch: 184, Loss: 2.7435, Train: 1.0865, Val: 1.1508, Test: 1.1448\n",
            "Epoch: 185, Loss: 2.7411, Train: 1.1026, Val: 1.1660, Test: 1.1597\n",
            "Epoch: 186, Loss: 2.7389, Train: 1.0852, Val: 1.1502, Test: 1.1442\n",
            "Epoch: 187, Loss: 2.7348, Train: 1.0971, Val: 1.1614, Test: 1.1555\n",
            "Epoch: 188, Loss: 2.7301, Train: 1.0909, Val: 1.1560, Test: 1.1502\n",
            "Epoch: 189, Loss: 2.7260, Train: 1.0909, Val: 1.1562, Test: 1.1504\n",
            "Epoch: 190, Loss: 2.7231, Train: 1.0954, Val: 1.1606, Test: 1.1547\n",
            "Epoch: 191, Loss: 2.7193, Train: 1.0830, Val: 1.1492, Test: 1.1433\n",
            "Epoch: 192, Loss: 2.7168, Train: 1.1002, Val: 1.1654, Test: 1.1591\n",
            "Epoch: 193, Loss: 2.7154, Train: 1.0778, Val: 1.1448, Test: 1.1387\n",
            "Epoch: 194, Loss: 2.7122, Train: 1.1006, Val: 1.1662, Test: 1.1602\n",
            "Epoch: 195, Loss: 2.7095, Train: 1.0778, Val: 1.1451, Test: 1.1397\n",
            "Epoch: 196, Loss: 2.7060, Train: 1.1029, Val: 1.1690, Test: 1.1628\n",
            "Epoch: 197, Loss: 2.7037, Train: 1.0733, Val: 1.1416, Test: 1.1354\n",
            "Epoch: 198, Loss: 2.6985, Train: 1.0940, Val: 1.1610, Test: 1.1553\n",
            "Epoch: 199, Loss: 2.6931, Train: 1.0818, Val: 1.1501, Test: 1.1444\n",
            "Epoch: 200, Loss: 2.6875, Train: 1.0883, Val: 1.1568, Test: 1.1506\n",
            "Epoch: 201, Loss: 2.6838, Train: 1.0827, Val: 1.1515, Test: 1.1455\n",
            "Epoch: 202, Loss: 2.6794, Train: 1.0851, Val: 1.1538, Test: 1.1483\n",
            "Epoch: 203, Loss: 2.6754, Train: 1.0835, Val: 1.1530, Test: 1.1473\n",
            "Epoch: 204, Loss: 2.6720, Train: 1.0784, Val: 1.1484, Test: 1.1425\n",
            "Epoch: 205, Loss: 2.6680, Train: 1.0840, Val: 1.1537, Test: 1.1481\n",
            "Epoch: 206, Loss: 2.6648, Train: 1.0778, Val: 1.1482, Test: 1.1429\n",
            "Epoch: 207, Loss: 2.6618, Train: 1.0928, Val: 1.1631, Test: 1.1571\n",
            "Epoch: 208, Loss: 2.6627, Train: 1.0572, Val: 1.1291, Test: 1.1243\n",
            "Epoch: 209, Loss: 2.6775, Train: 1.1326, Val: 1.1995, Test: 1.1940\n",
            "Epoch: 210, Loss: 2.7101, Train: 1.0368, Val: 1.1104, Test: 1.1051\n",
            "Epoch: 211, Loss: 2.7218, Train: 1.1110, Val: 1.1797, Test: 1.1748\n",
            "Epoch: 212, Loss: 2.6667, Train: 1.0950, Val: 1.1664, Test: 1.1605\n",
            "Epoch: 213, Loss: 2.6554, Train: 1.0360, Val: 1.1110, Test: 1.1062\n",
            "Epoch: 214, Loss: 2.7045, Train: 1.1347, Val: 1.2014, Test: 1.1979\n",
            "Epoch: 215, Loss: 2.6922, Train: 1.0834, Val: 1.1535, Test: 1.1493\n",
            "Epoch: 216, Loss: 2.6409, Train: 1.0358, Val: 1.1107, Test: 1.1051\n",
            "Epoch: 217, Loss: 2.6850, Train: 1.1151, Val: 1.1858, Test: 1.1803\n",
            "Epoch: 218, Loss: 2.6670, Train: 1.0837, Val: 1.1555, Test: 1.1516\n",
            "Epoch: 219, Loss: 2.6444, Train: 1.0523, Val: 1.1268, Test: 1.1226\n",
            "Epoch: 220, Loss: 2.6510, Train: 1.0992, Val: 1.1714, Test: 1.1666\n",
            "Epoch: 221, Loss: 2.6481, Train: 1.0682, Val: 1.1419, Test: 1.1372\n",
            "Epoch: 222, Loss: 2.6181, Train: 1.0581, Val: 1.1325, Test: 1.1277\n",
            "Epoch: 223, Loss: 2.6305, Train: 1.0931, Val: 1.1666, Test: 1.1612\n",
            "Epoch: 224, Loss: 2.6163, Train: 1.0759, Val: 1.1525, Test: 1.1464\n",
            "Epoch: 225, Loss: 2.6122, Train: 1.0489, Val: 1.1273, Test: 1.1210\n",
            "Epoch: 226, Loss: 2.6192, Train: 1.0931, Val: 1.1676, Test: 1.1624\n",
            "Epoch: 227, Loss: 2.6070, Train: 1.0821, Val: 1.1575, Test: 1.1521\n",
            "Epoch: 228, Loss: 2.5985, Train: 1.0487, Val: 1.1276, Test: 1.1215\n",
            "Epoch: 229, Loss: 2.6058, Train: 1.0803, Val: 1.1578, Test: 1.1521\n",
            "Epoch: 230, Loss: 2.5937, Train: 1.0775, Val: 1.1546, Test: 1.1494\n",
            "Epoch: 231, Loss: 2.5871, Train: 1.0585, Val: 1.1375, Test: 1.1319\n",
            "Epoch: 232, Loss: 2.5880, Train: 1.0774, Val: 1.1565, Test: 1.1507\n",
            "Epoch: 233, Loss: 2.5847, Train: 1.0608, Val: 1.1404, Test: 1.1346\n",
            "Epoch: 234, Loss: 2.5741, Train: 1.0634, Val: 1.1426, Test: 1.1368\n",
            "Epoch: 235, Loss: 2.5744, Train: 1.0787, Val: 1.1580, Test: 1.1526\n",
            "Epoch: 236, Loss: 2.5704, Train: 1.0543, Val: 1.1360, Test: 1.1303\n",
            "Epoch: 237, Loss: 2.5660, Train: 1.0651, Val: 1.1462, Test: 1.1407\n",
            "Epoch: 238, Loss: 2.5596, Train: 1.0745, Val: 1.1550, Test: 1.1496\n",
            "Epoch: 239, Loss: 2.5582, Train: 1.0512, Val: 1.1336, Test: 1.1275\n",
            "Epoch: 240, Loss: 2.5553, Train: 1.0661, Val: 1.1482, Test: 1.1424\n",
            "Epoch: 241, Loss: 2.5514, Train: 1.0621, Val: 1.1443, Test: 1.1382\n",
            "Epoch: 242, Loss: 2.5464, Train: 1.0615, Val: 1.1450, Test: 1.1384\n",
            "Epoch: 243, Loss: 2.5415, Train: 1.0621, Val: 1.1465, Test: 1.1396\n",
            "Epoch: 244, Loss: 2.5402, Train: 1.0540, Val: 1.1385, Test: 1.1313\n",
            "Epoch: 245, Loss: 2.5373, Train: 1.0710, Val: 1.1547, Test: 1.1481\n",
            "Epoch: 246, Loss: 2.5348, Train: 1.0469, Val: 1.1326, Test: 1.1255\n",
            "Epoch: 247, Loss: 2.5314, Train: 1.0659, Val: 1.1510, Test: 1.1443\n",
            "Epoch: 248, Loss: 2.5262, Train: 1.0553, Val: 1.1413, Test: 1.1342\n",
            "Epoch: 249, Loss: 2.5229, Train: 1.0591, Val: 1.1456, Test: 1.1388\n",
            "Epoch: 250, Loss: 2.5184, Train: 1.0521, Val: 1.1389, Test: 1.1320\n",
            "Epoch: 251, Loss: 2.5146, Train: 1.0614, Val: 1.1477, Test: 1.1412\n",
            "Epoch: 252, Loss: 2.5115, Train: 1.0492, Val: 1.1372, Test: 1.1301\n",
            "Epoch: 253, Loss: 2.5079, Train: 1.0588, Val: 1.1465, Test: 1.1397\n",
            "Epoch: 254, Loss: 2.5053, Train: 1.0475, Val: 1.1358, Test: 1.1285\n",
            "Epoch: 255, Loss: 2.5061, Train: 1.0708, Val: 1.1583, Test: 1.1517\n",
            "Epoch: 256, Loss: 2.5136, Train: 1.0247, Val: 1.1139, Test: 1.1060\n",
            "Epoch: 257, Loss: 2.5499, Train: 1.1183, Val: 1.2026, Test: 1.1960\n",
            "Epoch: 258, Loss: 2.5837, Train: 1.0104, Val: 1.1004, Test: 1.0915\n",
            "Epoch: 259, Loss: 2.6154, Train: 1.0870, Val: 1.1730, Test: 1.1665\n",
            "Epoch: 260, Loss: 2.5171, Train: 1.0601, Val: 1.1486, Test: 1.1417\n",
            "Epoch: 261, Loss: 2.4997, Train: 1.0080, Val: 1.0986, Test: 1.0902\n",
            "Epoch: 262, Loss: 2.5727, Train: 1.1123, Val: 1.1963, Test: 1.1907\n",
            "Epoch: 263, Loss: 2.5463, Train: 1.0495, Val: 1.1388, Test: 1.1315\n",
            "Epoch: 264, Loss: 2.4794, Train: 1.0079, Val: 1.1002, Test: 1.0915\n",
            "Epoch: 265, Loss: 2.5380, Train: 1.1083, Val: 1.1937, Test: 1.1873\n",
            "Epoch: 266, Loss: 2.5441, Train: 1.0483, Val: 1.1361, Test: 1.1284\n",
            "Epoch: 267, Loss: 2.5021, Train: 1.0164, Val: 1.1082, Test: 1.0994\n",
            "Epoch: 268, Loss: 2.4992, Train: 1.0932, Val: 1.1813, Test: 1.1732\n",
            "Epoch: 269, Loss: 2.5636, Train: 1.0342, Val: 1.1235, Test: 1.1145\n",
            "Epoch: 270, Loss: 2.5327, Train: 1.0453, Val: 1.1339, Test: 1.1253\n",
            "Epoch: 271, Loss: 2.5064, Train: 1.0866, Val: 1.1752, Test: 1.1676\n",
            "Epoch: 272, Loss: 2.5356, Train: 1.0156, Val: 1.1087, Test: 1.0999\n",
            "Epoch: 273, Loss: 2.4851, Train: 1.0469, Val: 1.1366, Test: 1.1281\n",
            "Epoch: 274, Loss: 2.4724, Train: 1.0893, Val: 1.1769, Test: 1.1694\n",
            "Epoch: 275, Loss: 2.4846, Train: 1.0261, Val: 1.1201, Test: 1.1116\n",
            "Epoch: 276, Loss: 2.4659, Train: 1.0247, Val: 1.1186, Test: 1.1102\n",
            "Epoch: 277, Loss: 2.4581, Train: 1.0829, Val: 1.1713, Test: 1.1640\n",
            "Epoch: 278, Loss: 2.4709, Train: 1.0432, Val: 1.1349, Test: 1.1265\n",
            "Epoch: 279, Loss: 2.4466, Train: 1.0177, Val: 1.1129, Test: 1.1042\n",
            "Epoch: 280, Loss: 2.4539, Train: 1.0625, Val: 1.1550, Test: 1.1470\n",
            "Epoch: 281, Loss: 2.4540, Train: 1.0424, Val: 1.1355, Test: 1.1263\n",
            "Epoch: 282, Loss: 2.4407, Train: 1.0279, Val: 1.1224, Test: 1.1131\n",
            "Epoch: 283, Loss: 2.4395, Train: 1.0564, Val: 1.1496, Test: 1.1420\n",
            "Epoch: 284, Loss: 2.4437, Train: 1.0292, Val: 1.1246, Test: 1.1161\n",
            "Epoch: 285, Loss: 2.4241, Train: 1.0334, Val: 1.1285, Test: 1.1191\n",
            "Epoch: 286, Loss: 2.4280, Train: 1.0589, Val: 1.1530, Test: 1.1444\n",
            "Epoch: 287, Loss: 2.4244, Train: 1.0261, Val: 1.1232, Test: 1.1143\n",
            "Epoch: 288, Loss: 2.4184, Train: 1.0270, Val: 1.1238, Test: 1.1148\n",
            "Epoch: 289, Loss: 2.4116, Train: 1.0576, Val: 1.1519, Test: 1.1435\n",
            "Epoch: 290, Loss: 2.4166, Train: 1.0275, Val: 1.1247, Test: 1.1159\n",
            "Epoch: 291, Loss: 2.4052, Train: 1.0253, Val: 1.1234, Test: 1.1147\n",
            "Epoch: 292, Loss: 2.4047, Train: 1.0477, Val: 1.1443, Test: 1.1356\n",
            "Epoch: 293, Loss: 2.4022, Train: 1.0281, Val: 1.1261, Test: 1.1166\n",
            "Epoch: 294, Loss: 2.4004, Train: 1.0307, Val: 1.1289, Test: 1.1206\n",
            "Epoch: 295, Loss: 2.3929, Train: 1.0371, Val: 1.1350, Test: 1.1272\n",
            "Epoch: 296, Loss: 2.3926, Train: 1.0235, Val: 1.1226, Test: 1.1135\n",
            "Epoch: 297, Loss: 2.3922, Train: 1.0413, Val: 1.1398, Test: 1.1313\n",
            "Epoch: 298, Loss: 2.3862, Train: 1.0259, Val: 1.1261, Test: 1.1175\n",
            "Epoch: 299, Loss: 2.3832, Train: 1.0229, Val: 1.1233, Test: 1.1144\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "user_id = 0\n",
        "num_movies = len(movie_mapping)\n",
        "\n",
        "row = torch.tensor([user_id] * num_movies)\n",
        "col = torch.arange(num_movies)\n",
        "edge_label_index = torch.stack([row, col], dim=0)\n",
        "\n",
        "pred = model(data.x_dict, data.edge_index_dict,\n",
        "              edge_label_index)\n",
        "pred = pred.clamp(min=0, max=5)"
      ],
      "metadata": {
        "id": "MfxjlfZuCSPZ"
      },
      "execution_count": 50,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "reverse_movie_mapping = dict(zip(movie_mapping.values(),movie_mapping.keys()))\n",
        "\n",
        "for i,j in enumerate(pred):\n",
        "  score = j.detach().numpy()\n",
        "  if score > 4:\n",
        "    movie_id = reverse_movie_mapping[i]\n",
        "    print(score, movie_id)"
      ],
      "metadata": {
        "id": "wn7KitYqgZx2",
        "outputId": "4902ede0-025c-4ee5-aa9f-ca25b037ef08",
        "colab": {
          "base_uri": "https://localhost:8080/"
        }
      },
      "execution_count": 60,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "4.0598617 110352\n",
            "4.0120296 108979\n",
            "4.1322947 62331\n",
            "4.185063 64695\n",
            "4.194876 71433\n",
            "4.548223 34435\n",
            "4.172068 26265\n",
            "4.475247 31923\n",
            "4.508934 8341\n",
            "4.0386944 8591\n",
            "4.7643223 8537\n",
            "5.0 7087\n",
            "4.2244244 7700\n",
            "4.04606 7771\n",
            "4.1852484 7840\n",
            "4.4977345 5300\n",
            "4.2502284 5069\n",
            "4.2183247 4688\n",
            "4.347323 4933\n",
            "4.8019185 4567\n",
            "4.5926213 4591\n",
            "4.1736975 3926\n",
            "4.191746 2893\n",
            "4.5188985 2880\n",
            "4.401515 2625\n",
            "4.417151 107559\n",
            "4.054242 106762\n",
            "5.0 83603\n",
            "5.0 72612\n",
            "4.0232973 66335\n",
            "4.2175603 67429\n",
            "4.2942324 31184\n",
            "4.03687 6713\n",
            "4.6466603 2810\n",
            "4.0433006 110461\n",
            "4.1122184 8761\n",
            "4.749738 755\n",
            "5.0 114265\n",
            "5.0 110645\n",
            "4.23756 109483\n",
            "4.144233 108514\n",
            "4.9258256 143859\n",
            "4.746245 148956\n",
            "4.2014356 141886\n",
            "4.0177484 129313\n",
            "4.3455296 128846\n",
            "5.0 82931\n",
            "4.3412914 80717\n",
            "4.3096323 79469\n",
            "4.4034314 77907\n",
            "4.2927575 72781\n",
            "4.4668264 99030\n",
            "4.025861 93855\n",
            "4.258225 91947\n",
            "5.0 91690\n",
            "4.854241 90061\n",
            "4.061634 60128\n",
            "4.9872293 61361\n",
            "4.0194736 64321\n",
            "4.299729 64660\n",
            "4.4593635 70201\n",
            "4.393086 48817\n",
            "4.2616 49013\n",
            "4.309868 52668\n",
            "5.0 54328\n",
            "4.5755296 55566\n",
            "4.885948 39416\n",
            "4.2300286 45000\n",
            "4.3147917 48560\n",
            "4.55891 47148\n",
            "4.2353196 26422\n",
            "4.007225 27685\n",
            "4.0723414 26788\n",
            "4.2357454 30883\n",
            "4.2257905 32591\n",
            "4.068458 32469\n",
            "4.1223893 8370\n",
            "4.193528 7888\n",
            "5.0 8989\n",
            "4.162474 25947\n",
            "4.291541 8689\n",
            "4.235349 8933\n",
            "4.074726 8872\n",
            "4.1171055 7073\n",
            "5.0 7086\n",
            "4.8135266 7832\n",
            "4.1101336 7207\n",
            "5.0 7352\n",
            "4.5494504 6528\n",
            "4.0897303 6404\n",
            "5.0 5243\n",
            "4.005517 5768\n",
            "4.014658 5525\n",
            "4.2398844 5602\n",
            "4.86654 5428\n",
            "4.1514773 5427\n",
            "5.0 4687\n",
            "4.4492483 4717\n",
            "4.0788565 4755\n",
            "5.0 4796\n",
            "4.835806 4493\n",
            "5.0 4600\n",
            "4.6683583 4584\n",
            "4.119969 3905\n",
            "5.0 4076\n",
            "4.2068076 3640\n",
            "4.1191955 3819\n",
            "4.0027814 2342\n",
            "4.0980983 1034\n",
            "5.0 1067\n",
            "4.0330157 946\n",
            "5.0 876\n",
            "5.0 617\n",
            "4.178244 582\n",
            "4.7762446 84\n",
            "4.15123 5182\n",
            "4.228307 102753\n",
            "4.4326005 85612\n",
            "4.203603 96911\n",
            "4.925684 64114\n",
            "5.0 65193\n",
            "4.797875 51471\n",
            "4.124559 26394\n",
            "4.129678 8459\n",
            "5.0 6883\n",
            "5.0 6722\n",
            "4.3460703 5826\n",
            "4.873979 4539\n",
            "4.6802783 3855\n",
            "4.4103546 2388\n",
            "4.4138627 764\n",
            "4.0038943 389\n",
            "4.7080083 301\n",
            "4.759151 112070\n",
            "4.993941 111913\n",
            "5.0 109249\n",
            "4.3541694 106452\n",
            "4.349671 104590\n",
            "4.413786 104303\n",
            "4.1313047 141890\n",
            "4.6775126 129653\n",
            "4.261273 131168\n",
            "4.7592225 130580\n",
            "4.44374 84116\n",
            "4.001243 77810\n",
            "4.043429 76091\n",
            "5.0 78903\n",
            "4.174827 72683\n",
            "4.594711 96373\n",
            "4.0997043 97826\n",
            "4.368005 98126\n",
            "4.0697637 90947\n",
            "4.376368 67957\n",
            "4.2364974 69761\n",
            "4.0397406 70678\n",
            "4.392351 71755\n",
            "4.4869046 50160\n",
            "4.4034095 52528\n",
            "4.564345 55417\n",
            "4.308442 55684\n",
            "4.3824525 58706\n",
            "4.6253257 37739\n",
            "4.820244 41336\n",
            "4.623121 40870\n",
            "5.0 45172\n",
            "4.2170873 26462\n",
            "4.1095433 26176\n",
            "5.0 26151\n",
            "5.0 26346\n",
            "5.0 27397\n",
            "4.3321533 27724\n",
            "5.0 26695\n",
            "4.5458255 27834\n",
            "4.8218255 8484\n",
            "4.118229 7947\n",
            "4.349904 7936\n",
            "4.6668296 7919\n",
            "5.0 8208\n",
            "4.054771 25755\n",
            "4.4685874 8951\n",
            "5.0 8730\n",
            "4.173735 8751\n",
            "4.669378 8899\n",
            "4.1092515 7040\n",
            "4.487986 7069\n",
            "5.0 7152\n",
            "5.0 7002\n",
            "4.3013964 7624\n",
            "4.0638304 7216\n",
            "4.352197 7238\n",
            "4.3686323 6114\n",
            "4.4672065 5986\n",
            "4.186474 5888\n",
            "4.16177 6786\n",
            "4.135402 6757\n",
            "4.3913507 6530\n",
            "4.2176175 5114\n",
            "5.0 5498\n",
            "4.3308625 4842\n",
            "4.0051556 4273\n",
            "4.1471105 4114\n",
            "4.868486 4095\n",
            "4.4818587 3746\n",
            "4.6795735 3202\n",
            "4.079862 3116\n",
            "4.785973 2758\n",
            "4.011664 2630\n",
            "4.383509 2207\n",
            "4.8761067 2284\n",
            "4.1860814 1933\n",
            "4.1190734 1609\n",
            "4.95842 1563\n",
            "4.157276 1572\n",
            "4.5282183 1575\n",
            "5.0 1428\n",
            "4.8305573 1164\n",
            "4.358039 1040\n",
            "5.0 845\n",
            "4.5444407 816\n",
            "4.02931 854\n",
            "4.0733695 549\n",
            "4.585633 245\n",
            "5.0 8142\n",
            "4.52484 7827\n",
            "5.0 401\n",
            "4.1397176 79677\n",
            "4.0353675 47728\n",
            "4.1466045 8765\n",
            "4.527819 4645\n",
            "4.093639 4957\n",
            "4.3570647 2511\n",
            "4.439151 1472\n",
            "4.1166463 107649\n",
            "4.0010843 44731\n",
            "4.3249645 25852\n",
            "4.7067113 7116\n",
            "4.7087255 5062\n",
            "4.031942 3892\n",
            "4.852548 1328\n",
            "4.213572 109205\n",
            "4.163547 105429\n",
            "4.315216 103444\n",
            "5.0 100553\n",
            "4.098902 101106\n",
            "5.0 100034\n",
            "4.2128906 99795\n",
            "4.2250147 139116\n",
            "4.2856045 83293\n",
            "4.2887 86504\n",
            "4.2730412 85394\n",
            "4.034184 80599\n",
            "4.3941326 79006\n",
            "4.3401628 96563\n",
            "4.72253 96901\n",
            "4.330721 97639\n",
            "4.1558304 97817\n",
            "4.0523887 93700\n",
            "4.1548944 93040\n",
            "4.4436684 60291\n",
            "4.1893272 65638\n",
            "4.4723644 68536\n",
            "4.395187 67504\n",
            "4.2850013 69529\n",
            "4.727995 71640\n",
            "4.68157 71390\n",
            "4.293312 53024\n",
            "4.2366767 53883\n",
            "4.212692 55687\n",
            "4.0573854 58889\n",
            "4.0035114 34153\n",
            "4.489734 45987\n",
            "4.222896 44511\n",
            "4.6867933 26797\n",
            "4.8097425 4459\n",
            "4.3138885 4515\n",
            "5.0 2984\n",
            "4.0028796 1652\n",
            "4.1043267 160590\n",
            "4.6860905 149532\n",
            "4.8634105 151307\n",
            "4.065282 140753\n",
            "4.1342134 128616\n",
            "4.703014 128620\n",
            "4.956445 122888\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        ""
      ],
      "metadata": {
        "id": "Rmrr3WNTgh8T"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}