[fd8900]: / matching / clinical_trial_matching.ipynb

Download this file

9698 lines (9698 with data), 529.8 kB

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "gpuType": "T4"
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU",
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "0044cf6034584371976bd34c2636df38": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_cc3a17ba703a4ffab2d3dcc814116cfd",
              "IPY_MODEL_625520fa23bb42f2b7ceb279a080144f",
              "IPY_MODEL_13eed3b66d424f8787f0bba1fd4ab7f9"
            ],
            "layout": "IPY_MODEL_e689e8b6d19341068a5ce1ecb9679edb"
          }
        },
        "cc3a17ba703a4ffab2d3dcc814116cfd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_205880e57da74abda5cbfd2930d6599f",
            "placeholder": "​",
            "style": "IPY_MODEL_df366d4d07d04554a4acc1d2531e92a5",
            "value": "Map: 100%"
          }
        },
        "625520fa23bb42f2b7ceb279a080144f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_50aac9cafdb84507af0c862c893271df",
            "max": 3633,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_012e679ae0064ca694bd267db13ee556",
            "value": 3633
          }
        },
        "13eed3b66d424f8787f0bba1fd4ab7f9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_3a0018a681ac449695c4b1206616e92c",
            "placeholder": "​",
            "style": "IPY_MODEL_2aa4e5b98dc34a73a4f085d27701b765",
            "value": " 3633/3633 [00:00<00:00, 4317.31 examples/s]"
          }
        },
        "e689e8b6d19341068a5ce1ecb9679edb": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "205880e57da74abda5cbfd2930d6599f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "df366d4d07d04554a4acc1d2531e92a5": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "50aac9cafdb84507af0c862c893271df": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "012e679ae0064ca694bd267db13ee556": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "3a0018a681ac449695c4b1206616e92c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "2aa4e5b98dc34a73a4f085d27701b765": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "4fcef18ad43a43af95a91441bb1a1dab": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_9f86e1dacbda4414aaeb65ea8a915a5c",
              "IPY_MODEL_5b5348e7cdd141bf847797aed7877e6d",
              "IPY_MODEL_601021e6b3964ebca79882456a04ac77"
            ],
            "layout": "IPY_MODEL_adf1a655c2c947c4bc782e39868828ba"
          }
        },
        "9f86e1dacbda4414aaeb65ea8a915a5c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_7b337d5fd00b42fc95fc741771b837a4",
            "placeholder": "​",
            "style": "IPY_MODEL_21b26aa6be6646098d05e273e416ba2d",
            "value": "Map: 100%"
          }
        },
        "5b5348e7cdd141bf847797aed7877e6d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_0a35a1a144fe42b7b75301c67e183da5",
            "max": 454,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_a11bd4b15170478897fdb502ccc09442",
            "value": 454
          }
        },
        "601021e6b3964ebca79882456a04ac77": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_002f1e1421eb483f82717dec3cc28fab",
            "placeholder": "​",
            "style": "IPY_MODEL_36d2f410a1744578bdd59da554ce740e",
            "value": " 454/454 [00:00<00:00, 2303.86 examples/s]"
          }
        },
        "adf1a655c2c947c4bc782e39868828ba": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "7b337d5fd00b42fc95fc741771b837a4": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "21b26aa6be6646098d05e273e416ba2d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "0a35a1a144fe42b7b75301c67e183da5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "a11bd4b15170478897fdb502ccc09442": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "002f1e1421eb483f82717dec3cc28fab": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "36d2f410a1744578bdd59da554ce740e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "5f2b45a0d8cd4713ab15b8f67e28c756": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_10a7beabd2204ea2942981f030deee63",
              "IPY_MODEL_94efefbead3d4b2d9f1079472468601e",
              "IPY_MODEL_f2d81f219b494caca47ef7d22e3a2117"
            ],
            "layout": "IPY_MODEL_9640260a88da41f5b5b2d669afe344e2"
          }
        },
        "10a7beabd2204ea2942981f030deee63": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_dc42502476d746e7adde1a4f5d229bbf",
            "placeholder": "​",
            "style": "IPY_MODEL_6047621b4fd74300a3cd97278dd0a880",
            "value": "Map: 100%"
          }
        },
        "94efefbead3d4b2d9f1079472468601e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_60916d1e3fe541dc85f4126cab2ddb61",
            "max": 455,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_f6367a21b26f4cdb926d53870aee0e18",
            "value": 455
          }
        },
        "f2d81f219b494caca47ef7d22e3a2117": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_1d535770898f45feaf57f95aef8af501",
            "placeholder": "​",
            "style": "IPY_MODEL_cc60e996b5d94979816a923a29045ec8",
            "value": " 455/455 [00:00<00:00, 2820.02 examples/s]"
          }
        },
        "9640260a88da41f5b5b2d669afe344e2": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "dc42502476d746e7adde1a4f5d229bbf": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6047621b4fd74300a3cd97278dd0a880": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "60916d1e3fe541dc85f4126cab2ddb61": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "f6367a21b26f4cdb926d53870aee0e18": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "1d535770898f45feaf57f95aef8af501": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "cc60e996b5d94979816a923a29045ec8": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "2adbe4d14ed448a0876bd84baafcd69d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_5f0217de9a9b4c6e835b381486ccde2d",
              "IPY_MODEL_d4d69709633c4a19acf28f67f962e474",
              "IPY_MODEL_0ee115186e5649528ff3a5605192bc7f"
            ],
            "layout": "IPY_MODEL_0b6714ad75dd4357a8a04255623b72e5"
          }
        },
        "5f0217de9a9b4c6e835b381486ccde2d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_6ef98027aed44d6ba192340ecec65afc",
            "placeholder": "​",
            "style": "IPY_MODEL_c6e3c22c52484dc78e4adda75f6e2dba",
            "value": "Downloading (…)okenizer_config.json: 100%"
          }
        },
        "d4d69709633c4a19acf28f67f962e474": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_200be19af1914804a4e5bbfbe909fa98",
            "max": 388,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_da68447cdf9240ce8a00776c7b7e7903",
            "value": 388
          }
        },
        "0ee115186e5649528ff3a5605192bc7f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_bf79122cfd2542d6b07c309df474abf9",
            "placeholder": "​",
            "style": "IPY_MODEL_2bd15504a02b4fc7b674ca3edebaa0df",
            "value": " 388/388 [00:00<00:00, 19.8kB/s]"
          }
        },
        "0b6714ad75dd4357a8a04255623b72e5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6ef98027aed44d6ba192340ecec65afc": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "c6e3c22c52484dc78e4adda75f6e2dba": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "200be19af1914804a4e5bbfbe909fa98": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "da68447cdf9240ce8a00776c7b7e7903": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "bf79122cfd2542d6b07c309df474abf9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "2bd15504a02b4fc7b674ca3edebaa0df": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d9ca7cbc1cbb49e398e615693dca2101": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_2d5a80732fd94d4bb5715c621be15a79",
              "IPY_MODEL_b4e91a1487224a3787c8c4b7d8ef9d66",
              "IPY_MODEL_992b3db6422545b0b778faba8b699a9b"
            ],
            "layout": "IPY_MODEL_18fa4473f5eb4679bef7f63ee10d4817"
          }
        },
        "2d5a80732fd94d4bb5715c621be15a79": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_0853b43c5a3e437395d0af21616ba644",
            "placeholder": "​",
            "style": "IPY_MODEL_145fc556fd0e4853948908dabf03ee22",
            "value": "Downloading (…)solve/main/vocab.txt: 100%"
          }
        },
        "b4e91a1487224a3787c8c4b7d8ef9d66": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_d93be8f2eb9846e1afecfc0bd00c8839",
            "max": 226150,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_bc96ccb3f010463facae4a9589850b65",
            "value": 226150
          }
        },
        "992b3db6422545b0b778faba8b699a9b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_1c51d8d7cf5b4a7dbec9a9e4e37668de",
            "placeholder": "​",
            "style": "IPY_MODEL_5ad1872df2bb4f6eb4fd476c1aa16122",
            "value": " 226k/226k [00:00<00:00, 3.31MB/s]"
          }
        },
        "18fa4473f5eb4679bef7f63ee10d4817": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "0853b43c5a3e437395d0af21616ba644": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "145fc556fd0e4853948908dabf03ee22": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d93be8f2eb9846e1afecfc0bd00c8839": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "bc96ccb3f010463facae4a9589850b65": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "1c51d8d7cf5b4a7dbec9a9e4e37668de": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "5ad1872df2bb4f6eb4fd476c1aa16122": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "c13124e6c8d14f319b372ff1621ea883": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_a21b9c1a85574673a8fd0f5dbd111f15",
              "IPY_MODEL_0d4b2b50d7cb4467988f3375b66c192c",
              "IPY_MODEL_7aa5fdb782a44a22b6d2219e5bcff354"
            ],
            "layout": "IPY_MODEL_093b83161a8448dd8f0f4c3da3bbbe8f"
          }
        },
        "a21b9c1a85574673a8fd0f5dbd111f15": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_29ca8221217748e68ce1588ee3f637a9",
            "placeholder": "​",
            "style": "IPY_MODEL_9c4256f00a6a4a8a85e20ddb53647424",
            "value": "Downloading (…)/main/tokenizer.json: 100%"
          }
        },
        "0d4b2b50d7cb4467988f3375b66c192c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_47af947481bc48aeb6ca2e0dcfdf6312",
            "max": 460876,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_4725600d7cb746589a140c1fe68bc04b",
            "value": 460876
          }
        },
        "7aa5fdb782a44a22b6d2219e5bcff354": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_9a0d5627edb5415c827ffd4ffdcb5a08",
            "placeholder": "​",
            "style": "IPY_MODEL_25d5b69cf04649aca3341f92b3327402",
            "value": " 461k/461k [00:00<00:00, 3.43MB/s]"
          }
        },
        "093b83161a8448dd8f0f4c3da3bbbe8f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "29ca8221217748e68ce1588ee3f637a9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "9c4256f00a6a4a8a85e20ddb53647424": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "47af947481bc48aeb6ca2e0dcfdf6312": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "4725600d7cb746589a140c1fe68bc04b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "9a0d5627edb5415c827ffd4ffdcb5a08": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "25d5b69cf04649aca3341f92b3327402": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "80a3951b0e644e0396a0f3322fc20cd3": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_1ee02de646534e77b681b2e3af2a8206",
              "IPY_MODEL_7a2139c45fc74306ad140dc65ca5b992",
              "IPY_MODEL_4016fbd574374622993d7b5ba90fe958"
            ],
            "layout": "IPY_MODEL_9aa801fa467745edb38c140436395158"
          }
        },
        "1ee02de646534e77b681b2e3af2a8206": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_8cc21f9f17de41bbb2a1bb79f16fadc8",
            "placeholder": "​",
            "style": "IPY_MODEL_72d302db18e74221be6c3518746ef20a",
            "value": "Downloading (…)cial_tokens_map.json: 100%"
          }
        },
        "7a2139c45fc74306ad140dc65ca5b992": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_dee076e04d0344a5a4436e925d58b797",
            "max": 112,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_d94934f5e2ac416494c7c85a089fdb7a",
            "value": 112
          }
        },
        "4016fbd574374622993d7b5ba90fe958": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_f29e903fc6c24ca9ae2dcb1b84fb68f0",
            "placeholder": "​",
            "style": "IPY_MODEL_4f58dab11b364d2c8da3e4a3cdf1b266",
            "value": " 112/112 [00:00<00:00, 8.87kB/s]"
          }
        },
        "9aa801fa467745edb38c140436395158": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "8cc21f9f17de41bbb2a1bb79f16fadc8": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "72d302db18e74221be6c3518746ef20a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "dee076e04d0344a5a4436e925d58b797": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "d94934f5e2ac416494c7c85a089fdb7a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "f29e903fc6c24ca9ae2dcb1b84fb68f0": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "4f58dab11b364d2c8da3e4a3cdf1b266": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "2cdc68bdf3ff47a8b0e29b94f5bd236b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_bb49a02a3ed5496cb8251cfc0f06ea1e",
              "IPY_MODEL_6bb13ea7aac84d399e34c8ed816d0c5b",
              "IPY_MODEL_668e815446ef483ea4839a8a8cc52937"
            ],
            "layout": "IPY_MODEL_d8e89bc18f284b43ab1261131ea2fdd8"
          }
        },
        "bb49a02a3ed5496cb8251cfc0f06ea1e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_e4f58c83b5f4402687aa8fd5e0d40409",
            "placeholder": "​",
            "style": "IPY_MODEL_4d11321f627b472f9862c719d0e184c0",
            "value": "Downloading (…)lve/main/config.json: 100%"
          }
        },
        "6bb13ea7aac84d399e34c8ed816d0c5b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_f431c96e37ea48f398a28be37e644c3c",
            "max": 667,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_cd3fee27ee5e48a68fffb8c343eb305e",
            "value": 667
          }
        },
        "668e815446ef483ea4839a8a8cc52937": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_843197ef31dd477d8a41876f4b212dec",
            "placeholder": "​",
            "style": "IPY_MODEL_c9199f83ce9444d09f8a9782aa0a99c4",
            "value": " 667/667 [00:00<00:00, 44.2kB/s]"
          }
        },
        "d8e89bc18f284b43ab1261131ea2fdd8": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "e4f58c83b5f4402687aa8fd5e0d40409": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "4d11321f627b472f9862c719d0e184c0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "f431c96e37ea48f398a28be37e644c3c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "cd3fee27ee5e48a68fffb8c343eb305e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "843197ef31dd477d8a41876f4b212dec": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "c9199f83ce9444d09f8a9782aa0a99c4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d1ab6716f79745b28c1e2acb3fa2a727": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_46e5c27000664e9299edea05c0f4b1a7",
              "IPY_MODEL_18f2040a12eb4e47a09b52a0b5dd9503",
              "IPY_MODEL_5501a73549e84c9b8c519934c13db6a2"
            ],
            "layout": "IPY_MODEL_74c25ffb54b64cee8d77f7752c2afa83"
          }
        },
        "46e5c27000664e9299edea05c0f4b1a7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_55580def279f4c6dabf56e2c01fd162a",
            "placeholder": "​",
            "style": "IPY_MODEL_80410cabadd24cf398a027fc19c6655a",
            "value": "Downloading pytorch_model.bin: 100%"
          }
        },
        "18f2040a12eb4e47a09b52a0b5dd9503": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_36333367d23f4d348836017cf842f76a",
            "max": 438010289,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_1ee7e50769c843eabe1b82c9749d2115",
            "value": 438010289
          }
        },
        "5501a73549e84c9b8c519934c13db6a2": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_b695b93e8a664e74abb36fd3074c1d57",
            "placeholder": "​",
            "style": "IPY_MODEL_17cf40348c0849a7a10b0c852a751600",
            "value": " 438M/438M [00:04<00:00, 76.4MB/s]"
          }
        },
        "74c25ffb54b64cee8d77f7752c2afa83": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "55580def279f4c6dabf56e2c01fd162a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "80410cabadd24cf398a027fc19c6655a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "36333367d23f4d348836017cf842f76a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "1ee7e50769c843eabe1b82c9749d2115": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "b695b93e8a664e74abb36fd3074c1d57": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "17cf40348c0849a7a10b0c852a751600": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        }
      }
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "source": [
        "# Clinical Trial Matching"
      ],
      "metadata": {
        "id": "QTbHPWYZUJoY"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "For the clinical trial matching, we will start by reading and processing the data from the second dataset so that it can be used for making predictions.\n",
        "\n",
        "The first document to process is the patient descriptions. We choose the first patient in order to be able to illustrate and test the application of the fine-tuned NER model using one single example."
      ],
      "metadata": {
        "id": "HEq_Qzn8Bn2m"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import pandas as pd\n",
        "import xml.etree.ElementTree as ET\n",
        "import re"
      ],
      "metadata": {
        "id": "NUFO2huEnq7F"
      },
      "execution_count": 7,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "patient_path='/content/patient_descriptions.topics'"
      ],
      "metadata": {
        "id": "PgOcJ66yDNXG"
      },
      "execution_count": 1,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "The content of the text file (patient_path) is splitted into individual sections based on the \"TOP\" delimiter, initializes an empty dictionary (patient_dict) and then processes each section to extract the \"NUM\" and \"TITLE\" information. The result is a dictionary, where the key is the patient description abd the value is the textual description."
      ],
      "metadata": {
        "id": "6Oqjz2ZM8aqb"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "with open(patient_path, 'r') as file:\n",
        "    file_content = file.read()\n",
        "\n",
        "sections = file_content.split('<TOP>\\n')\n",
        "patient_dict = {}\n",
        "\n",
        "for section in sections:\n",
        "    if section.strip():\n",
        "        num_start = section.find('<NUM>') + len('<NUM>')\n",
        "        num_end = section.find('</NUM>')\n",
        "        title_start = section.find('<TITLE>') + len('<TITLE>')\n",
        "        title_end = section.find('<NUM>', title_start)\n",
        "\n",
        "        num = section[num_start:num_end].strip()\n",
        "        title = section[title_start:title_end].strip()\n",
        "        title = title.replace(\"</TOP>\", \"\").replace(\"\\n\", \"\")\n",
        "\n",
        "        patient_dict[num] = title"
      ],
      "metadata": {
        "id": "Cr9wFjugDLIh"
      },
      "execution_count": 2,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "This is the patient description, we will use to match with trial descriptions:"
      ],
      "metadata": {
        "id": "ozRVs3999ElH"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "descrip_1 = list(patient_dict.values())[0]\n",
        "descrip_1"
      ],
      "metadata": {
        "id": "oyNyouTwSeTb",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 100
        },
        "outputId": "0d9d1f1c-1a9e-4206-a296-c4aa8f3329a5"
      },
      "execution_count": 3,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "'A 58-year-old African-American woman presents to the ER with episodic pressing/burning anterior chest pain that began two days earlier for the first time in her life. The pain started while she was walking, radiates to the back, and is accompanied by nausea, diaphoresis and mild dyspnea, but is not increased on inspiration. The latest episode of pain ended half an hour prior to her arrival. She is known to have hypertension and obesity. She denies smoking, diabetes, hypercholesterolemia, or a family history of heart disease. She currently takes no medications. Physical examination is normal. The EKG shows nonspecific changes.        '"
            ],
            "application/vnd.google.colaboratory.intrinsic+json": {
              "type": "string"
            }
          },
          "metadata": {},
          "execution_count": 3
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "In the 'matches.txt' file the patient numbers with selected trial descriptions are given, along with a number, either 0, 1 or 2.\n",
        "\n",
        "- 0 means that it is not a match\n",
        "- 1 means that it may be a match\n",
        "- 2 is a match between the two\n",
        "\n",
        "This information is stored in a table for the patient"
      ],
      "metadata": {
        "id": "VbUumiWq9PNi"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def get_matches_for_patient(input_number):\n",
        "    with open('/content/matches.txt', 'r') as file:\n",
        "        lines = file.readlines()\n",
        "\n",
        "    trials = []\n",
        "    matches = []\n",
        "\n",
        "    for line in lines:\n",
        "        parts = line.strip().split('\\t')\n",
        "        if len(parts) == 4:\n",
        "            number = int(parts[0])\n",
        "            trial = parts[2]\n",
        "            match = int(parts[3])\n",
        "            if number == input_number:\n",
        "                trials.append(trial)\n",
        "                matches.append(match)\n",
        "\n",
        "    data = pd.DataFrame({'trial': trials, 'match': matches})\n",
        "    return data\n",
        "\n",
        "input_number = 20141\n",
        "result = get_matches_for_patient(input_number)\n",
        "\n",
        "print(result)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "9DNj2Pd9E_R9",
        "outputId": "60724b83-251b-49af-f801-bf22174ef8fe"
      },
      "execution_count": 6,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "           trial  match\n",
            "0    NCT00000408      0\n",
            "1    NCT00000492      1\n",
            "2    NCT00000501      0\n",
            "3    NCT00001853      0\n",
            "4    NCT00004727      0\n",
            "..           ...    ...\n",
            "138  NCT02507050      0\n",
            "139  NCT02516839      0\n",
            "140  NCT02532699      2\n",
            "141  NCT02608255      1\n",
            "142  NCT02626741      0\n",
            "\n",
            "[143 rows x 2 columns]\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "The matching is used for simplified demonstration with 10 randomly selected trials, these are the selected files with their match value:\n",
        "\n",
        "- NCT00143195: 2\n",
        "- NCT01253486: 2\n",
        "- NCT00809029: 2\n",
        "- NCT01724567: 2\n",
        "- NCT00162344: 1\n",
        "- NCT01162902: 1\n",
        "- NCT00000408: 0\n",
        "- NCT00623454: 0\n",
        "- NCT00848250: 0\n",
        "- NCT00356707: 0\n"
      ],
      "metadata": {
        "id": "lx6jxXh4Xnfk"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "Next, only helpful information is extracted from the trial descriptions. The documents contain a lot of meta data, with the following relevant information being extracted: a short summary, a detailed description and the inclusion criteria."
      ],
      "metadata": {
        "id": "qa1Ea_Um_JBk"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def extract_criteria_text(xml_file):\n",
        "    tree = ET.parse(xml_file)\n",
        "    root = tree.getroot()\n",
        "\n",
        "    brief_summary_text = \"\"\n",
        "    detailed_description_text = \"\"\n",
        "    inclusion_criteria_text = \"\"\n",
        "\n",
        "    for textblock in root.findall(\".//detailed_description/textblock\"):\n",
        "        detailed_description_text = textblock.text.strip()\n",
        "\n",
        "    for textblock in root.findall(\".//brief_summary/textblock\"):\n",
        "        brief_summary_text = textblock.text.strip()\n",
        "\n",
        "    criteria_element = root.find(\".//criteria\")\n",
        "    if criteria_element is not None:\n",
        "        criteria_text = criteria_element.find(\".//textblock\").text\n",
        "\n",
        "        criteria_text_lower = criteria_text.lower()\n",
        "        inclusion_criteria_start = criteria_text_lower.find(\"inclusion criteria\")\n",
        "        exclusion_criteria_start = criteria_text_lower.find(\"exclusion criteria\")\n",
        "\n",
        "        if inclusion_criteria_start != -1:\n",
        "            if exclusion_criteria_start != -1:\n",
        "                inclusion_criteria_text = criteria_text[inclusion_criteria_start:exclusion_criteria_start]\n",
        "            else:\n",
        "                inclusion_criteria_text = criteria_text[inclusion_criteria_start:]\n",
        "\n",
        "    full_text = inclusion_criteria_text + brief_summary_text + detailed_description_text\n",
        "    full_text = full_text.replace(\"\\n\", \"\")\n",
        "    full_text = re.sub(r'\\s+', ' ', full_text)\n",
        "    return full_text"
      ],
      "metadata": {
        "id": "QT_itJ7oLdIG"
      },
      "execution_count": 8,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "In order to improve the matching and also include exclusion criteria, these are also extracted."
      ],
      "metadata": {
        "id": "blTZ2vl-_Vu4"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def extract_criteria_text_exclusion(xml_file):\n",
        "    tree = ET.parse(xml_file)\n",
        "    root = tree.getroot()\n",
        "\n",
        "    exclusion_criteria_text = \"\"\n",
        "\n",
        "    criteria_element = root.find(\".//criteria\")\n",
        "    if criteria_element is not None:\n",
        "        criteria_text = criteria_element.find(\".//textblock\").text\n",
        "\n",
        "        inclusion_criteria = \"\"\n",
        "        exclusion_criteria = \"\"\n",
        "\n",
        "        criteria_text_lower = criteria_text.lower()\n",
        "        if \"exclusion criteria\" in criteria_text_lower:\n",
        "            exclusion_start = criteria_text_lower.find(\"exclusion criteria\")\n",
        "            exclusion_criteria = criteria_text[exclusion_start:]\n",
        "\n",
        "        exclusion_criteria = exclusion_criteria.replace(\"\\n\", \"\")\n",
        "        exclusion_criteria = re.sub(r'\\s+', ' ', exclusion_criteria)\n",
        "        return exclusion_criteria"
      ],
      "metadata": {
        "id": "kPwABMJCwPSc"
      },
      "execution_count": 9,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Here are some examples for included and excluded data:"
      ],
      "metadata": {
        "id": "919JTO24B6wA"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "xml_file = '/content/trials/NCT00000408.xml'\n",
        "\n",
        "print(extract_criteria_text(xml_file))\n",
        "print(extract_criteria_text_exclusion(xml_file))"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "OEy5Gi_cXSke",
        "outputId": "7c46ca42-b37b-4994-d75d-44c459d9c2fd"
      },
      "execution_count": 10,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Inclusion Criteria: - Must live in the United States - Must understand and write English - Must have access to a computer with e-mail and expect to have this access for at least 3 years - Must be 18 years old - Must have seen a doctor for back pain at least once in the past year Back pain is one of the most common of all symptoms. It is also a great cause of days lost from work and visits to health care providers. This study will develop and evaluate an approach to low back pain that allows subjects to talk with each other and with health professionals via an Internet discussion group. Results we will look at include health behaviors, such as exercise; health status, such as pain and disability; and health care use, such as number of visits to doctors and other health care providers. Anyone 18 years old or older who lives in the United States and has ongoing Internet access can take part in the study. All subjects must have back pain and meet the eligibility criteria listed below.This study will develop and evaluate in a randomized trial a low back pain intervention that allows subjects to talk with each other and with health professionals via an Internet discussion group. The intervention consists of a book and a videotape and is based on interaction with other participants in the program and health professionals through a closed password protected moderated Internet discussion group. Outcome measures include health behaviors, such as exercise; health status, such as pain and disability; and health care use, such as number of visits to doctors and other health care providers. Patients will be randomized either to the treatment group and followed for 6, 12, 18, and 24 months with the intervention.\n",
            "Exclusion Criteria: - Pregnancy - Back surgery in the past 6 months - Expectation of having back surgery in the next 6 months - Back pain due to a car accident or other major injury within the last 6 months - Back pain or sciatica due to systemic disease (inflammatory rheumatic diseases, tumor, or other) - Major physical or mental health condition for which one is currently being treated that severely limits daily activities - Terminal illness - Receiving disability or workers compensation insurance payments for back pain or sciatica - Presently involved in legal proceedings because of back pain or sciatica - Difficulty with bladder or bowel control that began with back pain or sciatica - No visits to a doctor in the past year for back pain or sciatica - Numbness in crotch area that began with back pain or sciatica - Age under 18 \n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## Model\n",
        "\n",
        "BioClinicalBERT with best configuartion will be fine-tuned. Code used from fine-tuning file."
      ],
      "metadata": {
        "id": "me60mdGQMA4d"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import os\n",
        "import itertools\n",
        "import numpy as np\n",
        "!pip3 install datasets\n",
        "!pip3 install transformers\n",
        "from datasets import Dataset, DatasetDict\n",
        "from datasets import load_metric\n",
        "from transformers import AutoTokenizer\n",
        "from transformers import AutoModelForTokenClassification, TrainingArguments, Trainer\n",
        "from transformers import DataCollatorForTokenClassification\n",
        "import torch\n",
        "!pip install seqeval\n",
        "!pip install evaluate\n",
        "import evaluate\n",
        "metric = evaluate.load(\"seqeval\")\n",
        "import ast\n",
        "import pandas as pd\n",
        "from sklearn.model_selection import train_test_split\n",
        "! pip install -U accelerate\n",
        "! pip install -U transformers"
      ],
      "metadata": {
        "id": "PuABGMExMMVK"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "data = pd.read_csv(\"data.csv\")"
      ],
      "metadata": {
        "id": "NN96JXB7MQji"
      },
      "execution_count": 12,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def convert_tags(tags_string):\n",
        "    return ast.literal_eval(tags_string)\n",
        "\n",
        "data['tags'] = data['tags'].apply(convert_tags)\n",
        "data['numeric_tags'] = data['numeric_tags'].apply(convert_tags)\n",
        "data['tokens'] = data['tokens'].apply(convert_tags)"
      ],
      "metadata": {
        "id": "WQiDyqVcMRv2"
      },
      "execution_count": 13,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "label_dict = {'O': 0, 'B-Age': 1, 'I-Age': 2, 'B-Sex': 3, 'I-Sex': 4, 'B-Clinical_event': 5,\n",
        "              'I-Clinical_event': 6, 'B-Nonbiological_location': 7, 'I-Nonbiological_location': 8,\n",
        "              'B-Sign_symptom': 9, 'I-Sign_symptom': 10, 'B-Biological_structure': 11, 'I-Biological_structure': 12,\n",
        "              'B-Detailed_description': 13, 'I-Detailed_description': 14, 'B-History': 15, 'I-History': 16, 'B-Family_history': 17,\n",
        "              'I-Family_history': 18, 'B-Diagnostic_procedure': 19, 'I-Diagnostic_procedure': 20, 'B-Distance': 21,\n",
        "              'I-Distance': 22, 'B-Lab_value': 23, 'I-Lab_value': 24, 'B-Disease_disorder': 25, 'I-Disease_disorder': 26,\n",
        "              'B-Shape': 27, 'I-Shape': 28, 'B-Coreference': 29, 'I-Coreference': 30, 'B-Volume': 31, 'I-Volume': 32,\n",
        "              'B-Therapeutic_procedure': 33, 'I-Therapeutic_procedure': 34, 'B-Area': 35, 'I-Area': 36, 'B-Duration': 37,\n",
        "              'I-Duration': 38, 'B-Date': 39, 'I-Date': 40, 'B-Color': 41, 'I-Color': 42, 'B-Frequency': 43, 'I-Frequency': 44,\n",
        "              'B-Texture': 45, 'I-Texture': 46, 'B-Biological_attribute': 47, 'I-Biological_attribute': 48, 'B-Severity': 49,\n",
        "              'I-Severity': 50, 'B-Activity': 51, 'I-Activity': 52, 'B-Outcome': 53, 'I-Outcome': 54, 'B-Personal_background': 55,\n",
        "              'I-Personal_background': 56, 'B-Medication': 57, 'I-Medication': 58, 'B-Dosage': 59, 'I-Dosage': 60, 'B-Other_event': 61,\n",
        "              'I-Other_event': 62, 'B-Administration': 63, 'I-Administration': 64, 'B-Occupation': 65, 'I-Occupation': 66,\n",
        "              'B-Other_entity': 67, 'I-Other_entity': 68, 'B-Time': 69, 'I-Time': 70, 'B-Subject': 71, 'I-Subject': 72,\n",
        "              'B-Quantitative_concept': 73, 'I-Quantitative_concept': 74, 'B-Height': 75, 'I-Height': 76, 'B-Mass': 77, 'I-Mass': 78,\n",
        "              'B-Weight': 79, 'I-Weight': 80, 'B-Qualitative_concept': 81, 'I-Qualitative_concept': 82}"
      ],
      "metadata": {
        "id": "ohTCwHXbMTOX"
      },
      "execution_count": 14,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "id2label = {i: label for i, label in enumerate(label_dict)}\n",
        "label2id = {v: k for k, v in id2label.items()}"
      ],
      "metadata": {
        "id": "AxKi8-aNMUuN"
      },
      "execution_count": 15,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "X = data[\"sentence\"]\n",
        "y = data[\"tags\"]\n",
        "numeric_tags = data[\"numeric_tags\"]\n",
        "\n",
        "X_train, X_rest, y_train, y_rest, numeric_tags_train, numeric_tags_rest = train_test_split(X, y, numeric_tags, test_size=0.2, random_state=42)\n",
        "X_valid, X_test, y_valid, y_test, numeric_tags_valid, numeric_tags_test = train_test_split(X_rest, y_rest, numeric_tags_rest, test_size=0.5, random_state=42)\n",
        "\n",
        "train_df = pd.DataFrame({\"tags\": y_train, \"sentence\": X_train, \"numeric_tags\": numeric_tags_train, \"tokens\": data[\"tokens\"][X_train.index]})\n",
        "valid_df = pd.DataFrame({\"tags\": y_valid, \"sentence\": X_valid, \"numeric_tags\": numeric_tags_valid, \"tokens\": data[\"tokens\"][X_valid.index]})\n",
        "test_df = pd.DataFrame({\"tags\": y_test, \"sentence\": X_test, \"numeric_tags\": numeric_tags_test, \"tokens\": data[\"tokens\"][X_test.index]})\n",
        "\n",
        "train_dataset = Dataset.from_pandas(train_df)\n",
        "valid_dataset = Dataset.from_pandas(valid_df)\n",
        "test_dataset = Dataset.from_pandas(test_df)\n",
        "\n",
        "data_dict = DatasetDict({\n",
        "    \"train\": train_dataset,\n",
        "    \"validation\": valid_dataset,\n",
        "    \"test\": test_dataset\n",
        "})"
      ],
      "metadata": {
        "id": "lGs3yrcXMW-x"
      },
      "execution_count": 16,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def align_labels_with_tokens(labels, word_ids):\n",
        "    new_labels = []\n",
        "    current_word = None\n",
        "    for word_id in word_ids:\n",
        "        if word_id != current_word: # New word\n",
        "            current_word = word_id\n",
        "            try:\n",
        "              label = -100 if word_id is None else labels[word_id]\n",
        "            except:\n",
        "              label = -100\n",
        "            new_labels.append(label)\n",
        "        elif word_id is None: # Special token\n",
        "            new_labels.append(-100)\n",
        "        else: # Same word as previous token\n",
        "            try:\n",
        "              label = labels[word_id]\n",
        "              if label % 2 == 1: # if praefix is B-, it gets changed to I-\n",
        "                  label += 1\n",
        "            except:\n",
        "              label = -100\n",
        "            new_labels.append(label)\n",
        "\n",
        "    return new_labels"
      ],
      "metadata": {
        "id": "3TGRW5pXMdHN"
      },
      "execution_count": 17,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def tokenize_and_align_labels(examples, tokenizer):\n",
        "    tokenized_inputs = tokenizer(\n",
        "        examples[\"tokens\"], truncation=True, is_split_into_words=True\n",
        "    )\n",
        "    all_labels = examples[\"numeric_tags\"]\n",
        "    new_labels = []\n",
        "    for i, labels in enumerate(all_labels):\n",
        "        word_ids = tokenized_inputs.word_ids(i)\n",
        "        new_labels.append(align_labels_with_tokens(labels, word_ids))\n",
        "\n",
        "    tokenized_inputs[\"labels\"] = new_labels\n",
        "    return tokenized_inputs"
      ],
      "metadata": {
        "id": "l9CGzcYeMiRb"
      },
      "execution_count": 18,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def get_tokenized_data(tokenizer, data=data_dict ):\n",
        "    tokenized_datasets = data.map(\n",
        "        lambda examples: tokenize_and_align_labels(examples, tokenizer),\n",
        "        batched=True,\n",
        "        remove_columns=data[\"train\"].column_names,\n",
        "    )\n",
        "    return tokenized_datasets"
      ],
      "metadata": {
        "id": "h7TN_gupMkfX"
      },
      "execution_count": 19,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def compute_metrics(eval_preds):\n",
        "    logits, labels = eval_preds\n",
        "    predictions = np.argmax(logits, axis=-1)\n",
        "    true_labels = [[id2label[l] for l in label if l != -100] for label in labels]\n",
        "    true_predictions = [\n",
        "        [id2label[p] for (p, l) in zip(prediction, label) if l != -100]\n",
        "        for prediction, label in zip(predictions, labels)\n",
        "    ]\n",
        "    all_metrics = metric.compute(predictions=true_predictions, references=true_labels)\n",
        "    return {\n",
        "        \"precision\": all_metrics[\"overall_precision\"],\n",
        "        \"recall\": all_metrics[\"overall_recall\"],\n",
        "        \"f1\": all_metrics[\"overall_f1\"],\n",
        "        \"accuracy\": all_metrics[\"overall_accuracy\"],\n",
        "    }"
      ],
      "metadata": {
        "id": "OpmqyQ0YMssl"
      },
      "execution_count": 20,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def get_args(name_to_save, learning_rate=1e-4, num_train_epochs=3, weight_decay=0.01):\n",
        "  args = TrainingArguments(\n",
        "    name_to_save,\n",
        "    evaluation_strategy=\"epoch\",\n",
        "    save_strategy=\"epoch\",\n",
        "    learning_rate=learning_rate,\n",
        "    num_train_epochs=num_train_epochs,\n",
        "    weight_decay=weight_decay\n",
        "  )\n",
        "  return args"
      ],
      "metadata": {
        "id": "6hK8tkAiMt6M"
      },
      "execution_count": 21,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def get_model(checkpoint):\n",
        "  return AutoModelForTokenClassification.from_pretrained(\n",
        "    checkpoint,\n",
        "    id2label=id2label,\n",
        "    label2id=label2id,\n",
        ")"
      ],
      "metadata": {
        "id": "oI4r7rOWMwDL"
      },
      "execution_count": 22,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def get_trainer(model, args, training_set, eval_set, data_collator, tokenizer):\n",
        "  return Trainer(\n",
        "    model=model,\n",
        "    args=args,\n",
        "    train_dataset=training_set,\n",
        "    eval_dataset=eval_set,\n",
        "    data_collator=data_collator,\n",
        "    compute_metrics=compute_metrics,\n",
        "    tokenizer=tokenizer\n",
        ")"
      ],
      "metadata": {
        "id": "9NxTCMcbMxqj"
      },
      "execution_count": 23,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "checkpoint_clinicalBERT = \"emilyalsentzer/Bio_ClinicalBERT\"\n",
        "model_clinicalBERT = get_model(checkpoint_clinicalBERT)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "peBqvahqM0Qt",
        "outputId": "cac54b9a-03a6-4029-92e6-4f7b927cdc2f"
      },
      "execution_count": 24,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Some weights of BertForTokenClassification were not initialized from the model checkpoint at emilyalsentzer/Bio_ClinicalBERT and are newly initialized: ['classifier.weight', 'classifier.bias']\n",
            "You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "tokenizer_clinicalBERT = AutoTokenizer.from_pretrained(checkpoint_clinicalBERT)\n",
        "data_collator_clinicalBERT = DataCollatorForTokenClassification(tokenizer=tokenizer_clinicalBERT)"
      ],
      "metadata": {
        "id": "XyZ9rpsWM2YY"
      },
      "execution_count": 25,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "tokenized_datasets_clinicalBERT = get_tokenized_data(tokenizer_clinicalBERT)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 129,
          "referenced_widgets": [
            "0044cf6034584371976bd34c2636df38",
            "cc3a17ba703a4ffab2d3dcc814116cfd",
            "625520fa23bb42f2b7ceb279a080144f",
            "13eed3b66d424f8787f0bba1fd4ab7f9",
            "e689e8b6d19341068a5ce1ecb9679edb",
            "205880e57da74abda5cbfd2930d6599f",
            "df366d4d07d04554a4acc1d2531e92a5",
            "50aac9cafdb84507af0c862c893271df",
            "012e679ae0064ca694bd267db13ee556",
            "3a0018a681ac449695c4b1206616e92c",
            "2aa4e5b98dc34a73a4f085d27701b765",
            "4fcef18ad43a43af95a91441bb1a1dab",
            "9f86e1dacbda4414aaeb65ea8a915a5c",
            "5b5348e7cdd141bf847797aed7877e6d",
            "601021e6b3964ebca79882456a04ac77",
            "adf1a655c2c947c4bc782e39868828ba",
            "7b337d5fd00b42fc95fc741771b837a4",
            "21b26aa6be6646098d05e273e416ba2d",
            "0a35a1a144fe42b7b75301c67e183da5",
            "a11bd4b15170478897fdb502ccc09442",
            "002f1e1421eb483f82717dec3cc28fab",
            "36d2f410a1744578bdd59da554ce740e",
            "5f2b45a0d8cd4713ab15b8f67e28c756",
            "10a7beabd2204ea2942981f030deee63",
            "94efefbead3d4b2d9f1079472468601e",
            "f2d81f219b494caca47ef7d22e3a2117",
            "9640260a88da41f5b5b2d669afe344e2",
            "dc42502476d746e7adde1a4f5d229bbf",
            "6047621b4fd74300a3cd97278dd0a880",
            "60916d1e3fe541dc85f4126cab2ddb61",
            "f6367a21b26f4cdb926d53870aee0e18",
            "1d535770898f45feaf57f95aef8af501",
            "cc60e996b5d94979816a923a29045ec8"
          ]
        },
        "id": "RoCIyCdYM4LO",
        "outputId": "097c6802-5d0b-46b1-ab29-c71868c9489a"
      },
      "execution_count": 26,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Map:   0%|          | 0/3633 [00:00<?, ? examples/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "0044cf6034584371976bd34c2636df38"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Asking to truncate to max_length but no maximum length is provided and the model has no predefined maximum length. Default to no truncation.\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Map:   0%|          | 0/454 [00:00<?, ? examples/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "4fcef18ad43a43af95a91441bb1a1dab"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Map:   0%|          | 0/455 [00:00<?, ? examples/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "5f2b45a0d8cd4713ab15b8f67e28c756"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "trainer_clinicalBERT = get_trainer(model_clinicalBERT,\n",
        "                                   get_args(\"bioclinicalBERT_ner\"),\n",
        "                                    tokenized_datasets_clinicalBERT[\"train\"],\n",
        "                                    tokenized_datasets_clinicalBERT[\"validation\"],\n",
        "                                    data_collator_clinicalBERT,\n",
        "                                    tokenizer_clinicalBERT)"
      ],
      "metadata": {
        "id": "KhYBKnyBM5Qy"
      },
      "execution_count": 27,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "trainer_clinicalBERT.train()"
      ],
      "metadata": {
        "id": "PBZmJdF2M6sP",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 334
        },
        "outputId": "15fc33de-8c27-4b82-9e4b-acbc0241a61c"
      },
      "execution_count": 28,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "You're using a BertTokenizerFast tokenizer. Please note that with a fast tokenizer, using the `__call__` method is faster than using a method to encode the text followed by a call to the `pad` method to get a padded encoding.\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<IPython.core.display.HTML object>"
            ],
            "text/html": [
              "\n",
              "    <div>\n",
              "      \n",
              "      <progress value='1365' max='1365' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
              "      [1365/1365 03:45, Epoch 3/3]\n",
              "    </div>\n",
              "    <table border=\"1\" class=\"dataframe\">\n",
              "  <thead>\n",
              " <tr style=\"text-align: left;\">\n",
              "      <th>Epoch</th>\n",
              "      <th>Training Loss</th>\n",
              "      <th>Validation Loss</th>\n",
              "      <th>Precision</th>\n",
              "      <th>Recall</th>\n",
              "      <th>F1</th>\n",
              "      <th>Accuracy</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <td>1</td>\n",
              "      <td>No log</td>\n",
              "      <td>0.926511</td>\n",
              "      <td>0.433478</td>\n",
              "      <td>0.583750</td>\n",
              "      <td>0.497514</td>\n",
              "      <td>0.727637</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <td>2</td>\n",
              "      <td>1.181800</td>\n",
              "      <td>0.874900</td>\n",
              "      <td>0.507201</td>\n",
              "      <td>0.601667</td>\n",
              "      <td>0.550410</td>\n",
              "      <td>0.751611</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <td>3</td>\n",
              "      <td>0.643400</td>\n",
              "      <td>0.891894</td>\n",
              "      <td>0.535612</td>\n",
              "      <td>0.620417</td>\n",
              "      <td>0.574903</td>\n",
              "      <td>0.762144</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table><p>"
            ]
          },
          "metadata": {}
        },
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.10/dist-packages/seqeval/metrics/v1.py:57: UndefinedMetricWarning: Precision and F-score are ill-defined and being set to 0.0 in labels with no predicted samples. Use `zero_division` parameter to control this behavior.\n",
            "  _warn_prf(average, modifier, msg_start, len(result))\n",
            "/usr/local/lib/python3.10/dist-packages/seqeval/metrics/v1.py:57: UndefinedMetricWarning: Precision and F-score are ill-defined and being set to 0.0 in labels with no predicted samples. Use `zero_division` parameter to control this behavior.\n",
            "  _warn_prf(average, modifier, msg_start, len(result))\n",
            "/usr/local/lib/python3.10/dist-packages/seqeval/metrics/v1.py:57: UndefinedMetricWarning: Precision and F-score are ill-defined and being set to 0.0 in labels with no predicted samples. Use `zero_division` parameter to control this behavior.\n",
            "  _warn_prf(average, modifier, msg_start, len(result))\n"
          ]
        },
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "TrainOutput(global_step=1365, training_loss=0.7824189126709878, metrics={'train_runtime': 229.307, 'train_samples_per_second': 47.53, 'train_steps_per_second': 5.953, 'total_flos': 326628269271252.0, 'train_loss': 0.7824189126709878, 'epoch': 3.0})"
            ]
          },
          "metadata": {},
          "execution_count": 28
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "model = model_clinicalBERT\n",
        "tokenizer = tokenizer_clinicalBERT"
      ],
      "metadata": {
        "id": "QbzFQADeRZrw"
      },
      "execution_count": 29,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "*Copy prediction-functions from fine-tuning file to predict entities*"
      ],
      "metadata": {
        "id": "5NWapcdVUwSe"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "from transformers import pipeline"
      ],
      "metadata": {
        "id": "G7pJukg1U4p8"
      },
      "execution_count": 30,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "classifier = pipeline(\"ner\", model=model, tokenizer=tokenizer)\n",
        "model.to(\"cpu\")\n",
        "\n",
        "def predict_tags(sentence):\n",
        "  prediction = classifier(sentence)\n",
        "  return prediction"
      ],
      "metadata": {
        "id": "J9ZgdpVyU563"
      },
      "execution_count": 31,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "An example to try if everything works:"
      ],
      "metadata": {
        "id": "4lYluj3-CUde"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "predict_tags(patient_dict[\"20154\"])"
      ],
      "metadata": {
        "id": "t9k-WCNvgmLj",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "526c21b4-f39f-4810-8729-bc781fd3b0e8"
      },
      "execution_count": 32,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "[{'entity': 'B-Age',\n",
              "  'score': 0.99116594,\n",
              "  'index': 2,\n",
              "  'word': '82',\n",
              "  'start': 3,\n",
              "  'end': 5},\n",
              " {'entity': 'I-Age',\n",
              "  'score': 0.9957867,\n",
              "  'index': 3,\n",
              "  'word': '-',\n",
              "  'start': 5,\n",
              "  'end': 6},\n",
              " {'entity': 'I-Age',\n",
              "  'score': 0.99554574,\n",
              "  'index': 4,\n",
              "  'word': 'year',\n",
              "  'start': 6,\n",
              "  'end': 10},\n",
              " {'entity': 'I-Age',\n",
              "  'score': 0.9956742,\n",
              "  'index': 5,\n",
              "  'word': '-',\n",
              "  'start': 10,\n",
              "  'end': 11},\n",
              " {'entity': 'I-Age',\n",
              "  'score': 0.9952171,\n",
              "  'index': 6,\n",
              "  'word': 'old',\n",
              "  'start': 11,\n",
              "  'end': 14},\n",
              " {'entity': 'B-Sex',\n",
              "  'score': 0.9897379,\n",
              "  'index': 7,\n",
              "  'word': 'woman',\n",
              "  'start': 15,\n",
              "  'end': 20},\n",
              " {'entity': 'B-Clinical_event',\n",
              "  'score': 0.96544045,\n",
              "  'index': 8,\n",
              "  'word': 'comes',\n",
              "  'start': 21,\n",
              "  'end': 26},\n",
              " {'entity': 'B-Nonbiological_location',\n",
              "  'score': 0.9009364,\n",
              "  'index': 11,\n",
              "  'word': 'emergency',\n",
              "  'start': 34,\n",
              "  'end': 43},\n",
              " {'entity': 'I-Nonbiological_location',\n",
              "  'score': 0.9545496,\n",
              "  'index': 12,\n",
              "  'word': 'department',\n",
              "  'start': 44,\n",
              "  'end': 54},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.97602344,\n",
              "  'index': 15,\n",
              "  'word': 'chest',\n",
              "  'start': 66,\n",
              "  'end': 71},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9841467,\n",
              "  'index': 16,\n",
              "  'word': 'pain',\n",
              "  'start': 72,\n",
              "  'end': 76},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.98459816,\n",
              "  'index': 18,\n",
              "  'word': 'short',\n",
              "  'start': 81,\n",
              "  'end': 86},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.98669654,\n",
              "  'index': 19,\n",
              "  'word': '##ness',\n",
              "  'start': 86,\n",
              "  'end': 90},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.98793554,\n",
              "  'index': 20,\n",
              "  'word': 'of',\n",
              "  'start': 91,\n",
              "  'end': 93},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9861855,\n",
              "  'index': 21,\n",
              "  'word': 'breath',\n",
              "  'start': 94,\n",
              "  'end': 100},\n",
              " {'entity': 'B-Clinical_event',\n",
              "  'score': 0.46880144,\n",
              "  'index': 24,\n",
              "  'word': 'awakened',\n",
              "  'start': 113,\n",
              "  'end': 121},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.51456064,\n",
              "  'index': 26,\n",
              "  'word': 'the',\n",
              "  'start': 125,\n",
              "  'end': 128},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.660962,\n",
              "  'index': 27,\n",
              "  'word': 'morning',\n",
              "  'start': 129,\n",
              "  'end': 136},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.96790904,\n",
              "  'index': 29,\n",
              "  'word': 'stabbing',\n",
              "  'start': 140,\n",
              "  'end': 148},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.75009286,\n",
              "  'index': 30,\n",
              "  'word': 'sub',\n",
              "  'start': 149,\n",
              "  'end': 152},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.8468901,\n",
              "  'index': 31,\n",
              "  'word': '##ster',\n",
              "  'start': 152,\n",
              "  'end': 156},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.96003324,\n",
              "  'index': 32,\n",
              "  'word': '##nal',\n",
              "  'start': 156,\n",
              "  'end': 159},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9789039,\n",
              "  'index': 33,\n",
              "  'word': 'chest',\n",
              "  'start': 160,\n",
              "  'end': 165},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9781854,\n",
              "  'index': 34,\n",
              "  'word': 'pain',\n",
              "  'start': 166,\n",
              "  'end': 170},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9873776,\n",
              "  'index': 38,\n",
              "  'word': 'left',\n",
              "  'start': 188,\n",
              "  'end': 192},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98423505,\n",
              "  'index': 39,\n",
              "  'word': 'shoulder',\n",
              "  'start': 193,\n",
              "  'end': 201},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9849429,\n",
              "  'index': 41,\n",
              "  'word': 'jaw',\n",
              "  'start': 206,\n",
              "  'end': 209},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9073664,\n",
              "  'index': 46,\n",
              "  'word': 'h',\n",
              "  'start': 227,\n",
              "  'end': 228},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9537667,\n",
              "  'index': 47,\n",
              "  'word': '##yper',\n",
              "  'start': 228,\n",
              "  'end': 232},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.94417447,\n",
              "  'index': 48,\n",
              "  'word': '##tens',\n",
              "  'start': 232,\n",
              "  'end': 236},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9460092,\n",
              "  'index': 49,\n",
              "  'word': '##ion',\n",
              "  'start': 236,\n",
              "  'end': 239},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.8598172,\n",
              "  'index': 51,\n",
              "  'word': 're',\n",
              "  'start': 241,\n",
              "  'end': 243},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.58184004,\n",
              "  'index': 52,\n",
              "  'word': '##nal',\n",
              "  'start': 243,\n",
              "  'end': 246},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.4860088,\n",
              "  'index': 53,\n",
              "  'word': '-',\n",
              "  'start': 246,\n",
              "  'end': 247},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.3632132,\n",
              "  'index': 54,\n",
              "  'word': 'artery',\n",
              "  'start': 247,\n",
              "  'end': 253},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.41342488,\n",
              "  'index': 55,\n",
              "  'word': 's',\n",
              "  'start': 254,\n",
              "  'end': 255},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.68325895,\n",
              "  'index': 56,\n",
              "  'word': '##ten',\n",
              "  'start': 255,\n",
              "  'end': 258},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.54690886,\n",
              "  'index': 57,\n",
              "  'word': '##osis',\n",
              "  'start': 258,\n",
              "  'end': 262},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.69647986,\n",
              "  'index': 59,\n",
              "  'word': 'chronic',\n",
              "  'start': 268,\n",
              "  'end': 275},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.44666272,\n",
              "  'index': 60,\n",
              "  'word': 're',\n",
              "  'start': 276,\n",
              "  'end': 278},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.52430576,\n",
              "  'index': 61,\n",
              "  'word': '##nal',\n",
              "  'start': 278,\n",
              "  'end': 281},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.601716,\n",
              "  'index': 62,\n",
              "  'word': 'ins',\n",
              "  'start': 282,\n",
              "  'end': 285},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.6358053,\n",
              "  'index': 63,\n",
              "  'word': '##uff',\n",
              "  'start': 285,\n",
              "  'end': 288},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.5461173,\n",
              "  'index': 64,\n",
              "  'word': '##iciency',\n",
              "  'start': 288,\n",
              "  'end': 295},\n",
              " {'entity': 'B-Disease_disorder',\n",
              "  'score': 0.60557616,\n",
              "  'index': 66,\n",
              "  'word': 'h',\n",
              "  'start': 297,\n",
              "  'end': 298},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.5438338,\n",
              "  'index': 67,\n",
              "  'word': '##yper',\n",
              "  'start': 298,\n",
              "  'end': 302},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.5720529,\n",
              "  'index': 68,\n",
              "  'word': '##cho',\n",
              "  'start': 302,\n",
              "  'end': 305},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.59258604,\n",
              "  'index': 69,\n",
              "  'word': '##les',\n",
              "  'start': 305,\n",
              "  'end': 308},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.5814714,\n",
              "  'index': 70,\n",
              "  'word': '##tero',\n",
              "  'start': 308,\n",
              "  'end': 312},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.6595369,\n",
              "  'index': 71,\n",
              "  'word': '##lem',\n",
              "  'start': 312,\n",
              "  'end': 315},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.69859844,\n",
              "  'index': 72,\n",
              "  'word': '##ia',\n",
              "  'start': 315,\n",
              "  'end': 317},\n",
              " {'entity': 'B-Disease_disorder',\n",
              "  'score': 0.37963325,\n",
              "  'index': 74,\n",
              "  'word': 'o',\n",
              "  'start': 319,\n",
              "  'end': 320},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.63833153,\n",
              "  'index': 75,\n",
              "  'word': '##ste',\n",
              "  'start': 320,\n",
              "  'end': 323},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.5892713,\n",
              "  'index': 76,\n",
              "  'word': '##op',\n",
              "  'start': 323,\n",
              "  'end': 325},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.6723521,\n",
              "  'index': 77,\n",
              "  'word': '##oro',\n",
              "  'start': 325,\n",
              "  'end': 328},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.7956205,\n",
              "  'index': 78,\n",
              "  'word': '##sis',\n",
              "  'start': 328,\n",
              "  'end': 331},\n",
              " {'entity': 'B-Disease_disorder',\n",
              "  'score': 0.6458075,\n",
              "  'index': 80,\n",
              "  'word': 'dem',\n",
              "  'start': 336,\n",
              "  'end': 339},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.5967636,\n",
              "  'index': 81,\n",
              "  'word': '##ent',\n",
              "  'start': 339,\n",
              "  'end': 342},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.5743018,\n",
              "  'index': 82,\n",
              "  'word': '##ia',\n",
              "  'start': 342,\n",
              "  'end': 344},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9898717,\n",
              "  'index': 84,\n",
              "  'word': 'blood',\n",
              "  'start': 346,\n",
              "  'end': 351},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9931653,\n",
              "  'index': 85,\n",
              "  'word': 'pressure',\n",
              "  'start': 352,\n",
              "  'end': 360},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.925655,\n",
              "  'index': 87,\n",
              "  'word': '199',\n",
              "  'start': 365,\n",
              "  'end': 368},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9625185,\n",
              "  'index': 88,\n",
              "  'word': '/',\n",
              "  'start': 368,\n",
              "  'end': 369},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9013916,\n",
              "  'index': 89,\n",
              "  'word': '108',\n",
              "  'start': 369,\n",
              "  'end': 372},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.97280204,\n",
              "  'index': 90,\n",
              "  'word': 'mm',\n",
              "  'start': 373,\n",
              "  'end': 375},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9794272,\n",
              "  'index': 91,\n",
              "  'word': 'h',\n",
              "  'start': 376,\n",
              "  'end': 377},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.96906906,\n",
              "  'index': 92,\n",
              "  'word': '##g',\n",
              "  'start': 377,\n",
              "  'end': 378},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.97785735,\n",
              "  'index': 94,\n",
              "  'word': 'respiratory',\n",
              "  'start': 380,\n",
              "  'end': 391},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.984132,\n",
              "  'index': 95,\n",
              "  'word': 'rate',\n",
              "  'start': 392,\n",
              "  'end': 396},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.90291893,\n",
              "  'index': 96,\n",
              "  'word': '18',\n",
              "  'start': 397,\n",
              "  'end': 399},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9737846,\n",
              "  'index': 97,\n",
              "  'word': 'b',\n",
              "  'start': 400,\n",
              "  'end': 401},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.97588795,\n",
              "  'index': 98,\n",
              "  'word': '##pm',\n",
              "  'start': 401,\n",
              "  'end': 403},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9780385,\n",
              "  'index': 100,\n",
              "  'word': 'oxygen',\n",
              "  'start': 405,\n",
              "  'end': 411},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9866362,\n",
              "  'index': 101,\n",
              "  'word': 'sat',\n",
              "  'start': 412,\n",
              "  'end': 415},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9803744,\n",
              "  'index': 102,\n",
              "  'word': '##uration',\n",
              "  'start': 415,\n",
              "  'end': 422},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.9351836,\n",
              "  'index': 106,\n",
              "  'word': 'ambient',\n",
              "  'start': 430,\n",
              "  'end': 437},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.96065617,\n",
              "  'index': 107,\n",
              "  'word': 'air',\n",
              "  'start': 438,\n",
              "  'end': 441},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.93716145,\n",
              "  'index': 110,\n",
              "  'word': 'heart',\n",
              "  'start': 447,\n",
              "  'end': 452},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.95968276,\n",
              "  'index': 111,\n",
              "  'word': 'sounds',\n",
              "  'start': 453,\n",
              "  'end': 459},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.80004835,\n",
              "  'index': 113,\n",
              "  'word': 'rapid',\n",
              "  'start': 465,\n",
              "  'end': 470},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9015711,\n",
              "  'index': 117,\n",
              "  'word': 'murmurs',\n",
              "  'start': 483,\n",
              "  'end': 490},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.95810413,\n",
              "  'index': 119,\n",
              "  'word': 'c',\n",
              "  'start': 492,\n",
              "  'end': 493},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9770832,\n",
              "  'index': 120,\n",
              "  'word': '##k',\n",
              "  'start': 493,\n",
              "  'end': 494},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.92038035,\n",
              "  'index': 121,\n",
              "  'word': '-',\n",
              "  'start': 494,\n",
              "  'end': 495},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.92437655,\n",
              "  'index': 122,\n",
              "  'word': 'm',\n",
              "  'start': 495,\n",
              "  'end': 496},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9012284,\n",
              "  'index': 123,\n",
              "  'word': '##b',\n",
              "  'start': 496,\n",
              "  'end': 497},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.583351,\n",
              "  'index': 125,\n",
              "  'word': '10',\n",
              "  'start': 502,\n",
              "  'end': 504},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.7309681,\n",
              "  'index': 126,\n",
              "  'word': '.',\n",
              "  'start': 504,\n",
              "  'end': 505},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9233455,\n",
              "  'index': 128,\n",
              "  'word': 'ng',\n",
              "  'start': 507,\n",
              "  'end': 509},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.92423373,\n",
              "  'index': 129,\n",
              "  'word': '/',\n",
              "  'start': 509,\n",
              "  'end': 510},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.92800856,\n",
              "  'index': 130,\n",
              "  'word': 'm',\n",
              "  'start': 510,\n",
              "  'end': 511},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9016934,\n",
              "  'index': 131,\n",
              "  'word': '##l',\n",
              "  'start': 511,\n",
              "  'end': 512},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.96531016,\n",
              "  'index': 133,\n",
              "  'word': 'c',\n",
              "  'start': 514,\n",
              "  'end': 515},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.97421557,\n",
              "  'index': 134,\n",
              "  'word': '##k',\n",
              "  'start': 515,\n",
              "  'end': 516},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.7439547,\n",
              "  'index': 136,\n",
              "  'word': '89',\n",
              "  'start': 521,\n",
              "  'end': 523},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.92835444,\n",
              "  'index': 137,\n",
              "  'word': 'u',\n",
              "  'start': 524,\n",
              "  'end': 525},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.90280443,\n",
              "  'index': 138,\n",
              "  'word': '/',\n",
              "  'start': 525,\n",
              "  'end': 526},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.91042274,\n",
              "  'index': 139,\n",
              "  'word': 'l',\n",
              "  'start': 526,\n",
              "  'end': 527},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9701975,\n",
              "  'index': 141,\n",
              "  'word': 'c',\n",
              "  'start': 529,\n",
              "  'end': 530},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9778586,\n",
              "  'index': 142,\n",
              "  'word': '##k',\n",
              "  'start': 530,\n",
              "  'end': 531},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9580351,\n",
              "  'index': 143,\n",
              "  'word': 'index',\n",
              "  'start': 532,\n",
              "  'end': 537},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.94889206,\n",
              "  'index': 150,\n",
              "  'word': 't',\n",
              "  'start': 552,\n",
              "  'end': 553},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.965166,\n",
              "  'index': 151,\n",
              "  'word': '##rop',\n",
              "  'start': 553,\n",
              "  'end': 556},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.97131395,\n",
              "  'index': 152,\n",
              "  'word': '##oni',\n",
              "  'start': 556,\n",
              "  'end': 559},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9813743,\n",
              "  'index': 153,\n",
              "  'word': '##n',\n",
              "  'start': 559,\n",
              "  'end': 560},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.94735694,\n",
              "  'index': 154,\n",
              "  'word': 't',\n",
              "  'start': 561,\n",
              "  'end': 562},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.62030214,\n",
              "  'index': 156,\n",
              "  'word': '0',\n",
              "  'start': 567,\n",
              "  'end': 568},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.75664115,\n",
              "  'index': 157,\n",
              "  'word': '.',\n",
              "  'start': 568,\n",
              "  'end': 569},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.67661715,\n",
              "  'index': 158,\n",
              "  'word': '40',\n",
              "  'start': 569,\n",
              "  'end': 571},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.93408185,\n",
              "  'index': 159,\n",
              "  'word': 'ng',\n",
              "  'start': 572,\n",
              "  'end': 574},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9506873,\n",
              "  'index': 160,\n",
              "  'word': '/',\n",
              "  'start': 574,\n",
              "  'end': 575},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.94248897,\n",
              "  'index': 161,\n",
              "  'word': 'm',\n",
              "  'start': 575,\n",
              "  'end': 576},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.9473685,\n",
              "  'index': 162,\n",
              "  'word': '##l',\n",
              "  'start': 576,\n",
              "  'end': 577},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9835963,\n",
              "  'index': 165,\n",
              "  'word': 'e',\n",
              "  'start': 582,\n",
              "  'end': 583},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.992324,\n",
              "  'index': 166,\n",
              "  'word': '##k',\n",
              "  'start': 583,\n",
              "  'end': 584},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.99471045,\n",
              "  'index': 167,\n",
              "  'word': '##g',\n",
              "  'start': 584,\n",
              "  'end': 585},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.93072563,\n",
              "  'index': 169,\n",
              "  'word': 'sin',\n",
              "  'start': 593,\n",
              "  'end': 596},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.9385716,\n",
              "  'index': 170,\n",
              "  'word': '##us',\n",
              "  'start': 596,\n",
              "  'end': 598},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.36694235,\n",
              "  'index': 171,\n",
              "  'word': 'regular',\n",
              "  'start': 599,\n",
              "  'end': 606},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.8930933,\n",
              "  'index': 172,\n",
              "  'word': 'ta',\n",
              "  'start': 607,\n",
              "  'end': 609},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9420014,\n",
              "  'index': 173,\n",
              "  'word': '##chy',\n",
              "  'start': 609,\n",
              "  'end': 612},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9334554,\n",
              "  'index': 174,\n",
              "  'word': '##card',\n",
              "  'start': 612,\n",
              "  'end': 616},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.90553457,\n",
              "  'index': 175,\n",
              "  'word': '##ia',\n",
              "  'start': 616,\n",
              "  'end': 618},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.894049,\n",
              "  'index': 177,\n",
              "  'word': '119',\n",
              "  'start': 622,\n",
              "  'end': 625},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.94296044,\n",
              "  'index': 178,\n",
              "  'word': 'b',\n",
              "  'start': 626,\n",
              "  'end': 627},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.91782236,\n",
              "  'index': 179,\n",
              "  'word': '##pm',\n",
              "  'start': 627,\n",
              "  'end': 629},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9449503,\n",
              "  'index': 182,\n",
              "  'word': 's',\n",
              "  'start': 636,\n",
              "  'end': 637},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9509056,\n",
              "  'index': 183,\n",
              "  'word': '##t',\n",
              "  'start': 637,\n",
              "  'end': 638},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.97440344,\n",
              "  'index': 184,\n",
              "  'word': '-',\n",
              "  'start': 638,\n",
              "  'end': 639},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9700603,\n",
              "  'index': 185,\n",
              "  'word': 'segment',\n",
              "  'start': 639,\n",
              "  'end': 646},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9657587,\n",
              "  'index': 186,\n",
              "  'word': 'elevations',\n",
              "  'start': 647,\n",
              "  'end': 657},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.40225622,\n",
              "  'index': 189,\n",
              "  'word': '3',\n",
              "  'start': 664,\n",
              "  'end': 665},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.87922865,\n",
              "  'index': 190,\n",
              "  'word': 'mm',\n",
              "  'start': 666,\n",
              "  'end': 668},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.91303504,\n",
              "  'index': 192,\n",
              "  'word': 'v',\n",
              "  'start': 672,\n",
              "  'end': 673},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.8629679,\n",
              "  'index': 193,\n",
              "  'word': '##1',\n",
              "  'start': 673,\n",
              "  'end': 674},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9077325,\n",
              "  'index': 195,\n",
              "  'word': 'v',\n",
              "  'start': 676,\n",
              "  'end': 677},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9007026,\n",
              "  'index': 196,\n",
              "  'word': '##2',\n",
              "  'start': 677,\n",
              "  'end': 678},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.94149435,\n",
              "  'index': 199,\n",
              "  'word': 'v',\n",
              "  'start': 684,\n",
              "  'end': 685},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.89795464,\n",
              "  'index': 200,\n",
              "  'word': '##3',\n",
              "  'start': 685,\n",
              "  'end': 686},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9651459,\n",
              "  'index': 203,\n",
              "  'word': 'chest',\n",
              "  'start': 690,\n",
              "  'end': 695},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9871634,\n",
              "  'index': 204,\n",
              "  'word': 'radio',\n",
              "  'start': 696,\n",
              "  'end': 701},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9955179,\n",
              "  'index': 205,\n",
              "  'word': '##graph',\n",
              "  'start': 701,\n",
              "  'end': 706},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.9640857,\n",
              "  'index': 207,\n",
              "  'word': 'low',\n",
              "  'start': 714,\n",
              "  'end': 717},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.95851874,\n",
              "  'index': 208,\n",
              "  'word': 'lung',\n",
              "  'start': 718,\n",
              "  'end': 722},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.88191324,\n",
              "  'index': 209,\n",
              "  'word': 'volumes',\n",
              "  'start': 723,\n",
              "  'end': 730},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.95303077,\n",
              "  'index': 211,\n",
              "  'word': 'right',\n",
              "  'start': 735,\n",
              "  'end': 740},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98544353,\n",
              "  'index': 212,\n",
              "  'word': 'b',\n",
              "  'start': 741,\n",
              "  'end': 742},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98344857,\n",
              "  'index': 213,\n",
              "  'word': '##asi',\n",
              "  'start': 742,\n",
              "  'end': 745},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9876617,\n",
              "  'index': 214,\n",
              "  'word': '##lar',\n",
              "  'start': 745,\n",
              "  'end': 748},\n",
              " {'entity': 'B-Disease_disorder',\n",
              "  'score': 0.41654086,\n",
              "  'index': 215,\n",
              "  'word': 'sub',\n",
              "  'start': 749,\n",
              "  'end': 752},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.63412005,\n",
              "  'index': 216,\n",
              "  'word': '##se',\n",
              "  'start': 752,\n",
              "  'end': 754},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.312555,\n",
              "  'index': 217,\n",
              "  'word': '##gment',\n",
              "  'start': 754,\n",
              "  'end': 759},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.3790557,\n",
              "  'index': 218,\n",
              "  'word': '##al',\n",
              "  'start': 759,\n",
              "  'end': 761},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.5539687,\n",
              "  'index': 219,\n",
              "  'word': 'ate',\n",
              "  'start': 762,\n",
              "  'end': 765},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.6464451,\n",
              "  'index': 220,\n",
              "  'word': '##lect',\n",
              "  'start': 765,\n",
              "  'end': 769},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.5491357,\n",
              "  'index': 221,\n",
              "  'word': '##asis',\n",
              "  'start': 769,\n",
              "  'end': 773},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9807107,\n",
              "  'index': 223,\n",
              "  'word': 'co',\n",
              "  'start': 775,\n",
              "  'end': 777},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.82756627,\n",
              "  'index': 224,\n",
              "  'word': '##rona',\n",
              "  'start': 777,\n",
              "  'end': 781},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.48100108,\n",
              "  'index': 225,\n",
              "  'word': '##ry',\n",
              "  'start': 781,\n",
              "  'end': 783},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.6991683,\n",
              "  'index': 226,\n",
              "  'word': 'an',\n",
              "  'start': 784,\n",
              "  'end': 786},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9962095,\n",
              "  'index': 227,\n",
              "  'word': '##gio',\n",
              "  'start': 786,\n",
              "  'end': 789},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9955627,\n",
              "  'index': 228,\n",
              "  'word': '##graphy',\n",
              "  'start': 789,\n",
              "  'end': 795},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.6965778,\n",
              "  'index': 231,\n",
              "  'word': 's',\n",
              "  'start': 806,\n",
              "  'end': 807},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.81710297,\n",
              "  'index': 232,\n",
              "  'word': '##ten',\n",
              "  'start': 807,\n",
              "  'end': 810},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.65213704,\n",
              "  'index': 233,\n",
              "  'word': '##osis',\n",
              "  'start': 810,\n",
              "  'end': 814},\n",
              " {'entity': 'B-Disease_disorder',\n",
              "  'score': 0.8534777,\n",
              "  'index': 235,\n",
              "  'word': 'clinical',\n",
              "  'start': 818,\n",
              "  'end': 826},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.59690756,\n",
              "  'index': 236,\n",
              "  'word': '##ly',\n",
              "  'start': 826,\n",
              "  'end': 828},\n",
              " {'entity': 'I-Disease_disorder',\n",
              "  'score': 0.5669882,\n",
              "  'index': 237,\n",
              "  'word': 'significant',\n",
              "  'start': 829,\n",
              "  'end': 840},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.5180264,\n",
              "  'index': 238,\n",
              "  'word': 'disease',\n",
              "  'start': 841,\n",
              "  'end': 848},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.9688445,\n",
              "  'index': 240,\n",
              "  'word': 'left',\n",
              "  'start': 850,\n",
              "  'end': 854},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.7040885,\n",
              "  'index': 241,\n",
              "  'word': 'vent',\n",
              "  'start': 855,\n",
              "  'end': 859},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.7052556,\n",
              "  'index': 242,\n",
              "  'word': '##ric',\n",
              "  'start': 859,\n",
              "  'end': 862},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9833626,\n",
              "  'index': 243,\n",
              "  'word': '##ulo',\n",
              "  'start': 862,\n",
              "  'end': 865},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9918099,\n",
              "  'index': 244,\n",
              "  'word': '##graphy',\n",
              "  'start': 865,\n",
              "  'end': 871},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.9745953,\n",
              "  'index': 246,\n",
              "  'word': 'akin',\n",
              "  'start': 881,\n",
              "  'end': 885},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.94755137,\n",
              "  'index': 247,\n",
              "  'word': '##esis',\n",
              "  'start': 885,\n",
              "  'end': 889},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9775763,\n",
              "  'index': 250,\n",
              "  'word': 'anterior',\n",
              "  'start': 897,\n",
              "  'end': 905},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.99099755,\n",
              "  'index': 251,\n",
              "  'word': 'wall',\n",
              "  'start': 906,\n",
              "  'end': 910},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.93676615,\n",
              "  'index': 253,\n",
              "  'word': 'h',\n",
              "  'start': 912,\n",
              "  'end': 913},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.863527,\n",
              "  'index': 254,\n",
              "  'word': '##y',\n",
              "  'start': 913,\n",
              "  'end': 914},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.90121496,\n",
              "  'index': 255,\n",
              "  'word': '##po',\n",
              "  'start': 914,\n",
              "  'end': 916},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9113241,\n",
              "  'index': 256,\n",
              "  'word': '##kin',\n",
              "  'start': 916,\n",
              "  'end': 919},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9236261,\n",
              "  'index': 257,\n",
              "  'word': '##esis',\n",
              "  'start': 919,\n",
              "  'end': 923},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9475331,\n",
              "  'index': 260,\n",
              "  'word': 'a',\n",
              "  'start': 931,\n",
              "  'end': 932},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.97144115,\n",
              "  'index': 261,\n",
              "  'word': '##pical',\n",
              "  'start': 932,\n",
              "  'end': 937},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.914755,\n",
              "  'index': 262,\n",
              "  'word': 'and',\n",
              "  'start': 938,\n",
              "  'end': 941},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.46916676,\n",
              "  'index': 263,\n",
              "  'word': 'di',\n",
              "  'start': 942,\n",
              "  'end': 944},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.97046196,\n",
              "  'index': 264,\n",
              "  'word': '##stal',\n",
              "  'start': 944,\n",
              "  'end': 948},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9341015,\n",
              "  'index': 265,\n",
              "  'word': 'inferior',\n",
              "  'start': 949,\n",
              "  'end': 957},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.893528,\n",
              "  'index': 266,\n",
              "  'word': 'walls',\n",
              "  'start': 958,\n",
              "  'end': 963},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.65402335,\n",
              "  'index': 269,\n",
              "  'word': 'com',\n",
              "  'start': 969,\n",
              "  'end': 972},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.7301994,\n",
              "  'index': 270,\n",
              "  'word': '##pen',\n",
              "  'start': 972,\n",
              "  'end': 975},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.9417031,\n",
              "  'index': 271,\n",
              "  'word': '##sat',\n",
              "  'start': 975,\n",
              "  'end': 978},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.92024994,\n",
              "  'index': 272,\n",
              "  'word': '##ory',\n",
              "  'start': 978,\n",
              "  'end': 981},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.8847374,\n",
              "  'index': 273,\n",
              "  'word': 'h',\n",
              "  'start': 982,\n",
              "  'end': 983},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.9294455,\n",
              "  'index': 274,\n",
              "  'word': '##yper',\n",
              "  'start': 983,\n",
              "  'end': 987},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.8272192,\n",
              "  'index': 275,\n",
              "  'word': '##kin',\n",
              "  'start': 987,\n",
              "  'end': 990},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.92062825,\n",
              "  'index': 276,\n",
              "  'word': '##esis',\n",
              "  'start': 990,\n",
              "  'end': 994},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9693058,\n",
              "  'index': 279,\n",
              "  'word': 'basal',\n",
              "  'start': 1002,\n",
              "  'end': 1007},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.965882,\n",
              "  'index': 280,\n",
              "  'word': 'anterior',\n",
              "  'start': 1008,\n",
              "  'end': 1016},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.91551626,\n",
              "  'index': 281,\n",
              "  'word': 'and',\n",
              "  'start': 1017,\n",
              "  'end': 1020},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.73333627,\n",
              "  'index': 282,\n",
              "  'word': 'basal',\n",
              "  'start': 1021,\n",
              "  'end': 1026},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.96827286,\n",
              "  'index': 283,\n",
              "  'word': 'inferior',\n",
              "  'start': 1027,\n",
              "  'end': 1035},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9594169,\n",
              "  'index': 284,\n",
              "  'word': 'walls',\n",
              "  'start': 1036,\n",
              "  'end': 1041},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.5065216,\n",
              "  'index': 287,\n",
              "  'word': 'trans',\n",
              "  'start': 1045,\n",
              "  'end': 1050},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.6703042,\n",
              "  'index': 288,\n",
              "  'word': '##th',\n",
              "  'start': 1050,\n",
              "  'end': 1052},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9247918,\n",
              "  'index': 289,\n",
              "  'word': '##ora',\n",
              "  'start': 1052,\n",
              "  'end': 1055},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9310772,\n",
              "  'index': 290,\n",
              "  'word': '##ci',\n",
              "  'start': 1055,\n",
              "  'end': 1057},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9595867,\n",
              "  'index': 291,\n",
              "  'word': '##c',\n",
              "  'start': 1057,\n",
              "  'end': 1058},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.85125977,\n",
              "  'index': 292,\n",
              "  'word': 'echo',\n",
              "  'start': 1059,\n",
              "  'end': 1063},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9933014,\n",
              "  'index': 293,\n",
              "  'word': '##card',\n",
              "  'start': 1063,\n",
              "  'end': 1067},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9939976,\n",
              "  'index': 294,\n",
              "  'word': '##io',\n",
              "  'start': 1067,\n",
              "  'end': 1069},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9946114,\n",
              "  'index': 295,\n",
              "  'word': '##gram',\n",
              "  'start': 1069,\n",
              "  'end': 1073},\n",
              " {'entity': 'B-Severity',\n",
              "  'score': 0.9770867,\n",
              "  'index': 297,\n",
              "  'word': 'severe',\n",
              "  'start': 1081,\n",
              "  'end': 1087},\n",
              " {'entity': 'B-Detailed_description',\n",
              "  'score': 0.5328475,\n",
              "  'index': 298,\n",
              "  'word': 'segment',\n",
              "  'start': 1088,\n",
              "  'end': 1095},\n",
              " {'entity': 'I-Detailed_description',\n",
              "  'score': 0.89813167,\n",
              "  'index': 299,\n",
              "  'word': '##al',\n",
              "  'start': 1095,\n",
              "  'end': 1097},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.44993958,\n",
              "  'index': 300,\n",
              "  'word': 'left',\n",
              "  'start': 1098,\n",
              "  'end': 1102},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.8872347,\n",
              "  'index': 301,\n",
              "  'word': 'vent',\n",
              "  'start': 1103,\n",
              "  'end': 1107},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.8744576,\n",
              "  'index': 302,\n",
              "  'word': '##ric',\n",
              "  'start': 1107,\n",
              "  'end': 1110},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.8857194,\n",
              "  'index': 303,\n",
              "  'word': '##ular',\n",
              "  'start': 1110,\n",
              "  'end': 1114},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.8136631,\n",
              "  'index': 304,\n",
              "  'word': 'd',\n",
              "  'start': 1115,\n",
              "  'end': 1116},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.87526214,\n",
              "  'index': 305,\n",
              "  'word': '##ys',\n",
              "  'start': 1116,\n",
              "  'end': 1118},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.7074332,\n",
              "  'index': 306,\n",
              "  'word': '##function',\n",
              "  'start': 1118,\n",
              "  'end': 1126},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.9769646,\n",
              "  'index': 309,\n",
              "  'word': 'se',\n",
              "  'start': 1141,\n",
              "  'end': 1143},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.97921926,\n",
              "  'index': 310,\n",
              "  'word': '##pt',\n",
              "  'start': 1143,\n",
              "  'end': 1145},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98752546,\n",
              "  'index': 311,\n",
              "  'word': '##um',\n",
              "  'start': 1145,\n",
              "  'end': 1147},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.95023346,\n",
              "  'index': 313,\n",
              "  'word': 'ant',\n",
              "  'start': 1149,\n",
              "  'end': 1152},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98803306,\n",
              "  'index': 314,\n",
              "  'word': '##ero',\n",
              "  'start': 1152,\n",
              "  'end': 1155},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9886248,\n",
              "  'index': 315,\n",
              "  'word': '##se',\n",
              "  'start': 1155,\n",
              "  'end': 1157},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.9807222,\n",
              "  'index': 316,\n",
              "  'word': '##pta',\n",
              "  'start': 1157,\n",
              "  'end': 1160},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98848885,\n",
              "  'index': 317,\n",
              "  'word': '##l',\n",
              "  'start': 1160,\n",
              "  'end': 1161},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.98366725,\n",
              "  'index': 318,\n",
              "  'word': 'territory',\n",
              "  'start': 1162,\n",
              "  'end': 1171},\n",
              " {'entity': 'B-Biological_structure',\n",
              "  'score': 0.96185726,\n",
              "  'index': 321,\n",
              "  'word': 'apex',\n",
              "  'start': 1177,\n",
              "  'end': 1181},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.904804,\n",
              "  'index': 325,\n",
              "  'word': 'left',\n",
              "  'start': 1195,\n",
              "  'end': 1199},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.7759866,\n",
              "  'index': 326,\n",
              "  'word': 'vent',\n",
              "  'start': 1200,\n",
              "  'end': 1204},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.51953053,\n",
              "  'index': 327,\n",
              "  'word': '##ric',\n",
              "  'start': 1204,\n",
              "  'end': 1207},\n",
              " {'entity': 'I-Biological_structure',\n",
              "  'score': 0.84212273,\n",
              "  'index': 328,\n",
              "  'word': '##ular',\n",
              "  'start': 1207,\n",
              "  'end': 1211},\n",
              " {'entity': 'B-Diagnostic_procedure',\n",
              "  'score': 0.74838465,\n",
              "  'index': 329,\n",
              "  'word': 's',\n",
              "  'start': 1212,\n",
              "  'end': 1213},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9006081,\n",
              "  'index': 330,\n",
              "  'word': '##ys',\n",
              "  'start': 1213,\n",
              "  'end': 1215},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9443973,\n",
              "  'index': 331,\n",
              "  'word': '##to',\n",
              "  'start': 1215,\n",
              "  'end': 1217},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.908581,\n",
              "  'index': 332,\n",
              "  'word': '##lic',\n",
              "  'start': 1217,\n",
              "  'end': 1220},\n",
              " {'entity': 'I-Diagnostic_procedure',\n",
              "  'score': 0.9073442,\n",
              "  'index': 333,\n",
              "  'word': 'function',\n",
              "  'start': 1221,\n",
              "  'end': 1229},\n",
              " {'entity': 'B-Lab_value',\n",
              "  'score': 0.6683436,\n",
              "  'index': 335,\n",
              "  'word': 'mildly',\n",
              "  'start': 1234,\n",
              "  'end': 1240},\n",
              " {'entity': 'I-Lab_value',\n",
              "  'score': 0.92343223,\n",
              "  'index': 336,\n",
              "  'word': 'impaired',\n",
              "  'start': 1241,\n",
              "  'end': 1249},\n",
              " {'entity': 'B-Severity',\n",
              "  'score': 0.9684105,\n",
              "  'index': 340,\n",
              "  'word': 'mild',\n",
              "  'start': 1264,\n",
              "  'end': 1268},\n",
              " {'entity': 'B-Sign_symptom',\n",
              "  'score': 0.47800407,\n",
              "  'index': 341,\n",
              "  'word': 'mit',\n",
              "  'start': 1269,\n",
              "  'end': 1272},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.6886209,\n",
              "  'index': 342,\n",
              "  'word': '##ral',\n",
              "  'start': 1272,\n",
              "  'end': 1275},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.69865555,\n",
              "  'index': 343,\n",
              "  'word': 're',\n",
              "  'start': 1276,\n",
              "  'end': 1278},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.7023645,\n",
              "  'index': 344,\n",
              "  'word': '##gu',\n",
              "  'start': 1278,\n",
              "  'end': 1280},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.7601678,\n",
              "  'index': 345,\n",
              "  'word': '##rg',\n",
              "  'start': 1280,\n",
              "  'end': 1282},\n",
              " {'entity': 'I-Sign_symptom',\n",
              "  'score': 0.6566562,\n",
              "  'index': 346,\n",
              "  'word': '##itation',\n",
              "  'start': 1282,\n",
              "  'end': 1289}]"
            ]
          },
          "metadata": {},
          "execution_count": 32
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "def create_token_table(sentence):\n",
        "    sentence = sentence.replace(\"\\u2005\", \"\").replace(\"\\u200a\", \"\").replace(\"\\u2009\", \"\").replace(\"\\n\", \"\")\n",
        "    tokens = sentence.split()\n",
        "\n",
        "    token_table = []\n",
        "    start = 0\n",
        "\n",
        "    for token in tokens:\n",
        "        end = start + len(token)\n",
        "        token_info = {\n",
        "            \"token\": token,\n",
        "            \"start\": start,\n",
        "            \"end\": end\n",
        "        }\n",
        "        token_table.append(token_info)\n",
        "\n",
        "        start = end + 1\n",
        "\n",
        "    return token_table"
      ],
      "metadata": {
        "id": "0EqTlC3BU9Ax"
      },
      "execution_count": 33,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Originally, tags were fulfilled with a low probability. However, since matching was more difficult than expected, this threshold was removed again."
      ],
      "metadata": {
        "id": "R9MnljN8CeHs"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def preprocess_entities(sentence_info):\n",
        "    #sentence_info = [info for info in sentence_info if info['score'] >= 0.4]\n",
        "    cleaned_entities = []\n",
        "    current_entity = None\n",
        "    current_start = None\n",
        "    current_end = None\n",
        "\n",
        "    for info in sentence_info:\n",
        "        entity = info['entity']\n",
        "        start = info['start']\n",
        "        end = info['end']\n",
        "\n",
        "        if current_entity is None:\n",
        "            current_entity = entity.replace('B-', '').replace('I-', '')\n",
        "            current_start = start\n",
        "            current_end = end\n",
        "        elif current_end == start:\n",
        "            current_entity = entity.replace('B-', '').replace('I-', '')\n",
        "            current_end = end\n",
        "        else:\n",
        "            cleaned_entities.append((current_entity, current_start, current_end))\n",
        "            current_entity = entity.replace('B-', '').replace('I-', '')\n",
        "            current_start = start\n",
        "            current_end = end\n",
        "\n",
        "    if current_entity is not None:\n",
        "        cleaned_entities.append((current_entity, current_start, current_end))\n",
        "\n",
        "    return cleaned_entities"
      ],
      "metadata": {
        "id": "_GxFH_10VANt"
      },
      "execution_count": 34,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def generate_tags(predicted_entities, sentence_entities):\n",
        "    tags = [\"O\"] * len(sentence_entities)\n",
        "\n",
        "    for entity, start, end in predicted_entities:\n",
        "        for i, token_info in enumerate(sentence_entities):\n",
        "            token_start = token_info['start']\n",
        "            token_end = token_info['end']\n",
        "\n",
        "            if token_start == start and token_end == end:\n",
        "                tags[i] = entity\n",
        "    return tags"
      ],
      "metadata": {
        "id": "XUYZ6oqJVB8Y"
      },
      "execution_count": 35,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def format_tags(tags):\n",
        "    formatted_tags = []\n",
        "    current_entity = None\n",
        "\n",
        "    for tag in tags:\n",
        "        if tag == 'O':\n",
        "            formatted_tags.append(tag)\n",
        "            current_entity = None\n",
        "        else:\n",
        "            entity_type = tag\n",
        "            if current_entity == entity_type:\n",
        "                formatted_tags.append('I-' + entity_type)\n",
        "            else:\n",
        "                formatted_tags.append('B-' + entity_type)\n",
        "                current_entity = entity_type\n",
        "\n",
        "    return formatted_tags"
      ],
      "metadata": {
        "id": "hcqjANrpVEA6"
      },
      "execution_count": 36,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def get_sentence_tags(sentence):\n",
        "  token_table = create_token_table(sentence)\n",
        "  sentence_info = predict_tags(sentence)\n",
        "  predicted_entities = preprocess_entities(sentence_info)\n",
        "  general_tags = generate_tags(predicted_entities, token_table)\n",
        "  return general_tags"
      ],
      "metadata": {
        "id": "fquq-LLnVG0n"
      },
      "execution_count": 37,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "# Turn entities into values\n",
        "\n",
        "Now that every function is defined, we can start with making predictions."
      ],
      "metadata": {
        "id": "skR6_z3wWKtg"
      }
    },
    {
      "cell_type": "code",
      "execution_count": 38,
      "metadata": {
        "id": "DnRyLacvUDFH",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "82b91f2b-38db-4e99-e60e-d08ccfbcfb79"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "[nltk_data] Downloading package punkt to /root/nltk_data...\n",
            "[nltk_data]   Unzipping tokenizers/punkt.zip.\n"
          ]
        },
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "True"
            ]
          },
          "metadata": {},
          "execution_count": 38
        }
      ],
      "source": [
        "from nltk import tokenize\n",
        "import nltk\n",
        "from transformers import AutoTokenizer, AutoModel\n",
        "import torch\n",
        "import numpy as np\n",
        "nltk.download('punkt')"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "descrip_1 = list(patient_dict.values())[0]\n",
        "list_tags = get_sentence_tags(descrip_1)\n",
        "list_descrip = descrip_1.split()"
      ],
      "metadata": {
        "id": "bUTZkROcUPF3"
      },
      "execution_count": 39,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "result_dict = {}\n",
        "for key, value in zip(list_tags, list_descrip):\n",
        "    if key in result_dict:\n",
        "        result_dict[key].append(value)\n",
        "    else:\n",
        "        result_dict[key] = [value]"
      ],
      "metadata": {
        "id": "5IFzfOuQUQq2"
      },
      "execution_count": 40,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "if 'O' in result_dict:\n",
        "    del result_dict['O']"
      ],
      "metadata": {
        "id": "admPBDzUVe_4"
      },
      "execution_count": 41,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "result_dict"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "6Fc5otAeDbPO",
        "outputId": "43c1c58d-e504-4ec5-fe9e-5ddcea8a1589"
      },
      "execution_count": 42,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "{'Age': ['58-year-old'],\n",
              " 'Personal_background': ['African-American'],\n",
              " 'Sex': ['woman'],\n",
              " 'Clinical_event': ['presents'],\n",
              " 'Nonbiological_location': ['ER'],\n",
              " 'Detailed_description': ['episodic', 'pressing/burning'],\n",
              " 'Biological_structure': ['anterior', 'chest'],\n",
              " 'Sign_symptom': ['pain', 'pain', 'diaphoresis', 'increased', 'pain'],\n",
              " 'Date': ['two', 'days', 'earlier', 'half', 'an', 'prior'],\n",
              " 'Duration': ['time'],\n",
              " 'Severity': ['mild'],\n",
              " 'Time': ['hour'],\n",
              " 'History': ['hypertension', 'history', 'of', 'heart'],\n",
              " 'Family_history': ['family'],\n",
              " 'Diagnostic_procedure': ['Physical', 'examination', 'EKG']}"
            ]
          },
          "metadata": {},
          "execution_count": 42
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "To turn the entities into embeddings, we use BERT finetuned with Medical Texts (https://huggingface.co/pritamdeka/S-PubMedBert-MS-MARCO)."
      ],
      "metadata": {
        "id": "pRA24dtL5pqF"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "model_name = \"pritamdeka/S-PubMedBert-MS-MARCO\"\n",
        "tokenizer_embeddings = AutoTokenizer.from_pretrained(model_name)\n",
        "model_embeddings = AutoModel.from_pretrained(model_name)"
      ],
      "metadata": {
        "id": "kTIRb3TQWXFu",
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 208,
          "referenced_widgets": [
            "2adbe4d14ed448a0876bd84baafcd69d",
            "5f0217de9a9b4c6e835b381486ccde2d",
            "d4d69709633c4a19acf28f67f962e474",
            "0ee115186e5649528ff3a5605192bc7f",
            "0b6714ad75dd4357a8a04255623b72e5",
            "6ef98027aed44d6ba192340ecec65afc",
            "c6e3c22c52484dc78e4adda75f6e2dba",
            "200be19af1914804a4e5bbfbe909fa98",
            "da68447cdf9240ce8a00776c7b7e7903",
            "bf79122cfd2542d6b07c309df474abf9",
            "2bd15504a02b4fc7b674ca3edebaa0df",
            "d9ca7cbc1cbb49e398e615693dca2101",
            "2d5a80732fd94d4bb5715c621be15a79",
            "b4e91a1487224a3787c8c4b7d8ef9d66",
            "992b3db6422545b0b778faba8b699a9b",
            "18fa4473f5eb4679bef7f63ee10d4817",
            "0853b43c5a3e437395d0af21616ba644",
            "145fc556fd0e4853948908dabf03ee22",
            "d93be8f2eb9846e1afecfc0bd00c8839",
            "bc96ccb3f010463facae4a9589850b65",
            "1c51d8d7cf5b4a7dbec9a9e4e37668de",
            "5ad1872df2bb4f6eb4fd476c1aa16122",
            "c13124e6c8d14f319b372ff1621ea883",
            "a21b9c1a85574673a8fd0f5dbd111f15",
            "0d4b2b50d7cb4467988f3375b66c192c",
            "7aa5fdb782a44a22b6d2219e5bcff354",
            "093b83161a8448dd8f0f4c3da3bbbe8f",
            "29ca8221217748e68ce1588ee3f637a9",
            "9c4256f00a6a4a8a85e20ddb53647424",
            "47af947481bc48aeb6ca2e0dcfdf6312",
            "4725600d7cb746589a140c1fe68bc04b",
            "9a0d5627edb5415c827ffd4ffdcb5a08",
            "25d5b69cf04649aca3341f92b3327402",
            "80a3951b0e644e0396a0f3322fc20cd3",
            "1ee02de646534e77b681b2e3af2a8206",
            "7a2139c45fc74306ad140dc65ca5b992",
            "4016fbd574374622993d7b5ba90fe958",
            "9aa801fa467745edb38c140436395158",
            "8cc21f9f17de41bbb2a1bb79f16fadc8",
            "72d302db18e74221be6c3518746ef20a",
            "dee076e04d0344a5a4436e925d58b797",
            "d94934f5e2ac416494c7c85a089fdb7a",
            "f29e903fc6c24ca9ae2dcb1b84fb68f0",
            "4f58dab11b364d2c8da3e4a3cdf1b266",
            "2cdc68bdf3ff47a8b0e29b94f5bd236b",
            "bb49a02a3ed5496cb8251cfc0f06ea1e",
            "6bb13ea7aac84d399e34c8ed816d0c5b",
            "668e815446ef483ea4839a8a8cc52937",
            "d8e89bc18f284b43ab1261131ea2fdd8",
            "e4f58c83b5f4402687aa8fd5e0d40409",
            "4d11321f627b472f9862c719d0e184c0",
            "f431c96e37ea48f398a28be37e644c3c",
            "cd3fee27ee5e48a68fffb8c343eb305e",
            "843197ef31dd477d8a41876f4b212dec",
            "c9199f83ce9444d09f8a9782aa0a99c4",
            "d1ab6716f79745b28c1e2acb3fa2a727",
            "46e5c27000664e9299edea05c0f4b1a7",
            "18f2040a12eb4e47a09b52a0b5dd9503",
            "5501a73549e84c9b8c519934c13db6a2",
            "74c25ffb54b64cee8d77f7752c2afa83",
            "55580def279f4c6dabf56e2c01fd162a",
            "80410cabadd24cf398a027fc19c6655a",
            "36333367d23f4d348836017cf842f76a",
            "1ee7e50769c843eabe1b82c9749d2115",
            "b695b93e8a664e74abb36fd3074c1d57",
            "17cf40348c0849a7a10b0c852a751600"
          ]
        },
        "outputId": "4fccce88-6010-490e-f8bc-0808d27240e8"
      },
      "execution_count": 43,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading (…)okenizer_config.json:   0%|          | 0.00/388 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "2adbe4d14ed448a0876bd84baafcd69d"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading (…)solve/main/vocab.txt:   0%|          | 0.00/226k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "d9ca7cbc1cbb49e398e615693dca2101"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading (…)/main/tokenizer.json:   0%|          | 0.00/461k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "c13124e6c8d14f319b372ff1621ea883"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading (…)cial_tokens_map.json:   0%|          | 0.00/112 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "80a3951b0e644e0396a0f3322fc20cd3"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading (…)lve/main/config.json:   0%|          | 0.00/667 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "2cdc68bdf3ff47a8b0e29b94f5bd236b"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading pytorch_model.bin:   0%|          | 0.00/438M [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "d1ab6716f79745b28c1e2acb3fa2a727"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "if torch.cuda.is_available():\n",
        "    device = torch.device(\"cuda\")\n",
        "else:\n",
        "    device = torch.device(\"cpu\")\n",
        "\n",
        "model_embeddings = model_embeddings.to(device)"
      ],
      "metadata": {
        "id": "5i8eKj8mjJk4"
      },
      "execution_count": 44,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "With `create_word_vectors`, every entity cluster will be turned into one vector."
      ],
      "metadata": {
        "id": "7F_N8-QcDjp9"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def create_word_vectors(texts, tokenizer=tokenizer_embeddings, model=model_embeddings, device=device):\n",
        "    embeddings = []\n",
        "    for text in texts:\n",
        "        tokens = tokenizer(text, padding=True, truncation=True, return_tensors=\"pt\").to(device)\n",
        "        with torch.no_grad():\n",
        "            model = model.to(device)\n",
        "            output = model(**tokens)\n",
        "            embeddings.append(output.last_hidden_state.mean(dim=1).squeeze().cpu().numpy())\n",
        "    return embeddings"
      ],
      "metadata": {
        "id": "f1dg7_EXkd0C"
      },
      "execution_count": 45,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "First, the patient descriptions will be turned into a semantic vector by applying the above defined function."
      ],
      "metadata": {
        "id": "QmHoX0neWvlM"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "vector_dict_patient = {}\n",
        "for key, values in result_dict.items():\n",
        "    vectors = create_word_vectors(values)\n",
        "    if vectors:\n",
        "        averaged_vector = np.mean(vectors, axis=0)\n",
        "        vector_dict_patient[key] = averaged_vector\n",
        "\n",
        "print(vector_dict_patient)"
      ],
      "metadata": {
        "id": "t3w1TRTVWvNu",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "aff2c29d-ba0e-4081-85dd-5454f3b156f7"
      },
      "execution_count": 46,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Asking to truncate to max_length but no maximum length is provided and the model has no predefined maximum length. Default to no truncation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "{'Age': array([-4.48633522e-01, -5.57422221e-01, -6.25807941e-01, -6.50893092e-01,\n",
            "       -2.29620993e-01, -1.82942122e-01, -6.47665381e-01,  8.48993361e-01,\n",
            "       -1.27741784e-01, -2.12214768e-01,  2.88278669e-01,  5.50653994e-01,\n",
            "        1.88696459e-01, -1.27048090e-01, -1.07246256e+00,  4.82443459e-02,\n",
            "        2.16187134e-01,  3.06083590e-01, -7.88488537e-02,  7.07307756e-01,\n",
            "        5.30802667e-01,  8.10717225e-01, -1.98618099e-01, -1.29397362e-01,\n",
            "       -2.42943540e-01,  7.89049119e-02, -1.42334998e-01,  2.71630615e-01,\n",
            "        6.71719909e-01,  4.65820253e-01,  1.33993447e-01,  3.55350077e-01,\n",
            "       -5.45750260e-01,  9.65225875e-01,  1.82438582e-01, -2.94252276e-01,\n",
            "        1.23208717e-01, -2.88575411e-01,  3.93170893e-01, -8.43763500e-02,\n",
            "        2.91413993e-01,  2.67682046e-01,  9.93976220e-02, -1.12106480e-01,\n",
            "       -9.48969275e-02,  6.60806119e-01, -1.08379729e-01, -5.39819002e-01,\n",
            "        1.16810776e-01, -1.40077382e-01, -1.92926452e-01, -1.13017648e-01,\n",
            "        1.20609246e-01,  3.88935059e-02, -3.47747713e-01, -1.49967372e-01,\n",
            "       -2.09349785e-02,  1.94212750e-01, -5.59273899e-01,  3.86523418e-02,\n",
            "       -4.04717624e-01,  3.42481315e-01, -1.95857555e-01, -8.39294419e-02,\n",
            "        2.41877347e-01, -1.58531964e-01, -2.39683986e-01, -6.49351597e-01,\n",
            "       -4.75347281e-01, -4.84792367e-02, -1.35228902e-01,  2.48948455e-01,\n",
            "        3.00307184e-01, -1.37642071e-01,  9.79602113e-02, -2.68712103e-01,\n",
            "        9.89452451e-02, -6.13525808e-02, -2.81012543e-02, -3.55495363e-01,\n",
            "       -3.85543257e-02,  2.03999087e-01, -2.80654043e-01, -4.75474983e-01,\n",
            "       -2.66524702e-01, -3.54372472e-01,  2.59203404e-01,  6.37339577e-02,\n",
            "       -2.88414598e-01,  1.19719394e-01, -1.59508765e-01, -4.59613502e-01,\n",
            "       -2.54821986e-01, -2.97886759e-01, -7.04868138e-02, -1.67555422e-01,\n",
            "       -1.42002642e-01, -1.15110539e-02,  1.31047532e-01, -3.86549205e-01,\n",
            "        1.05021641e-01,  5.13716228e-02,  3.88298929e-02, -2.76856869e-01,\n",
            "       -8.16228017e-02, -6.26798630e-01,  2.42332876e-01,  3.51241708e-01,\n",
            "       -2.41591513e-01, -6.13621064e-02, -1.07717678e-01, -7.06832528e-01,\n",
            "        4.25605476e-01, -5.86694419e-01,  5.68033643e-02, -1.74087316e-01,\n",
            "        6.45520866e-01,  1.69120222e-01,  5.02394773e-02,  4.84045267e-01,\n",
            "        3.88996154e-01, -4.14165288e-01, -1.00246876e-01, -1.74915716e-01,\n",
            "        2.57601291e-01, -3.32156539e-01,  6.19499624e-01, -3.24038208e-01,\n",
            "        2.56839275e-01,  7.13806212e-01,  1.36942551e-01,  1.86762303e-01,\n",
            "       -5.69413900e-01, -2.60016769e-01, -1.57817647e-01,  4.15672868e-01,\n",
            "       -1.55928180e-01, -2.14399740e-01,  3.71823758e-01,  2.12169945e-01,\n",
            "        2.70188361e-01, -3.58712792e-01, -2.86715388e-01, -7.86934793e-02,\n",
            "       -1.05967805e-01, -1.80954635e-01, -9.25149620e-02, -3.13925177e-01,\n",
            "       -6.48829281e-01, -2.32564896e-01, -4.14991826e-01, -5.69085300e-01,\n",
            "        6.48851812e-01, -1.34646267e-01,  2.72624016e-01, -3.35685819e-01,\n",
            "        4.36561108e-01, -3.41026098e-01,  2.60867536e-01, -2.54173670e-02,\n",
            "        3.31945658e-01, -3.18427950e-01,  6.57235503e-01, -8.88331383e-02,\n",
            "        4.88568619e-02,  4.79817763e-02,  3.91759247e-01, -1.41909301e-01,\n",
            "        1.44742638e-01,  6.55900598e-01,  5.42888157e-02, -1.51776463e-01,\n",
            "        3.93182814e-01, -2.40871251e-01, -1.40936568e-01, -3.78728355e-03,\n",
            "       -8.74351524e-03, -3.69594246e-01, -1.26828447e-01,  5.87230086e-01,\n",
            "        3.29454839e-01, -1.39454758e+00, -4.63709205e-01, -5.12296975e-01,\n",
            "       -6.56829357e-01, -5.85327744e-01, -8.56457800e-02, -3.37496027e-02,\n",
            "        4.08806354e-01, -5.52485101e-02, -3.93645316e-01, -8.25331435e-02,\n",
            "       -2.76518852e-01, -1.48040369e-01, -8.27965662e-02,  2.34969452e-01,\n",
            "        5.14665127e-01,  4.13146526e-01, -1.99327305e-01,  2.22251594e-01,\n",
            "       -3.66798311e-01,  1.76834211e-01,  2.16330230e-01,  3.07345092e-01,\n",
            "       -3.54555488e-01,  2.97481745e-01, -3.72646272e-01,  3.57793272e-01,\n",
            "        1.48975998e-01, -2.44631827e-01, -3.18011016e-01, -1.95185438e-01,\n",
            "       -6.24323010e-01,  1.00005597e-01, -5.56971841e-02, -4.22279015e-02,\n",
            "       -2.81456143e-01, -2.14748517e-01,  4.67964768e-01, -1.42005950e-01,\n",
            "       -2.30167076e-01, -9.25473794e-02, -2.97058672e-01, -6.02706254e-01,\n",
            "        6.90520346e-01,  1.16409853e-01, -1.10876761e-01,  3.52979273e-01,\n",
            "        3.35029177e-02, -1.70866981e-01,  2.78172940e-01, -6.79602623e-02,\n",
            "        2.68391788e-01,  2.02002391e-01, -2.05884740e-01, -3.24503273e-01,\n",
            "       -2.53779650e-01,  5.55332661e-01, -1.07955649e-01, -7.12347552e-02,\n",
            "       -3.21225256e-01, -3.18498105e-01, -4.54095036e-01,  5.93536273e-02,\n",
            "        4.64713722e-01, -2.05139518e-02,  2.15803325e-01,  5.85648298e-01,\n",
            "        5.88494241e-01,  4.55665916e-01,  1.74447045e-01,  2.94153601e-01,\n",
            "        2.46572420e-01, -5.80743179e-02, -7.44962990e-01, -2.63995022e-01,\n",
            "       -2.96324808e-02, -2.42643446e-01,  1.67620391e-01, -3.57197225e-01,\n",
            "       -1.58620998e-02, -1.56122774e-01, -1.14659760e-02,  4.16212827e-01,\n",
            "       -1.46572649e-01, -6.64108694e-02,  3.95962089e-01, -2.12196901e-01,\n",
            "       -5.23753576e-02,  5.29459953e-01, -1.04503088e-01,  5.82980871e-01,\n",
            "        5.84231457e-03, -3.93378854e-01, -5.78170955e-01, -5.46892583e-01,\n",
            "        2.58537233e-01,  1.21460203e-02, -5.51225916e-02, -8.76633823e-01,\n",
            "       -2.33193845e-01, -2.30494469e-01,  6.01652898e-02, -4.49987501e-01,\n",
            "       -4.34125625e-02,  3.74875516e-01, -9.73468721e-02, -2.57091552e-01,\n",
            "       -1.74810261e-01,  2.20608652e-01, -3.81518513e-01,  1.12728454e-01,\n",
            "       -2.79114515e-01,  1.88281566e-01,  2.82465219e-01,  4.33873646e-02,\n",
            "        2.16935828e-01,  2.71033436e-01, -1.86503455e-02, -1.03403730e-02,\n",
            "        2.73152322e-01,  3.70239556e-01,  3.93713303e-02, -3.30473274e-01,\n",
            "       -1.91308528e-01,  1.24450594e-01, -6.98245615e-02,  5.25255680e-01,\n",
            "       -4.26596850e-01, -3.02774072e-01, -3.73191386e-02,  7.12268949e-01,\n",
            "       -3.79971117e-01,  2.90493760e-02, -8.20190683e-02, -1.06533933e+00,\n",
            "       -2.19533131e-01, -4.59325343e-01, -2.24625126e-01, -4.83484626e-01,\n",
            "        3.32655519e-01,  6.44793630e-01,  8.77790302e-02, -2.24700496e-01,\n",
            "       -3.90663683e-01,  5.42559385e-01,  1.08815002e+00, -2.17364281e-01,\n",
            "        1.13188609e-01, -2.17002705e-01, -1.70110613e-01, -2.95943260e-01,\n",
            "        5.32026635e-03, -2.96891063e-01,  4.67480421e-01,  4.85664427e-01,\n",
            "        3.23299766e-01, -2.29448929e-01,  6.12030700e-02, -3.00004601e-01,\n",
            "        5.02888083e-01,  2.43638441e-01,  3.05909384e-02,  5.18405139e-01,\n",
            "       -4.81188476e-01,  4.49489981e-01,  1.60223708e-01, -1.25148579e-01,\n",
            "       -7.94886202e-02,  3.54024380e-01,  3.23432803e-01, -5.12003362e-01,\n",
            "       -7.74925351e-01, -1.18616000e-01,  1.05676010e-01,  5.70122659e-01,\n",
            "        4.80318367e-01, -2.49805540e-01,  4.12736088e-01,  1.64776623e-01,\n",
            "       -3.77731472e-01,  1.27091512e-01, -1.37242243e-01, -4.26481813e-02,\n",
            "        8.29230696e-02,  4.35599625e-01,  2.05172285e-01, -1.47495463e-01,\n",
            "       -3.40133309e-01, -3.76019701e-02, -7.20730983e-03, -1.73476115e-01,\n",
            "       -1.31330878e-01, -2.62411889e-02,  4.02724057e-01, -1.53599471e-01,\n",
            "       -5.15895605e-01,  3.81137282e-01, -1.89739957e-01, -8.61485824e-02,\n",
            "        2.65447706e-01,  3.45231332e-02,  4.30164248e-01, -2.71844864e-01,\n",
            "       -3.09380770e-01, -4.90219221e-02,  1.45462140e-01,  1.48916483e-01,\n",
            "        3.00331205e-01, -1.06667203e-03,  1.44345919e-02, -4.43530113e-01,\n",
            "       -2.60388315e-01,  8.00703466e-01, -3.07635874e-01, -3.09319079e-01,\n",
            "        4.63583648e-01, -2.74080664e-01,  2.02858448e-02,  5.65542221e-01,\n",
            "       -4.18821484e-01, -4.52708066e-01, -1.76083654e-01,  3.52097631e-01,\n",
            "        7.40390643e-02, -2.23894373e-01, -2.15822980e-01, -1.41840205e-01,\n",
            "       -5.49488142e-02,  1.90942675e-01,  4.58094537e-01,  2.52027363e-01,\n",
            "        1.52976617e-01, -1.79761782e-01,  1.03208460e-02,  1.82077557e-01,\n",
            "       -3.00850034e-01, -4.23337594e-02, -1.31804749e-01, -1.10538498e-01,\n",
            "        3.74611944e-01,  2.23534897e-01, -2.82732815e-01,  4.04834539e-01,\n",
            "       -1.14791813e+01,  3.99177670e-01,  3.85767370e-01, -1.81277409e-01,\n",
            "        9.72939953e-02, -1.91119872e-02, -5.61935194e-02, -3.82798821e-01,\n",
            "        1.80826172e-01, -1.79166228e-01, -1.18032828e-01, -6.03263490e-02,\n",
            "       -2.28242785e-01, -2.32154965e-01, -5.33489883e-01,  1.47520334e-01,\n",
            "        4.08855021e-01, -2.54136741e-01, -5.20654507e-02, -2.00187892e-01,\n",
            "       -4.97103706e-02, -1.47574544e-01, -4.13921475e-02,  4.20727819e-01,\n",
            "       -3.95565242e-01,  1.20853484e-01, -4.75921184e-01,  9.46011022e-02,\n",
            "        7.05795875e-03,  1.01865634e-01, -2.78384507e-01, -1.42943770e-01,\n",
            "       -1.90996692e-01, -4.27340297e-03,  5.55985495e-02, -3.76597375e-01,\n",
            "       -3.48448046e-02, -2.26451352e-01, -9.12140086e-02, -3.79615687e-02,\n",
            "       -1.32270902e-01, -5.77389784e-02, -1.85417831e-01,  6.95824698e-02,\n",
            "        3.75065714e-01,  7.10927188e-01,  3.24322790e-01,  4.67923790e-01,\n",
            "        3.94934505e-01,  3.32024060e-02, -5.40759042e-02, -8.28496695e-01,\n",
            "       -3.71790584e-03, -7.43039668e-01, -1.96070284e-01, -2.16428250e-01,\n",
            "        2.91990101e-01, -4.40602303e-01,  2.50422627e-01, -4.80595499e-01,\n",
            "       -2.44005412e-01,  1.35152385e-01,  8.33977908e-02,  8.83602649e-02,\n",
            "       -6.46735013e-01, -3.12336147e-01,  2.01355129e-01, -4.22157533e-02,\n",
            "        1.87228307e-01,  1.86957031e-01,  2.61119217e-01,  1.65560246e-01,\n",
            "       -1.07465582e-02,  3.51299858e-03, -6.00933544e-02, -2.62005299e-01,\n",
            "       -2.39014357e-01, -9.43214595e-02,  2.20336342e+00,  1.16050439e-02,\n",
            "        4.03494328e-01,  9.62852966e-03, -2.33023062e-01, -1.84761614e-01,\n",
            "       -6.80104434e-01, -1.45580433e-02, -7.28513226e-02, -4.32398587e-01,\n",
            "        4.89858091e-01,  4.30620074e-01, -1.85727432e-01, -1.06730387e-01,\n",
            "       -3.22346613e-02,  1.88565046e-01,  3.61409247e-01,  5.74580193e-01,\n",
            "       -6.56786859e-01,  2.56110549e-01, -1.56001091e-01,  4.10489857e-01,\n",
            "       -3.39762241e-01,  2.61068463e-01, -3.17685008e-01, -5.07257402e-01,\n",
            "       -3.61039609e-01, -5.63184321e-01, -9.20619890e-02,  2.52407491e-01,\n",
            "        2.16258943e-01,  3.97854209e-01, -4.14346784e-01, -2.75992572e-01,\n",
            "        3.34827036e-01, -5.46876788e-02, -3.05129617e-01, -3.67553741e-01,\n",
            "        3.72403204e-01,  2.06281915e-01, -4.02931780e-01,  1.12070777e-02,\n",
            "        6.93210959e-02,  1.24797523e-01,  5.01439236e-02, -1.30406365e-01,\n",
            "       -4.60032284e-01, -3.93585175e-01, -5.65626502e-01,  2.80327529e-01,\n",
            "        3.11977535e-01,  2.57211506e-01,  7.75202930e-01,  9.85526145e-02,\n",
            "        1.63330197e-01, -7.47593865e-02, -1.93457752e-02, -2.47060731e-01,\n",
            "       -7.02279449e-01,  2.38785759e-01,  6.03729248e-01, -1.79640919e-01,\n",
            "       -8.84617046e-02, -2.48485044e-01, -8.86340067e-03,  9.33389887e-02,\n",
            "        2.24339440e-01,  4.94053848e-02,  2.14057133e-01,  3.23743254e-01,\n",
            "       -2.44612262e-01,  1.12378597e-01, -1.06772497e-01, -1.68238148e-01,\n",
            "        3.25110145e-02, -2.58543819e-01, -1.07627576e-02, -7.25593388e-01,\n",
            "       -5.41868865e-01, -6.95664585e-02, -9.04661044e-02, -2.15002850e-01,\n",
            "        1.69545963e-01, -2.83632845e-01, -3.18445385e-01, -6.40374005e-01,\n",
            "        4.18592542e-01,  1.28965691e-01,  1.49063691e-01, -1.67507336e-01,\n",
            "        5.66805480e-03, -1.43061817e-01,  8.30798522e-02,  6.30212545e-01,\n",
            "       -2.11347356e-01, -4.78116542e-01, -9.37204421e-01, -1.93069920e-01,\n",
            "       -3.67170602e-01,  8.98217261e-01, -3.19701254e-01, -3.84734660e-01,\n",
            "        4.07744467e-01,  5.89326859e-01, -1.52901541e-02,  1.93315312e-01,\n",
            "       -9.33393650e-03,  3.48273158e-01, -5.16305685e-01, -3.04532528e-01,\n",
            "        4.39314514e-01,  2.14515422e-02,  5.84600806e-01,  1.23783201e-01,\n",
            "        2.75735676e-01,  9.85424072e-02,  3.84256512e-01,  2.45255455e-02,\n",
            "       -1.98670372e-01,  2.56633192e-01, -1.55594811e-01,  4.70094055e-01,\n",
            "        2.06715450e-01, -3.14363807e-01, -2.76753545e-01, -1.92449227e-01,\n",
            "        8.81374106e-02, -2.67719746e-01,  5.19906878e-01,  8.43227983e-01,\n",
            "       -2.12645575e-01,  5.84650831e-03,  9.37482893e-01,  1.62161648e-01,\n",
            "        4.05637205e-01,  2.69975752e-01,  4.16469723e-01,  2.26568222e-01,\n",
            "        5.01175046e-01, -3.21366012e-01, -5.20464897e-01, -1.37472779e-01,\n",
            "        1.40462995e-01,  8.96289885e-01, -4.42510039e-01, -8.59418064e-02,\n",
            "       -1.91096827e-01,  7.63250589e-01,  1.12818450e-01,  3.13159138e-01,\n",
            "        3.10592800e-01,  4.73737538e-01,  3.32675099e-01, -5.49989581e-01,\n",
            "        3.25710863e-01,  5.67354500e-01,  1.60860151e-01, -1.30830467e-01,\n",
            "       -3.07710588e-01, -1.32562384e-01, -2.52063692e-01, -1.73718423e-01,\n",
            "       -1.11538261e-01,  1.27709597e-01, -7.25213826e-01,  5.24739772e-02,\n",
            "        6.32865906e-01, -7.69645989e-01, -3.50813493e-02,  8.08128119e-02,\n",
            "       -1.86018094e-01,  8.60690176e-02,  1.57031029e-01,  1.40293404e-01,\n",
            "        1.68738633e-01, -2.44348690e-01, -4.23132062e-01,  1.87414214e-01,\n",
            "        4.71066415e-01,  5.67361005e-02, -1.43290669e-01, -2.27158487e-01,\n",
            "       -2.24558771e-01, -2.20165342e-01, -3.39365125e-01, -2.48558342e-01,\n",
            "        3.18173051e-01, -2.30273664e-01, -1.38835358e-02, -1.20124638e-01,\n",
            "        7.18381628e-02,  4.14602965e-01,  5.37968893e-03,  3.00998896e-01,\n",
            "        4.90406781e-01,  7.98197925e-01,  1.01428173e-01, -1.51375562e-01,\n",
            "        9.48397294e-02,  2.13610858e-01, -1.68495402e-01,  8.25394988e-01,\n",
            "        3.08393296e-02, -3.23659182e-02, -3.20630223e-01,  3.87138456e-01,\n",
            "       -2.63400972e-01,  7.86952153e-02,  2.21400321e-01,  1.14470348e-01,\n",
            "       -2.91541249e-01,  7.83236790e-03,  1.67233318e-01,  1.54021293e-01,\n",
            "        2.87380219e-01, -7.52287626e-01,  1.61777794e-01,  8.75403434e-02,\n",
            "       -5.87614849e-02,  6.35820746e-01, -1.37566954e-01,  6.71388686e-01,\n",
            "       -5.68491399e-01,  1.45103484e-01, -1.13522100e+00, -1.25285074e-01,\n",
            "        3.11989397e-01, -4.26644921e-01, -5.33318937e-01, -2.28838295e-01,\n",
            "        1.50360197e-01,  1.44945413e-01,  4.80045080e-01,  8.57253313e-01,\n",
            "        3.12886357e-01, -6.64175332e-01, -9.42407027e-02, -1.54231280e-01,\n",
            "       -3.03545237e-01, -1.44119844e-01,  4.44748193e-01,  6.55772686e-02,\n",
            "        2.84926414e-01, -4.56728816e-01, -3.61577183e-01,  7.17855543e-02,\n",
            "        1.17815629e-01, -1.28183112e-01,  8.24673951e-01,  9.69168022e-02,\n",
            "        2.84529150e-01,  4.70065922e-01,  1.82213649e-01, -1.79135546e-01,\n",
            "        6.19428992e-01, -7.11093703e-03, -4.21273053e-01,  4.01082903e-01,\n",
            "        9.04886201e-02, -1.75858170e-01,  1.38924688e-01, -7.79632747e-01,\n",
            "        1.20350555e-01, -8.72766450e-02,  6.13099515e-01, -4.08509150e-02],\n",
            "      dtype=float32), 'Personal_background': array([-4.75006312e-01, -8.56213748e-01, -9.75947201e-01, -3.17955136e-01,\n",
            "       -2.73713470e-02, -2.88194954e-01, -4.59507793e-01,  7.11282551e-01,\n",
            "        3.48237157e-01, -2.72304267e-01,  9.41637065e-03,  3.50204796e-01,\n",
            "       -7.71314427e-02, -3.39573741e-01, -3.83714736e-01, -7.37047521e-03,\n",
            "        3.24472375e-02,  2.75390297e-01,  9.51823890e-02,  7.26817623e-02,\n",
            "        3.78977835e-01,  3.66617918e-01, -2.14450121e-01, -7.57507607e-02,\n",
            "       -4.72485512e-01, -1.14541374e-01, -2.01666951e-01, -3.47324386e-02,\n",
            "        4.22900826e-01, -5.34434259e-01,  2.76262010e-03,  4.79826510e-01,\n",
            "       -2.62909830e-01,  8.56756210e-01,  4.72256914e-02, -2.75231063e-01,\n",
            "       -1.02545276e-01, -5.03664732e-01,  3.61031353e-01, -3.20867412e-02,\n",
            "       -7.88530558e-02,  2.36739188e-01,  4.99321371e-01, -1.04676917e-01,\n",
            "       -2.57141352e-01,  4.72885698e-01, -2.17440212e-03, -4.28606600e-01,\n",
            "        3.60639155e-01,  1.03724711e-01, -7.18157947e-01,  1.06168151e-01,\n",
            "        3.78200412e-01,  4.33508866e-02, -4.20238703e-01, -1.05620526e-01,\n",
            "        2.52781719e-01,  2.04026178e-01, -3.71407896e-01,  6.08151376e-01,\n",
            "       -3.49440426e-01,  5.98821938e-01, -1.41410351e-01,  1.19339675e-02,\n",
            "       -8.38849768e-02, -1.38907060e-01, -5.34197569e-01, -3.65708828e-01,\n",
            "       -5.07636666e-01,  1.71974525e-01, -3.53028387e-01,  3.70030850e-01,\n",
            "       -1.66805426e-03, -1.71996251e-01,  7.38659725e-02, -1.75681904e-01,\n",
            "        1.76093027e-01,  1.23585060e-01, -4.64517504e-01, -3.01645130e-01,\n",
            "        1.21911718e-02,  5.86738586e-01, -1.66968733e-01, -3.54045838e-01,\n",
            "       -2.80840069e-01,  6.25533462e-02, -9.61601734e-04,  8.43177959e-02,\n",
            "        2.82348305e-01,  2.46305153e-01, -2.68261619e-02, -2.62671918e-01,\n",
            "       -4.14766610e-01, -7.32145160e-02,  1.89835012e-01, -2.90366888e-01,\n",
            "        9.04184505e-02, -4.08170432e-01,  1.87407643e-01, -5.20799398e-01,\n",
            "        4.17323261e-02, -1.05065681e-01, -2.69195326e-02, -2.13809997e-01,\n",
            "       -3.78255367e-01, -3.78122479e-01,  1.43851146e-01, -1.52069390e-01,\n",
            "       -2.15163097e-01, -5.98183833e-02, -1.71015665e-01, -3.16872120e-01,\n",
            "        4.65690829e-02, -5.43246210e-01,  3.89454097e-01, -4.00906876e-02,\n",
            "        2.19944552e-01,  4.29837182e-02, -4.97688383e-01,  1.57027408e-01,\n",
            "        5.94522655e-01, -2.33299024e-02,  1.81938648e-01,  4.39048827e-01,\n",
            "        5.57666242e-01, -5.20675480e-02,  1.00277901e+00, -1.49661124e-01,\n",
            "        4.16083664e-01,  6.75360143e-01, -3.53408128e-01, -2.99608707e-01,\n",
            "       -8.24404180e-01, -3.80178481e-01,  9.69695449e-02,  1.39487326e-01,\n",
            "       -4.24344063e-01, -2.23798752e-01,  3.51440996e-01,  2.76442200e-01,\n",
            "       -4.37862396e-01, -1.20874636e-01, -3.73713255e-01, -2.29038578e-02,\n",
            "       -2.13639051e-01, -3.63308974e-02,  2.11851642e-01, -1.06262080e-01,\n",
            "        4.02562320e-02, -3.77055407e-01, -1.01898313e-02, -2.25046039e-01,\n",
            "        1.00113904e+00, -1.56104520e-01,  5.78485541e-02, -6.28726363e-01,\n",
            "        2.57623047e-01, -1.66562945e-01,  1.03332877e-01,  5.51949203e-01,\n",
            "        7.75726736e-01, -4.70543355e-01, -9.53909680e-02, -5.88071719e-02,\n",
            "       -2.30685361e-02,  8.81972015e-02, -2.95121521e-01, -1.37959540e-01,\n",
            "       -1.46973237e-01,  2.43696213e-01,  9.52072293e-02, -5.02455235e-01,\n",
            "        2.95302063e-01, -1.82630971e-01,  1.27158508e-01, -3.75862926e-01,\n",
            "        1.15038179e-01,  6.82710484e-03, -6.89430460e-02, -3.20715234e-02,\n",
            "        5.33038497e-01, -1.83808100e+00,  3.12912375e-01, -2.93408155e-01,\n",
            "        2.73007393e-01, -3.86166096e-01,  3.63357276e-01,  5.39592087e-01,\n",
            "        3.51619571e-01, -2.02897310e-01,  1.98096633e-01,  1.32960556e-02,\n",
            "       -3.50928754e-01, -6.57482967e-02, -1.85107455e-01,  2.80945688e-01,\n",
            "        3.28462452e-01,  1.50743559e-01,  5.24583995e-01,  4.58805896e-02,\n",
            "       -2.93342359e-02,  3.58316809e-01,  2.78165668e-01, -1.74642324e-01,\n",
            "        9.84041840e-02, -2.09512398e-01, -2.68782675e-01,  1.86917707e-01,\n",
            "       -2.61034817e-01,  1.44420668e-01, -1.26171440e-01, -6.06861673e-02,\n",
            "       -2.58235842e-01, -9.97015163e-02,  1.24746755e-01,  3.17541927e-01,\n",
            "       -4.52410132e-01, -6.83788002e-01,  3.34403634e-01, -1.03564458e-02,\n",
            "       -2.04495201e-03,  7.90248532e-03, -3.24156523e-01, -2.68948793e-01,\n",
            "       -1.33385897e-01,  3.19773734e-01, -4.17160034e-01,  1.43948200e-04,\n",
            "       -2.55220473e-01, -1.24762706e-01,  1.17778137e-01, -8.44534189e-02,\n",
            "        5.44601381e-01,  1.39401332e-01,  5.44915907e-02, -2.83168942e-01,\n",
            "       -3.11634928e-01,  2.98839986e-01,  8.25109184e-02,  3.29382271e-01,\n",
            "       -3.51771712e-01, -3.50374550e-01, -4.46747124e-01,  1.05279863e-01,\n",
            "       -8.17081705e-02,  3.62819545e-02, -3.07050142e-02,  6.37933984e-02,\n",
            "        4.71555382e-01,  4.28430706e-01,  4.76085395e-02, -7.45976642e-02,\n",
            "        1.90125331e-01,  1.77071124e-01, -3.35548967e-01, -6.30665049e-02,\n",
            "        1.44772723e-01, -1.79947376e-01,  5.21607958e-02, -7.56740034e-01,\n",
            "       -2.45048389e-01, -2.66545005e-02,  8.65936279e-04,  6.35370433e-01,\n",
            "       -5.02602272e-02,  1.72850773e-01, -1.93504050e-01,  2.20063731e-01,\n",
            "       -4.38897908e-01,  1.36469617e-01, -1.80749938e-01,  4.52538103e-01,\n",
            "       -6.91152215e-01,  2.10515842e-01, -6.93438947e-01, -4.25122470e-01,\n",
            "       -3.55584502e-01, -3.32829654e-02,  4.23885494e-01, -3.96966517e-01,\n",
            "       -1.43884331e-01, -5.65100789e-01,  1.21944904e-01, -3.68308395e-01,\n",
            "       -2.20074847e-01,  2.66823471e-01, -2.27994114e-01, -3.04623812e-01,\n",
            "       -1.61748350e-01, -2.32315585e-01, -7.37320662e-01, -8.02433640e-02,\n",
            "       -5.92142582e-01, -2.04029441e-01,  7.90817365e-02, -5.21426685e-02,\n",
            "       -9.18139145e-02,  1.70415267e-01, -1.00473322e-01, -2.34371468e-01,\n",
            "       -1.14752352e-03,  4.72695321e-01,  1.59683391e-01,  4.29171696e-02,\n",
            "        3.24132107e-02,  3.24254423e-01,  1.47078767e-01,  3.07994429e-02,\n",
            "       -6.43524006e-02, -3.26462477e-01,  7.48974755e-02,  4.73003864e-01,\n",
            "       -2.35243350e-01, -1.76897328e-02,  5.27447350e-02, -9.21280861e-01,\n",
            "        1.08609676e-01,  1.01872444e-01, -3.15242261e-01, -3.72064441e-01,\n",
            "        1.74287513e-01,  5.53315759e-01, -3.86935681e-01, -2.00480461e-01,\n",
            "       -2.73276716e-01,  4.89272207e-01,  1.19732571e+00, -1.76724076e-01,\n",
            "        5.14783040e-02, -5.52269876e-01, -1.46812722e-01, -4.28706616e-01,\n",
            "       -1.38333708e-01, -3.12565237e-01,  1.87297255e-01,  3.36261839e-01,\n",
            "       -1.01580620e-01,  1.35236740e-01, -5.56229018e-02, -2.23742679e-01,\n",
            "        4.25195426e-01,  4.12582792e-02, -5.71125448e-01,  2.87186831e-01,\n",
            "       -4.66403633e-01,  3.68374199e-01,  5.03572404e-01, -6.40813351e-01,\n",
            "        5.52709512e-02,  2.82228291e-01, -3.04063149e-02, -2.16806129e-01,\n",
            "       -4.99448866e-01, -7.22221315e-01,  9.33076799e-01,  5.64029992e-01,\n",
            "        3.56498390e-01,  1.05546035e-01,  7.79082894e-01,  2.41360590e-01,\n",
            "        7.08541200e-02,  2.39419505e-01, -3.65186989e-01, -6.57153904e-01,\n",
            "       -1.84599891e-01,  3.44187170e-01,  4.55168158e-01, -4.42204803e-01,\n",
            "        1.78041086e-01, -5.63390195e-01, -6.83481023e-02, -4.41020578e-01,\n",
            "       -8.14388320e-02,  1.21244289e-01,  3.84312004e-01, -3.55551213e-01,\n",
            "       -3.82215410e-01,  2.80895561e-01, -7.21791267e-01, -5.37633240e-01,\n",
            "       -1.82519421e-01,  1.21633425e-01,  1.53317720e-01, -2.59022620e-02,\n",
            "       -6.98411986e-02, -3.57798666e-01, -5.74521601e-01,  6.44825920e-02,\n",
            "       -2.41396036e-02,  2.79995412e-01,  1.08862400e-01, -3.28943819e-01,\n",
            "       -2.22946793e-01,  4.44727510e-01,  1.28197432e-01, -1.55303478e-01,\n",
            "        5.15732706e-01, -3.74464244e-01,  2.85599679e-01,  4.50726122e-01,\n",
            "       -3.76820415e-01, -5.93516052e-01, -6.79883182e-01,  4.72326666e-01,\n",
            "        4.17885870e-01, -3.75392705e-01, -2.43370533e-01,  2.90788412e-01,\n",
            "       -1.24098696e-01,  1.59511775e-01, -2.16240957e-01,  3.78695339e-01,\n",
            "       -4.35018927e-01, -3.53999138e-01, -2.85129845e-01,  1.16487421e-01,\n",
            "       -1.44459922e-02,  6.37156963e-02,  3.92379999e-01, -7.74437264e-02,\n",
            "        7.20699847e-01, -4.33683060e-02,  1.12085663e-01,  5.15260339e-01,\n",
            "       -1.11658278e+01,  2.96276510e-01,  2.66795158e-01, -1.72335535e-01,\n",
            "        5.75995624e-01, -3.87144297e-01,  2.09810540e-01, -3.03263754e-01,\n",
            "        2.49285027e-01, -6.36187851e-01,  1.36698512e-02,  2.51081496e-01,\n",
            "        3.75434905e-01,  4.31612246e-02, -4.32482302e-01, -3.57686609e-01,\n",
            "       -4.05800581e-01, -5.86744308e-01,  1.21910028e-01, -3.86623681e-01,\n",
            "        1.02967419e-01, -5.27502954e-01, -1.84230655e-01,  3.65662545e-01,\n",
            "        3.00384103e-03, -5.32844989e-03, -8.55476186e-02,  5.16124554e-02,\n",
            "        6.55161858e-01, -1.48059160e-01, -2.21475273e-01,  9.46372375e-02,\n",
            "       -2.47948542e-02,  2.44111091e-01, -5.53841479e-02, -6.42173588e-01,\n",
            "        1.62227437e-01, -1.17527865e-01,  6.55833423e-01, -1.86322957e-01,\n",
            "       -6.11793637e-01, -2.88855523e-01, -1.15789272e-01,  4.47117150e-01,\n",
            "        6.67179227e-01,  5.19037843e-01,  9.45845127e-01,  5.74757159e-01,\n",
            "        3.50794822e-01, -8.86365771e-02, -2.29199454e-01, -7.92365253e-01,\n",
            "        4.88541238e-02, -5.78620493e-01,  1.03270806e-01,  1.07880034e-01,\n",
            "        1.35036588e-01, -5.18778026e-01,  2.27590248e-01, -5.21632195e-01,\n",
            "        2.95825094e-01,  8.09925675e-01,  1.89017996e-01, -4.93796408e-01,\n",
            "       -6.69729471e-01, -8.92070979e-02,  4.35564816e-01, -1.77482933e-01,\n",
            "        2.16298819e-01, -3.39279413e-01, -2.55614400e-01,  1.33602664e-01,\n",
            "        5.44542611e-01,  7.26467967e-02,  4.85453427e-01,  1.47064209e-01,\n",
            "       -2.22908214e-01, -5.95793070e-04,  1.45388782e+00, -1.11917235e-01,\n",
            "        1.58339426e-01,  1.29802553e-02,  2.19510585e-01, -3.68886217e-02,\n",
            "       -4.30383295e-01, -2.47435402e-02, -4.24935907e-01, -4.91760164e-01,\n",
            "        9.69794020e-02,  6.18526578e-01, -2.49834687e-01,  1.04731873e-01,\n",
            "       -2.29019880e-01,  2.39549637e-01,  3.56725901e-01,  7.74199009e-01,\n",
            "       -6.67602658e-01, -1.20124988e-01, -5.16778648e-01,  5.61272204e-01,\n",
            "       -1.65650636e-01, -6.02031127e-02, -7.56135136e-02, -2.33316571e-01,\n",
            "        7.77224600e-02,  1.54268965e-01,  2.75470853e-01,  7.62763852e-03,\n",
            "        2.29427651e-01,  3.07577223e-01,  1.51994690e-01,  8.68775174e-02,\n",
            "        3.62097293e-01, -9.03340336e-03, -1.75406411e-01,  4.32225615e-01,\n",
            "       -3.73876467e-02,  5.06333768e-01, -5.30178666e-01, -4.66838092e-01,\n",
            "        1.24004364e-01,  2.88397878e-01,  3.58077854e-01,  5.21054976e-02,\n",
            "       -9.25125778e-02,  2.11265132e-01, -6.24504328e-01,  1.93297297e-01,\n",
            "        1.46188214e-01, -1.10487401e-01,  6.12124622e-01,  6.68794811e-01,\n",
            "       -4.84372914e-01,  2.62246132e-01, -5.90107918e-01, -1.03572786e-01,\n",
            "       -1.04129064e+00, -3.97392884e-02,  5.22732377e-01, -1.13421895e-01,\n",
            "        1.81515381e-01, -5.12585402e-01, -7.11936445e-04, -1.59848019e-01,\n",
            "        3.67904276e-01,  4.56355542e-01,  5.66160142e-01, -1.29843568e-02,\n",
            "        1.45634329e-02,  5.92723154e-02, -5.26260294e-04, -2.22327739e-01,\n",
            "       -3.74914676e-01, -9.72050354e-02, -1.37295304e-02, -2.49583479e-02,\n",
            "       -9.11740027e-03,  4.71937135e-02, -4.73742560e-02,  4.19239104e-01,\n",
            "        5.59417009e-02,  2.43367609e-02, -5.82077764e-02, -3.58068384e-02,\n",
            "        6.14802003e-01, -4.80452180e-02,  7.53868837e-03, -4.78032410e-01,\n",
            "       -3.87415320e-01,  2.52033714e-02,  2.08995983e-01, -1.95261165e-01,\n",
            "        1.77902669e-01, -2.63201058e-01, -1.33498818e-01, -2.44301841e-01,\n",
            "       -3.63227241e-02,  7.76296556e-01, -2.19590783e-01, -3.64123225e-01,\n",
            "        2.28695460e-02,  7.72955239e-01, -3.86596173e-01, -5.80430865e-01,\n",
            "        6.80257156e-02,  1.38978720e-01, -3.20346951e-01,  8.28190148e-03,\n",
            "        3.87788504e-01,  2.30803683e-01,  2.36231372e-01,  2.65630931e-01,\n",
            "        4.08732980e-01, -2.68627852e-01, -5.05620316e-02,  8.14274609e-01,\n",
            "       -5.43306649e-01,  1.80956230e-01, -5.45421779e-01,  1.28474578e-01,\n",
            "        2.00640783e-01, -2.31784821e-01, -6.86395541e-02,  2.29804993e-01,\n",
            "        2.69535393e-01,  3.94584626e-01,  2.70152539e-01,  9.65160489e-01,\n",
            "        1.30727127e-01,  5.16914204e-02,  1.08950734e+00, -5.39599219e-03,\n",
            "        2.86281437e-01,  3.46875906e-01,  3.79117697e-01,  2.52386630e-01,\n",
            "        6.30266845e-01, -4.20457363e-01, -3.11819166e-01,  1.80298686e-01,\n",
            "        2.33622432e-01, -2.29925275e-01, -1.88017473e-01, -3.01276475e-01,\n",
            "       -2.39173770e-01,  3.30795139e-01,  3.76616679e-02, -4.00230698e-02,\n",
            "       -3.00749660e-01,  1.09547926e-02, -3.60428877e-02, -8.57881188e-01,\n",
            "       -3.36861849e-01,  6.07087672e-01,  3.80378932e-01, -2.79592305e-01,\n",
            "       -2.58566052e-01, -4.47149813e-01, -1.18089698e-01, -2.28193045e-01,\n",
            "        3.68711054e-01,  5.16736090e-01, -1.21268618e+00, -1.17982663e-01,\n",
            "        6.06396794e-01, -3.70720834e-01,  1.52039394e-01, -3.70667651e-02,\n",
            "        2.41273507e-01,  3.72021288e-01, -5.61743975e-02,  2.85676420e-01,\n",
            "        8.47765431e-02, -3.45730782e-01, -1.12775110e-01,  2.79601485e-01,\n",
            "        2.74997205e-01,  1.76738769e-01, -4.95677032e-02, -8.46746489e-02,\n",
            "       -1.27902895e-01, -3.56198162e-01, -2.78652072e-01, -3.85348290e-01,\n",
            "       -1.81562394e-01, -2.68923670e-01, -1.28573269e-01,  9.26016867e-02,\n",
            "       -1.69558689e-01,  5.23394011e-02, -1.51190355e-01,  4.01330233e-01,\n",
            "        8.60256016e-01,  6.14358842e-01,  1.90761741e-02, -4.69651312e-01,\n",
            "        2.23627314e-01,  2.47112736e-01, -2.65970379e-01,  7.45621681e-01,\n",
            "        2.06085637e-01, -2.29468063e-01, -3.93451899e-02,  4.38752741e-01,\n",
            "       -6.33210182e-01,  4.25158113e-01, -3.14341545e-01,  2.38089655e-02,\n",
            "       -2.03792930e-01, -2.61216640e-01,  1.48080319e-01,  2.34176800e-01,\n",
            "       -3.09062954e-02, -5.37461460e-01,  5.62482588e-02,  5.63233614e-01,\n",
            "       -1.16586208e-01,  2.49443695e-01, -7.97840282e-02,  3.96609455e-01,\n",
            "       -4.93813336e-01, -1.67529389e-01, -6.28629327e-01, -1.88079759e-01,\n",
            "       -4.17017117e-02,  3.55887026e-01, -6.32327437e-01, -7.64231160e-02,\n",
            "        4.93985862e-01, -7.97062963e-02,  9.72899139e-01,  5.92025280e-01,\n",
            "        1.67058498e-01, -8.78825963e-01, -3.61264884e-01, -2.26550326e-01,\n",
            "       -1.54998884e-01,  2.77253181e-01,  7.00906813e-01,  6.14025407e-02,\n",
            "       -3.24895866e-02, -5.58161557e-01, -1.76207185e-01,  8.03664848e-02,\n",
            "        7.22693428e-02, -2.01505765e-01,  5.30872285e-01,  2.29085192e-01,\n",
            "        2.80973643e-01,  3.68656218e-01,  5.08992851e-01,  1.30797923e-01,\n",
            "        6.99691474e-01,  4.62755978e-01,  6.10535331e-02,  1.34878829e-02,\n",
            "       -1.77349076e-01, -4.12355512e-01, -2.58930176e-01, -8.40814888e-01,\n",
            "        2.02319741e-01, -5.23778163e-02,  1.10654819e+00,  4.74316210e-01],\n",
            "      dtype=float32), 'Sex': array([ 3.10874693e-02, -9.14818525e-01, -5.22118211e-01, -3.51312220e-01,\n",
            "       -3.34370226e-01, -6.06455505e-02, -3.51468563e-01,  4.19559419e-01,\n",
            "        2.66607642e-01,  1.92199588e-01,  1.92787975e-01,  4.10771459e-01,\n",
            "       -2.77610123e-01, -2.35676408e-01, -8.52644444e-02, -3.38949502e-01,\n",
            "       -1.56522632e-01,  4.54887807e-01, -1.86172768e-01, -2.85535883e-02,\n",
            "        3.58607858e-01,  2.62043059e-01, -3.54571581e-01,  3.31212953e-02,\n",
            "       -4.07557428e-01, -3.24978307e-02,  5.86053133e-02,  2.42311120e-01,\n",
            "        2.14032769e-01, -3.33289444e-01, -1.07136652e-01,  6.06400073e-01,\n",
            "       -5.84932268e-01,  1.17600191e+00,  1.75939083e-01, -1.28079101e-01,\n",
            "        9.61073767e-03, -2.24003583e-01,  1.10140073e+00,  3.08086872e-01,\n",
            "        5.98045997e-03, -5.00368595e-04, -1.04710788e-01, -1.65829450e-01,\n",
            "        2.74406433e-01,  6.30034685e-01,  1.54171392e-01, -4.27126527e-01,\n",
            "        4.53094810e-01, -2.37097010e-01, -2.26366624e-01, -1.00161873e-01,\n",
            "        1.31761700e-01,  3.44158351e-01, -2.63236821e-01, -8.49871561e-02,\n",
            "        2.86137432e-01,  2.36362338e-01, -3.20925176e-01,  1.21094190e-01,\n",
            "       -1.74664944e-01,  4.95001853e-01, -3.34553182e-01, -1.41774341e-01,\n",
            "       -4.79897559e-02,  2.41883770e-02, -6.59682751e-01, -3.00411403e-01,\n",
            "       -5.55356145e-01,  2.61590835e-02,  7.30448887e-02,  2.29233980e-01,\n",
            "        5.42701595e-02, -3.06848045e-02,  6.18455708e-02, -1.47176310e-01,\n",
            "        1.40591329e-02, -5.15399389e-02,  8.06259811e-02, -1.87206551e-01,\n",
            "        2.09905714e-01,  5.17604351e-01, -2.24016216e-02, -1.52246833e-01,\n",
            "       -1.29003942e-01, -2.64837444e-01,  4.23340499e-01,  6.24349415e-02,\n",
            "        1.60283726e-02, -2.55164504e-02,  3.34916949e-01,  5.73338419e-02,\n",
            "       -1.70845896e-01, -3.14815819e-01,  1.81166157e-01, -3.25028896e-01,\n",
            "        1.11625969e-01,  1.36893630e-01, -2.67399997e-02, -2.25491762e-01,\n",
            "        3.82129252e-02, -1.56849712e-01, -5.37442379e-02, -4.25430154e-03,\n",
            "       -8.21009874e-02, -4.82127398e-01,  2.33936638e-01,  5.85925542e-02,\n",
            "       -4.97763455e-01,  8.55255127e-02, -2.97557116e-01,  1.36969341e-02,\n",
            "        1.32668763e-01, -2.95919210e-01, -1.07237920e-01, -8.38574767e-02,\n",
            "        5.20308018e-01, -1.27864808e-01, -9.95829981e-03, -2.48582393e-01,\n",
            "        1.99691340e-01, -5.46694756e-01,  1.83924399e-02,  1.95934325e-01,\n",
            "        3.88046741e-01, -2.17284530e-01,  5.35652161e-01, -2.29175743e-02,\n",
            "        3.10390651e-01,  4.11268711e-01,  1.05436251e-01, -4.14555311e-01,\n",
            "       -2.15621799e-01,  9.48001146e-02,  2.01101601e-02, -1.85935661e-01,\n",
            "        1.85062498e-01,  4.28558439e-02,  3.66237044e-01,  1.57302111e-01,\n",
            "       -3.98204140e-02, -2.73292720e-01, -4.63502645e-01, -1.84145331e-01,\n",
            "       -5.35511300e-02, -8.25493112e-02,  7.75632143e-01, -3.79975885e-01,\n",
            "       -2.76307195e-01, -3.76686394e-01, -9.41236973e-01, -3.08326960e-01,\n",
            "        6.11056566e-01, -1.06524102e-01,  1.76234007e-01, -3.70837748e-01,\n",
            "        3.42790961e-01,  3.40199798e-01, -1.30905345e-01,  2.72067487e-01,\n",
            "        4.53580379e-01, -4.19591591e-02,  4.05549020e-01, -2.05022424e-01,\n",
            "       -1.16065387e-02, -1.10707348e-02, -3.58229160e-01, -4.57830906e-01,\n",
            "       -1.93638057e-01, -1.39602870e-01,  4.19905663e-01, -2.92000890e-01,\n",
            "       -4.71541062e-02, -2.56976366e-01,  1.12922098e-02, -4.56157148e-01,\n",
            "        2.77902067e-01,  1.18824914e-02, -9.48714465e-02, -2.83929445e-02,\n",
            "        3.63362163e-01, -2.23593879e+00, -4.62849557e-01,  1.93025172e-02,\n",
            "       -2.62537479e-01, -5.12091994e-01, -1.66622400e-01,  2.03582317e-01,\n",
            "        5.00384390e-01,  1.39417991e-01,  1.06937729e-01,  1.64026201e-01,\n",
            "       -9.47469473e-02,  9.96685307e-03,  2.61090845e-02,  3.47300589e-01,\n",
            "        3.12584132e-01,  4.36152071e-01,  2.84190863e-01,  1.45996287e-01,\n",
            "       -3.07924837e-01,  2.26299450e-01,  1.72182113e-01, -3.01562287e-02,\n",
            "       -3.05683941e-01, -1.64380372e-01, -2.93103069e-01,  1.98821515e-01,\n",
            "        1.06121801e-01,  1.88042790e-01, -4.01748747e-01,  1.65609300e-01,\n",
            "       -5.44909239e-01,  5.15270159e-02, -5.80912363e-03,  7.04460084e-01,\n",
            "        6.17346466e-02, -2.85923332e-01,  2.30177090e-01,  8.22990835e-02,\n",
            "       -7.11041540e-02, -3.86800826e-01, -1.01931199e-01, -1.38033897e-01,\n",
            "        8.73841047e-02,  3.65194023e-01, -6.23725533e-01,  2.93144315e-01,\n",
            "        2.19402775e-01, -3.86187315e-01,  5.58374703e-01, -2.55631953e-01,\n",
            "        5.41555166e-01,  2.79919386e-01,  1.73181176e-01,  2.75776863e-01,\n",
            "       -2.19275698e-01,  3.72155905e-01, -7.66707286e-02,  1.27810091e-01,\n",
            "       -2.67326176e-01, -4.50614303e-01, -5.59504449e-01,  3.05846512e-01,\n",
            "        4.18302029e-01, -3.10821161e-02,  3.73609245e-01,  6.00736141e-01,\n",
            "        5.39570451e-01,  6.18574321e-01, -1.61043778e-02, -2.31690332e-01,\n",
            "        5.43358922e-01,  3.85974407e-01, -3.38842124e-01,  8.82519782e-02,\n",
            "        1.63191944e-01,  1.12051122e-01,  1.64674729e-01, -1.30833268e-01,\n",
            "       -4.63068902e-01, -1.07855842e-01,  1.17126279e-01,  6.51766062e-01,\n",
            "       -1.79040045e-01, -5.58063626e-01,  5.27522080e-02, -4.22920913e-01,\n",
            "       -3.11574101e-01,  3.13356519e-01, -1.77492589e-01,  4.74729300e-01,\n",
            "        2.47399792e-01, -2.85944462e-01, -6.54500484e-01, -8.45160961e-01,\n",
            "       -1.39707774e-02, -6.32246673e-01,  3.48187387e-01, -7.90566385e-01,\n",
            "       -5.77638984e-01, -1.71627164e-01, -5.12437038e-02, -3.23674008e-02,\n",
            "       -1.87448129e-01,  3.33629966e-01,  1.82944059e-01,  6.48986250e-02,\n",
            "       -5.37485957e-01, -4.08832639e-01, -1.10258833e-01,  2.04053268e-01,\n",
            "       -4.10925359e-01, -2.85742849e-01,  5.95296733e-02,  4.25062418e-01,\n",
            "       -1.87813025e-02,  1.00377142e-01, -7.27542788e-02, -4.01268959e-01,\n",
            "       -1.05184451e-01,  1.68875635e-01,  2.92151690e-01, -5.30828089e-02,\n",
            "        7.22361952e-02, -1.09867916e-01,  1.28060505e-01,  3.71068954e-01,\n",
            "       -3.73347819e-01, -3.93338650e-01, -2.74948739e-02, -8.39961767e-02,\n",
            "        7.35607296e-02, -1.12631738e-01, -3.11320543e-01, -5.67092299e-01,\n",
            "        2.63486942e-03,  1.67034209e-01, -3.70762795e-01, -1.97131813e-01,\n",
            "        4.45085943e-01,  7.90868282e-01, -1.66598767e-01,  3.85595411e-02,\n",
            "       -2.63547711e-02,  3.12202275e-01, -3.95301208e-02,  4.43378776e-01,\n",
            "        5.96792921e-02, -8.52155909e-02, -9.57012475e-02, -3.05112749e-01,\n",
            "       -7.22366869e-02, -3.97318542e-01, -7.74100274e-02,  2.46588677e-01,\n",
            "        3.21563691e-01, -3.89283746e-01,  2.35834166e-01, -9.43513005e-04,\n",
            "        4.03283805e-01, -4.25999045e-01, -2.99811929e-01,  6.07685924e-01,\n",
            "       -4.36310396e-02,  1.02207325e-01,  3.72260809e-01, -3.87160748e-01,\n",
            "        1.51024505e-01, -2.17914786e-02,  4.29764032e-01, -4.59785074e-01,\n",
            "       -6.79093227e-02, -1.30280420e-01,  4.02309388e-01,  4.61357117e-01,\n",
            "        3.50893438e-01,  9.65324268e-02,  1.29220724e-01,  4.72768754e-01,\n",
            "        5.94108403e-02,  2.22211152e-01, -3.85520071e-01, -6.69272766e-02,\n",
            "       -5.00924885e-02,  3.30549657e-01,  1.59697428e-01, -7.18496799e-01,\n",
            "        7.50701576e-02, -1.41004413e-01,  1.30715966e-01, -2.77677417e-01,\n",
            "       -2.49836490e-01,  1.64283350e-01, -2.19563887e-01, -7.13236690e-01,\n",
            "       -3.68145943e-01, -1.80979297e-02,  1.75947234e-01, -4.38483655e-01,\n",
            "        2.22734272e-01, -3.95708531e-02, -7.26986825e-02, -3.79750133e-01,\n",
            "       -5.10802343e-02, -2.46790648e-01,  1.20954230e-01,  5.42971075e-01,\n",
            "       -3.94536322e-03,  2.87808895e-01, -1.02156095e-01, -2.71104090e-02,\n",
            "       -2.19856501e-01,  7.52934456e-01,  1.71439767e-01, -4.94861007e-02,\n",
            "        2.80286491e-01, -1.56148244e-02,  1.48389712e-01,  4.70364571e-01,\n",
            "       -1.52843326e-01, -1.96889549e-01, -4.69257295e-01,  4.58625883e-01,\n",
            "       -1.57068670e-01, -2.41857648e-01, -2.62701243e-01, -1.66501433e-01,\n",
            "       -4.96596005e-03, -2.89719254e-02, -6.81139231e-02,  5.73237062e-01,\n",
            "       -1.42958686e-01, -4.78478670e-02,  1.02784000e-01,  4.93288159e-01,\n",
            "       -1.86886340e-02,  2.08536312e-01,  2.87249029e-01, -5.45955598e-01,\n",
            "        2.77833641e-01,  1.54268488e-01, -4.04544771e-02,  2.05337867e-01,\n",
            "       -1.16922302e+01,  1.46178097e-01,  6.00690246e-01,  2.65570223e-01,\n",
            "        1.83969992e-03, -1.22070163e-01,  1.89502954e-01, -1.45083070e-01,\n",
            "        4.25540537e-01, -5.78607380e-01, -7.49748871e-02,  1.76230356e-01,\n",
            "       -3.22582424e-01, -1.55901715e-01, -2.66346991e-01,  1.81972593e-01,\n",
            "       -6.87814355e-02, -5.52516401e-01,  2.61001945e-01, -3.91093433e-01,\n",
            "        3.24861914e-01, -4.12319243e-01, -3.13296229e-01,  4.29639161e-01,\n",
            "       -2.16030508e-01, -4.44431871e-01, -2.69387603e-01,  9.73710697e-03,\n",
            "       -2.85434455e-01,  2.33836472e-01, -4.70572770e-01, -2.75848031e-01,\n",
            "        1.48356585e-02,  4.22696471e-01, -3.35231006e-01, -4.33528990e-01,\n",
            "        3.44793499e-01, -2.68360436e-01,  1.06485188e-01, -2.52615333e-01,\n",
            "       -4.80743647e-02,  3.66266787e-01,  8.70078802e-02,  1.96599945e-01,\n",
            "        1.98492408e-01,  5.77032685e-01,  6.39705598e-01,  8.93819749e-01,\n",
            "       -1.29469573e-01, -4.23454762e-01,  1.28075421e-01, -7.91039109e-01,\n",
            "        2.95597792e-01, -8.84656191e-01,  3.61248434e-01, -8.98985490e-02,\n",
            "        3.38857889e-01, -1.05166048e-01,  2.20504612e-01, -1.64927185e-01,\n",
            "        1.61771961e-02,  2.27491185e-01, -6.56358525e-02, -4.38174121e-02,\n",
            "       -5.09356797e-01, -9.13142692e-03,  6.43154621e-01, -1.39349490e-01,\n",
            "       -2.38032907e-01, -5.11849165e-01, -1.95460796e-01,  1.61010623e-02,\n",
            "        6.45874619e-01,  3.99478674e-01, -2.28350341e-01,  1.07516736e-01,\n",
            "       -1.47101998e-01,  2.74145842e-01,  6.73821568e-01, -4.64558676e-02,\n",
            "        7.78301358e-02,  2.17120290e-01,  1.71602756e-01, -6.61730096e-02,\n",
            "       -5.20137131e-01,  7.58551881e-02,  1.59942836e-01, -3.93310562e-02,\n",
            "       -1.08185738e-01,  1.94583386e-01, -2.45149180e-01,  4.59262490e-01,\n",
            "        9.43395018e-04, -1.56016976e-01,  2.60419667e-01,  7.64293075e-01,\n",
            "       -6.69007540e-01,  1.03472650e-01, -4.39123176e-02,  1.91163510e-01,\n",
            "        1.88152399e-02,  2.78304905e-01, -5.44944942e-01, -5.48081398e-01,\n",
            "        3.75712126e-01, -2.19371364e-01,  1.25439987e-01,  2.23678112e-01,\n",
            "        3.65811795e-01,  5.73314354e-02,  1.25413060e-01, -6.17363839e-04,\n",
            "       -8.98740441e-02,  2.70662487e-01, -4.76816505e-01, -2.37363577e-01,\n",
            "        2.56246805e-01,  2.22656652e-01, -5.97189665e-01, -3.40769619e-01,\n",
            "       -4.84914362e-01, -2.37259746e-01,  2.46191084e-01, -5.54840416e-02,\n",
            "       -8.17076489e-02,  1.77265227e-01, -4.01689231e-01, -2.22394884e-01,\n",
            "        3.35208118e-01, -1.11809395e-01,  4.46163118e-01,  5.23883283e-01,\n",
            "       -1.31372020e-01, -4.07574117e-01, -6.39608204e-02, -3.56393427e-01,\n",
            "       -9.54030395e-01,  9.34264511e-02,  2.22653091e-01,  1.44923879e-02,\n",
            "        1.56552345e-01, -1.94304705e-01, -9.37952846e-02, -2.41726965e-01,\n",
            "        4.80004042e-01,  5.06507531e-02,  4.39061493e-01,  4.98469993e-02,\n",
            "       -5.80725312e-01, -1.18519709e-01, -2.45922327e-01, -5.68367481e-01,\n",
            "        2.40268633e-01, -2.82442421e-01, -2.42826447e-01, -2.10556716e-01,\n",
            "        1.37807429e-01, -5.23571596e-02, -1.58502519e-01, -2.89342046e-01,\n",
            "       -1.01779483e-01, -1.00943297e-01,  1.16201013e-01, -1.47851050e-01,\n",
            "        5.53242445e-01,  2.65416592e-01, -7.51556158e-02, -3.29936802e-01,\n",
            "       -4.63973045e-01, -1.42247841e-01,  2.88543046e-01,  4.15557235e-01,\n",
            "        7.32143670e-02,  5.24086207e-02, -5.72504818e-01,  7.37245828e-02,\n",
            "       -1.69242188e-01,  6.74547672e-01, -3.70692581e-01, -4.92053479e-01,\n",
            "       -2.03968622e-02,  1.97692066e-01, -4.83338922e-01, -3.22198242e-01,\n",
            "        1.74228102e-01, -1.80310935e-01, -1.29723817e-01,  2.61322439e-01,\n",
            "        6.97788000e-02, -3.43018174e-01,  5.57350636e-01, -2.65497118e-01,\n",
            "        2.74526596e-01, -5.00685930e-01,  1.46658018e-01, -4.36312780e-02,\n",
            "       -4.04823542e-01, -4.18876037e-02, -2.69324362e-01, -2.64537573e-01,\n",
            "        9.58869159e-01,  3.47753391e-02, -5.64319491e-02, -1.41833663e-01,\n",
            "       -1.14042819e-01, -1.13192424e-01,  2.53016204e-01,  1.10027444e+00,\n",
            "       -2.73354292e-01, -1.55908614e-01,  4.84601438e-01,  1.64809644e-01,\n",
            "        1.18993126e-01,  2.37215757e-01,  2.46716961e-01,  5.42315058e-02,\n",
            "        6.28596723e-01,  2.60059573e-02, -6.43383384e-01,  2.39643231e-02,\n",
            "        2.44306952e-01,  4.27868307e-01, -4.42707956e-01, -6.04745388e-01,\n",
            "        1.76578183e-02,  5.75588942e-02,  2.01863080e-01,  2.67197013e-01,\n",
            "       -4.85219359e-01,  2.76228428e-01,  1.35685727e-01, -9.24743772e-01,\n",
            "        7.85842389e-02, -4.02431339e-01, -1.15616158e-01, -2.69432664e-01,\n",
            "       -2.09655255e-01, -3.23111385e-01, -1.64569244e-01, -4.19922352e-01,\n",
            "        9.27318335e-02,  4.00974482e-01, -3.31855953e-01, -1.91606730e-01,\n",
            "        2.37962246e-01, -6.45478427e-01,  4.03896809e-01, -3.87572753e-03,\n",
            "       -6.85167834e-02,  2.34747976e-01,  9.02728960e-02, -4.10937034e-02,\n",
            "        2.18682885e-01, -7.63048679e-02, -4.94402975e-01,  2.95593143e-01,\n",
            "        3.76015007e-01,  3.97206545e-02, -2.28624903e-02,  1.62212715e-01,\n",
            "        2.57351935e-01, -1.88936338e-01,  2.22224798e-02, -2.52732813e-01,\n",
            "        6.81226403e-02, -3.26935314e-02, -1.66098285e-03, -1.78865597e-01,\n",
            "        1.91074878e-01,  4.41371918e-01, -5.11225641e-01,  2.34212086e-01,\n",
            "        4.63749141e-01,  1.15653408e+00, -1.04051251e-02, -3.95357609e-01,\n",
            "        2.76683539e-01,  3.80615801e-01,  1.20822988e-01,  4.89382595e-01,\n",
            "       -7.46456161e-02, -5.20632491e-02, -1.50847390e-01,  2.98820734e-01,\n",
            "       -1.56098932e-01,  2.76467681e-01,  6.28815545e-03, -2.21233845e-01,\n",
            "       -3.25023234e-01, -7.92516321e-02,  1.00309655e-01,  5.09594940e-02,\n",
            "        1.78282186e-01, -6.48568451e-01, -6.48520291e-02,  4.89906490e-01,\n",
            "        2.96356604e-02,  3.74657214e-01, -1.00411728e-01,  4.27701771e-01,\n",
            "       -2.28576874e-03, -1.15874439e-01, -6.15667045e-01, -3.28566372e-01,\n",
            "        7.37433229e-03,  4.88564298e-02, -5.23746967e-01, -7.13260770e-02,\n",
            "        2.38358259e-01,  1.39829963e-01,  4.67785299e-01,  5.98080039e-01,\n",
            "       -9.85383019e-02, -7.03001022e-01, -4.22064722e-01, -4.43608969e-01,\n",
            "       -2.39098430e-01, -9.47386622e-02,  5.49322009e-01,  4.36947227e-01,\n",
            "       -2.21379206e-01, -4.23089683e-01,  1.71899915e-01, -1.35138690e-01,\n",
            "        2.09767073e-01, -4.01997179e-01,  8.58619392e-01,  8.68667364e-02,\n",
            "        2.84873039e-01,  3.17176044e-01,  3.96327078e-02,  1.67915985e-01,\n",
            "        7.12576091e-01,  8.70413184e-02, -5.27666152e-01, -7.01099187e-02,\n",
            "        1.03710458e-01, -2.73026735e-01,  4.00419533e-02, -5.92701912e-01,\n",
            "        3.58687460e-01, -2.83157587e-01,  3.65589678e-01,  5.78229189e-01],\n",
            "      dtype=float32), 'Clinical_event': array([-3.33408296e-01, -6.17860854e-01, -6.68913066e-01, -2.86092013e-01,\n",
            "       -4.13030088e-01, -1.42831996e-01, -6.15187645e-01,  5.14931917e-01,\n",
            "       -7.32297972e-02,  2.35729426e-01, -2.69692808e-01, -3.52559268e-01,\n",
            "        9.10826847e-02, -2.89104939e-01, -5.32227755e-01,  2.12621853e-01,\n",
            "       -2.41572000e-02, -2.86869943e-01,  1.26651347e-01,  3.59757125e-01,\n",
            "        6.94902182e-01,  4.15937543e-01,  2.01844480e-02, -1.49168909e-01,\n",
            "       -4.64275181e-01, -1.32450983e-02,  1.96930349e-01, -3.25767577e-01,\n",
            "        2.92596877e-01,  4.42308754e-01,  1.51650533e-01,  8.06655347e-01,\n",
            "       -2.61762023e-01,  1.09720486e-03, -3.10853541e-01, -3.32925022e-01,\n",
            "        6.15267735e-03, -5.22845313e-02,  9.96302605e-01,  1.22525096e-01,\n",
            "        6.99476719e-01,  7.54543424e-01,  2.64218688e-01,  2.59231269e-01,\n",
            "        4.52765405e-01,  1.44710720e-01, -2.41666824e-01, -6.58042058e-02,\n",
            "       -1.66238211e-02, -4.56817001e-01, -1.32876977e-01,  3.21678966e-01,\n",
            "        4.78435844e-01, -4.29965347e-01, -6.96642876e-01,  2.00559124e-01,\n",
            "       -3.44330788e-01,  2.36507937e-01, -5.15304446e-01, -6.50480539e-02,\n",
            "        3.73348802e-01,  6.98267162e-01, -1.72589630e-01, -1.53814584e-01,\n",
            "       -7.95734450e-02, -2.49567077e-01, -2.75546014e-01, -4.91998792e-01,\n",
            "       -7.41869390e-01, -2.53879756e-01,  4.46900427e-02,  3.36776912e-01,\n",
            "       -2.12133929e-01, -3.81579638e-01,  2.52951086e-01, -2.14121252e-01,\n",
            "        8.29833269e-01, -7.17435837e-01, -9.19045806e-02, -6.00071669e-01,\n",
            "        3.38517904e-01,  3.17398489e-01,  2.49767587e-01, -4.60883081e-02,\n",
            "       -7.00093687e-01,  8.56222063e-02,  3.07498574e-01,  4.74623770e-01,\n",
            "       -6.26328513e-02, -5.03822044e-03,  7.03491449e-01, -8.61700997e-02,\n",
            "       -1.28176332e-01, -3.94037187e-01,  1.66472584e-01, -2.03205988e-01,\n",
            "        2.70756394e-01, -1.74845681e-01,  1.04331877e-02, -1.25305086e-01,\n",
            "       -1.20682620e-01,  2.64663339e-01,  3.27500999e-01, -2.19047725e-01,\n",
            "       -3.58836979e-01, -4.40018773e-01,  4.38452452e-01, -1.50137395e-01,\n",
            "       -7.85672426e-01, -1.10190019e-01, -1.62541673e-01,  2.42469069e-02,\n",
            "        1.02823257e-01, -7.01161146e-01, -1.92892238e-01,  1.17292397e-01,\n",
            "        4.82906669e-01, -1.90866753e-01, -5.32026768e-01,  1.51831284e-02,\n",
            "       -1.96468234e-01, -6.62481179e-03,  4.59983826e-01,  1.13176212e-01,\n",
            "       -2.42452677e-02,  2.38664374e-02,  4.00940835e-01, -2.08705179e-02,\n",
            "        3.85084093e-01, -3.54695730e-02, -1.09533481e-01, -1.21197164e-01,\n",
            "       -7.41540313e-01,  3.96548659e-02, -4.45753068e-01,  1.21729299e-01,\n",
            "       -4.01696146e-01, -5.09696305e-01, -9.30362865e-02,  3.01209450e-01,\n",
            "        2.86787208e-02,  2.10216850e-01, -5.29596746e-01,  1.90411910e-01,\n",
            "       -1.16662703e-01, -3.19037080e-01,  2.26459593e-01, -3.38062704e-01,\n",
            "       -4.68064427e-01, -3.06685507e-01, -2.85567045e-01, -7.79726207e-01,\n",
            "        4.85901296e-01,  2.39094347e-02, -1.21502325e-01, -5.66925168e-01,\n",
            "        1.21429741e-01, -1.20363519e-01, -7.32339825e-03,  3.39574575e-01,\n",
            "        8.60434055e-01, -1.05110750e-01,  3.00723791e-01, -6.22473173e-02,\n",
            "       -7.92668685e-02,  1.43743515e-01, -5.54706514e-01, -3.54230493e-01,\n",
            "        1.01522438e-01,  2.31888145e-01,  1.77621752e-01,  8.09365883e-02,\n",
            "        4.42114919e-01, -6.59297585e-01, -4.36948061e-01, -2.46891081e-01,\n",
            "       -9.93480086e-02,  1.37215834e-02, -2.45757222e-01, -2.56298333e-02,\n",
            "       -7.57914484e-02, -2.44095850e+00, -2.14699149e-01, -3.48040909e-02,\n",
            "        2.58273389e-02,  1.09794855e-01, -4.23042715e-01, -1.88370988e-01,\n",
            "        4.05417621e-01, -1.58712208e-01, -4.77848589e-01,  1.05186187e-01,\n",
            "       -3.23082030e-01, -2.19212413e-01, -1.61218852e-01, -4.82169390e-02,\n",
            "        8.96111667e-01,  6.93043470e-01, -1.09495118e-01,  2.26748556e-01,\n",
            "       -4.94536459e-01,  1.05657473e-01,  4.92397547e-02, -4.44056243e-02,\n",
            "        5.56176603e-02,  5.90573430e-01, -5.20518959e-01,  5.63798845e-01,\n",
            "       -1.06235594e-01, -5.70984855e-02, -1.61968112e-01, -1.85910046e-01,\n",
            "       -1.33980200e-01, -2.07736999e-01,  2.86810454e-02,  6.17214739e-01,\n",
            "       -3.49060774e-01, -4.38221991e-01,  7.30192900e-01, -4.62940447e-02,\n",
            "       -4.35082436e-01,  6.04397655e-02, -1.31846428e-01, -3.43657702e-01,\n",
            "        1.00340202e-01,  7.30926275e-01, -6.40509427e-01,  6.87893152e-01,\n",
            "       -1.11988798e-01, -2.60455012e-01, -2.45876104e-01, -6.06258273e-01,\n",
            "        1.59238249e-01,  4.10623968e-01,  1.97493315e-01, -4.48204502e-02,\n",
            "       -2.88709879e-01,  4.80742097e-01, -1.46085069e-01,  1.36767268e-01,\n",
            "        1.61962241e-01, -2.78177857e-01, -8.39723572e-02, -3.89031589e-01,\n",
            "        1.92582995e-01, -1.95605516e-01, -2.05026969e-01,  5.73655248e-01,\n",
            "        3.02879035e-01,  5.14607728e-01,  2.17621967e-01, -3.42387408e-02,\n",
            "        1.89988896e-01,  6.17892504e-01, -1.69515401e-01, -2.44470388e-01,\n",
            "        6.98546350e-01, -1.05245539e-03,  2.61002332e-01, -1.37054443e-01,\n",
            "       -5.06866515e-01, -7.54654557e-02,  1.93240881e-01,  3.10351461e-01,\n",
            "        2.94460319e-02,  5.97839832e-01,  2.12442324e-01, -3.87740225e-01,\n",
            "       -5.09878039e-01, -9.42645222e-02, -6.04584157e-01,  2.80591458e-01,\n",
            "        2.98831642e-01, -2.82848477e-01, -7.21104324e-01, -1.39246702e+00,\n",
            "       -1.03075624e-01, -5.34263313e-01, -1.15311675e-01, -6.04786158e-01,\n",
            "       -3.12974215e-01,  1.30713344e-01,  3.44929874e-01, -3.68343234e-01,\n",
            "       -7.12593049e-02,  1.56691179e-01, -2.12055728e-01,  3.26628685e-01,\n",
            "       -4.47068304e-01, -4.85717982e-01,  5.18471263e-02, -5.03288582e-02,\n",
            "       -2.30766580e-01, -1.11874759e-01,  1.32920370e-01, -2.04541266e-01,\n",
            "        1.04529202e-01,  3.36001158e-01,  7.27260560e-02,  4.90760729e-02,\n",
            "       -1.32797107e-01, -1.26366258e-01, -7.22148782e-03,  1.95794195e-01,\n",
            "        1.36609495e-01,  1.18695006e-01,  2.03749359e-01,  1.04520321e+00,\n",
            "       -2.01304957e-01,  9.33629498e-02,  2.03015074e-01,  4.82348293e-01,\n",
            "       -7.29656443e-02, -5.40368617e-01,  2.04357952e-01, -3.22792143e-01,\n",
            "        8.43139738e-02, -2.50408649e-01, -1.39795989e-01, -1.57702044e-01,\n",
            "        7.15259314e-01, -3.86182964e-01, -1.11520857e-01,  3.51151913e-01,\n",
            "       -3.42079967e-01,  5.57900608e-01,  6.11047089e-01, -6.24209285e-01,\n",
            "        2.15901956e-02, -2.57395148e-01,  3.68683226e-02,  2.41338741e-02,\n",
            "       -3.95269006e-01,  1.21782959e-01,  6.30366951e-02,  7.43041694e-01,\n",
            "        4.87141982e-02,  1.48090586e-01, -4.94424179e-02,  3.93012822e-01,\n",
            "        1.95641264e-01, -2.54660985e-03, -1.46702915e-01,  4.39946860e-01,\n",
            "        3.61084312e-01, -4.57217187e-01,  2.96156824e-01, -5.84355056e-01,\n",
            "       -9.55137014e-02, -2.22281024e-01,  1.11225210e-01,  4.27656099e-02,\n",
            "       -9.59132612e-02, -1.80222720e-01,  4.19086576e-01,  5.44269025e-01,\n",
            "       -3.48412767e-02, -1.82945773e-01,  2.81594507e-02,  6.10963106e-01,\n",
            "       -4.38791335e-01,  3.30060542e-01,  2.32136711e-01, -1.53792918e-01,\n",
            "       -2.16531008e-01,  1.38883293e-01, -9.25714597e-02, -3.41372371e-01,\n",
            "       -5.90942681e-01, -5.84668666e-02,  1.39368445e-01, -5.93111329e-02,\n",
            "       -2.87502855e-01,  5.07940888e-01,  1.05431902e+00, -2.75403559e-01,\n",
            "       -4.78999317e-01,  2.88327932e-01, -5.23385525e-01, -1.38815194e-01,\n",
            "        5.19371545e-03, -7.16414005e-02, -7.77747333e-02, -4.32641625e-01,\n",
            "       -7.74979472e-01, -3.20624709e-01,  4.66540635e-01,  5.58467269e-01,\n",
            "        3.95934641e-01,  1.92612350e-01,  4.35829014e-01,  8.52665752e-02,\n",
            "       -2.24832207e-01,  7.26336658e-01, -2.65731096e-01, -4.84100282e-01,\n",
            "        3.09020609e-01,  1.71979308e-01,  3.41100395e-01,  4.21652719e-02,\n",
            "       -2.83689611e-02, -3.81955951e-01, -2.00427979e-01,  3.48392338e-01,\n",
            "        5.17179847e-01, -1.05403915e-01,  1.90335631e-01,  4.61261570e-01,\n",
            "       -6.24059200e-01, -1.26809403e-01,  3.67345065e-01,  3.83243978e-01,\n",
            "       -6.69920623e-01, -9.47818905e-02, -2.80904382e-01,  6.39710307e-01,\n",
            "       -3.32735270e-01,  2.72064149e-01, -1.33003578e-01, -5.71005762e-01,\n",
            "        3.84389997e-01,  1.53034568e-01,  6.13021804e-03,  1.99964568e-01,\n",
            "       -1.09795971e+01,  3.01313579e-01,  6.97573900e-01, -5.68921447e-01,\n",
            "       -4.38937172e-02,  2.55530804e-01,  2.27498531e-01, -4.52918932e-02,\n",
            "       -2.12850347e-02, -6.04475848e-02, -3.41424525e-01, -9.24401879e-01,\n",
            "        5.71552962e-02, -2.88276374e-01, -1.66831657e-01, -1.51772005e-02,\n",
            "        9.47270021e-02, -5.18301129e-01,  3.69395912e-01,  1.28039658e-01,\n",
            "       -1.16390169e-01, -2.49235004e-01,  3.64067405e-02,  1.37495637e-01,\n",
            "       -4.66131419e-01,  1.72723055e-01, -3.41257542e-01, -4.41473097e-01,\n",
            "        4.43733931e-01,  4.75835443e-01,  4.56791632e-02, -2.28743359e-01,\n",
            "       -1.68933794e-01,  3.25625777e-01, -4.63764518e-01, -3.45846146e-01,\n",
            "        2.68547475e-01, -4.02793139e-01,  1.66390687e-01, -3.35951656e-01,\n",
            "       -1.27739951e-01, -6.55073971e-02,  1.85535759e-01, -1.84638798e-01,\n",
            "        4.77708906e-01,  5.11457622e-01,  1.03506339e+00,  7.14107513e-01,\n",
            "       -2.05851436e-01, -8.99471492e-02, -5.85232019e-01, -5.60522377e-01,\n",
            "        1.98745415e-01, -3.08299392e-01,  1.97146058e-01, -5.44712618e-02,\n",
            "        1.90510318e-01, -6.65490746e-01,  1.34946138e-01, -3.28141093e-01,\n",
            "        2.95859873e-01, -1.66648000e-01,  3.99357498e-01,  2.14515433e-01,\n",
            "       -1.41133413e-01,  9.04475600e-02,  3.14188242e-01, -3.63670439e-01,\n",
            "       -8.15435022e-04, -5.83330095e-01,  3.92139494e-01, -4.24451053e-01,\n",
            "        6.72517776e-01,  2.23303780e-01, -8.77296925e-01,  1.06996335e-02,\n",
            "        1.05550826e-01,  9.39784199e-03,  1.67031324e+00, -8.65833983e-02,\n",
            "        1.88644528e-01,  2.48001367e-01,  3.06481898e-01, -6.11990988e-02,\n",
            "       -2.57319808e-01, -1.54756010e-01,  2.80514270e-01,  6.35920092e-02,\n",
            "        5.60911655e-01,  3.84178698e-01, -3.45296323e-01, -9.57833156e-02,\n",
            "        2.25914028e-02, -8.77269208e-02,  3.15492243e-01,  2.80244231e-01,\n",
            "       -8.65951896e-01,  2.40986407e-01,  5.60813174e-02,  6.00775123e-01,\n",
            "        4.77654152e-02,  4.04151320e-01, -3.08442950e-01, -3.86932254e-01,\n",
            "        2.44071037e-01,  3.66938949e-01,  2.77903289e-01, -7.73792565e-02,\n",
            "       -1.56482577e-01,  3.40694427e-01,  2.96126425e-01,  3.37673187e-01,\n",
            "        3.81729215e-01,  1.04991779e-01, -4.57054824e-01, -9.97741461e-01,\n",
            "        3.76293004e-01,  1.05100684e-01, -5.88472724e-01, -4.65247512e-01,\n",
            "       -1.20459631e-01, -1.77385911e-01,  4.14302588e-01, -3.90173674e-01,\n",
            "       -5.16335890e-02,  5.28032720e-01, -6.39020046e-03, -3.48359406e-01,\n",
            "        1.71206594e-01,  5.23670912e-01,  4.85376179e-01,  6.60956383e-01,\n",
            "       -1.80820435e-01, -2.42210597e-01, -5.31443596e-01, -5.08782089e-01,\n",
            "       -1.09002471e+00,  2.30087683e-01,  6.63001388e-02, -2.48005800e-02,\n",
            "        2.99769998e-01, -1.45320594e-01, -4.64688778e-01, -2.24935740e-01,\n",
            "        2.90222943e-01,  4.04492080e-01, -7.21795857e-02,  5.96253932e-01,\n",
            "       -1.16902232e-01,  1.58819556e-01, -1.12480551e-01, -2.24233419e-01,\n",
            "       -2.85632849e-01,  4.72616218e-02, -2.25129843e-01, -6.10945940e-01,\n",
            "       -4.68256056e-01, -3.04821461e-01, -1.70563191e-01, -4.88195062e-01,\n",
            "       -3.24538022e-01,  1.33084282e-02, -3.03083789e-02, -2.42886707e-01,\n",
            "        2.28942215e-01,  3.98852348e-01,  1.58074319e-01, -3.16922843e-01,\n",
            "       -2.89822578e-01, -2.97038089e-02,  1.61808670e-01,  1.69226110e-01,\n",
            "        1.47747457e-01, -1.31059468e-01,  4.00300801e-01, -2.09341675e-01,\n",
            "       -1.32403344e-01,  9.04255390e-01, -5.66224813e-01, -4.78990048e-01,\n",
            "       -2.19851896e-01,  5.27867913e-01, -2.68354177e-01, -3.08330119e-01,\n",
            "       -6.08059131e-02, -1.74844861e-01,  1.96077272e-01, -1.82856366e-01,\n",
            "        1.86973691e-01,  2.11575001e-01,  1.33640140e-01,  4.54415858e-01,\n",
            "        1.86486036e-01, -5.55686474e-01,  2.19479561e-01,  2.61661500e-01,\n",
            "       -1.84887171e-01,  5.08503541e-02, -1.35922268e-01, -5.07303357e-01,\n",
            "        5.83412349e-02, -2.50935078e-01,  4.30866003e-01, -3.89311641e-01,\n",
            "        7.73913741e-01, -6.52158856e-01, -2.27689847e-01,  1.43251896e+00,\n",
            "       -3.25753093e-01, -1.31643992e-02,  5.36501884e-01,  2.53169358e-01,\n",
            "       -4.21203464e-01,  4.99129772e-01,  3.20359439e-01,  2.34253556e-02,\n",
            "        1.87577248e-01,  6.05540983e-02,  1.65414616e-01,  1.35974258e-01,\n",
            "        2.68407464e-01,  3.10104609e-01, -1.00458241e+00, -5.98236732e-02,\n",
            "       -1.39744639e-01, -2.11598203e-01,  2.72720307e-03,  7.43656099e-01,\n",
            "       -1.08266376e-01, -1.45750074e-02,  5.99992990e-01, -4.97043908e-01,\n",
            "        2.30927512e-01,  2.51767427e-01,  1.55820370e-01,  1.64022744e-01,\n",
            "       -4.71444547e-01,  4.10243422e-02, -1.31086826e-01, -7.32345507e-02,\n",
            "        2.04034150e-02,  3.59479070e-01, -8.15045536e-01,  3.25368494e-01,\n",
            "        9.77285087e-01, -1.34921789e-01,  1.00195855e-01,  1.63152069e-01,\n",
            "        8.19055915e-01, -4.05387461e-01, -8.56849253e-02, -1.85951456e-01,\n",
            "        1.83331653e-01, -2.90377676e-01, -7.78498352e-01,  6.09872043e-01,\n",
            "        8.54754806e-01,  1.81107193e-01, -1.93114683e-01,  8.41029435e-02,\n",
            "       -2.50652909e-01, -1.03606686e-01,  7.62557983e-02, -2.70391315e-01,\n",
            "        3.05856913e-02, -2.95053184e-01,  1.95755772e-02, -2.24867389e-01,\n",
            "       -1.44501150e-01, -1.77713424e-01, -9.86445323e-03, -1.40884548e-01,\n",
            "        4.70098734e-01,  8.90147924e-01, -4.91914332e-01, -7.48599991e-02,\n",
            "        5.27137816e-02,  5.61114728e-01,  6.59810677e-02,  3.15217078e-01,\n",
            "       -2.33859032e-01, -3.47883165e-01,  4.54957426e-01, -1.51971370e-01,\n",
            "       -4.20805901e-01,  2.18115658e-01, -1.64397925e-01, -1.31335706e-01,\n",
            "       -5.37895381e-01, -2.76745528e-01,  1.24996722e-01,  1.46704614e-01,\n",
            "        9.74741131e-02, -2.81045884e-01, -1.09557986e-01,  4.38656926e-01,\n",
            "        2.48194098e-01,  8.94333839e-01, -2.41691247e-02,  1.03899872e+00,\n",
            "       -2.31341004e-01,  6.31451607e-02, -5.90385318e-01, -9.31500923e-03,\n",
            "       -1.87374093e-02, -1.66773307e-03, -2.95116842e-01, -1.24922454e-01,\n",
            "        3.08914185e-01,  4.87257659e-01,  4.34090376e-01,  1.90555155e-01,\n",
            "       -1.66984841e-01, -7.24267662e-01, -6.49251223e-01, -6.50982261e-02,\n",
            "        1.81645691e-01, -7.50705823e-02,  5.56377247e-02, -2.05511853e-01,\n",
            "       -2.34518290e-01, -4.85889673e-01,  2.66507626e-01,  5.31764841e-03,\n",
            "        2.37322450e-01, -3.73579562e-01,  1.69333696e-01, -1.83856376e-02,\n",
            "       -8.42161924e-02,  8.98427367e-01,  6.93948865e-01,  1.51370823e-01,\n",
            "        4.53433618e-02,  3.27608846e-02, -6.63084835e-02, -2.25535035e-02,\n",
            "        3.30624878e-01,  3.30586098e-02, -3.71592134e-01, -1.17451763e+00,\n",
            "        1.07461713e-01,  3.34647894e-01,  9.67338562e-01,  3.42991650e-01],\n",
            "      dtype=float32), 'Nonbiological_location': array([ 1.36887059e-02, -3.36531878e-01, -4.52316940e-01, -3.86965930e-01,\n",
            "       -2.69152820e-01, -1.74658716e-01, -4.59443778e-01,  8.82764980e-02,\n",
            "        8.22342932e-02,  6.11499667e-01,  3.61392587e-01,  3.19468170e-01,\n",
            "       -7.57241994e-02, -2.15949774e-01, -8.83115530e-01,  7.45318532e-02,\n",
            "       -2.80511588e-01,  7.07308173e-01, -2.74571441e-02,  3.05682421e-01,\n",
            "        3.30915779e-01,  2.99654484e-01,  3.18715453e-01, -1.02904007e-01,\n",
            "        5.12350118e-03, -2.14243695e-01,  6.83219209e-02,  2.40287274e-01,\n",
            "        2.82158583e-01, -4.71644729e-01,  2.52773434e-01,  7.69909084e-01,\n",
            "       -6.28489017e-01,  5.34729481e-01, -4.65726480e-02, -3.25625420e-01,\n",
            "       -1.35412320e-01, -6.06152296e-01,  5.29154301e-01, -4.82982099e-01,\n",
            "        2.81269908e-01,  3.96487087e-01,  3.33844006e-01,  2.18228363e-02,\n",
            "        2.57108092e-01,  1.22371696e-01,  2.68893149e-02, -4.54478711e-01,\n",
            "        4.83630776e-01, -3.18446159e-01, -3.45241874e-01, -5.46876416e-02,\n",
            "       -1.65769503e-01,  3.90460226e-03, -5.10756075e-01,  1.00372240e-01,\n",
            "       -3.61035913e-02,  4.24643278e-01, -4.54847664e-01,  4.06151742e-01,\n",
            "        5.02998158e-02,  4.77893829e-01, -1.89004749e-01, -2.91625470e-01,\n",
            "        7.92040378e-02,  3.25464547e-01, -5.80311954e-01, -5.67494631e-01,\n",
            "       -3.73547018e-01, -6.27958775e-02, -5.95208406e-02,  6.93685338e-02,\n",
            "        3.55286780e-03, -3.47957015e-01,  1.83439642e-01, -1.41873471e-02,\n",
            "        3.23885500e-01, -1.70079231e-01, -2.32735559e-01, -2.74267137e-01,\n",
            "        2.84298301e-01,  6.87968254e-01, -2.71277517e-01, -1.56900451e-01,\n",
            "        2.09999323e-01,  8.42416584e-02,  4.73340392e-01,  2.79399753e-01,\n",
            "       -1.47377491e-01,  1.85414001e-01,  4.12479967e-01,  2.89933830e-01,\n",
            "       -5.68444848e-01, -2.57728726e-01,  3.96659940e-01,  8.23219270e-02,\n",
            "       -1.87149900e-03, -4.08107162e-01, -1.00194596e-01, -2.43433297e-01,\n",
            "       -1.27234370e-01, -1.15012303e-02,  9.31909084e-02,  4.78091352e-02,\n",
            "       -3.40069056e-01, -4.39017951e-01,  3.85447502e-01,  1.12383895e-01,\n",
            "       -3.81638587e-01, -3.41745466e-01, -9.76589024e-02, -1.13485515e-01,\n",
            "        1.82991624e-01, -2.90842116e-01, -2.90235486e-02,  1.89659759e-01,\n",
            "        5.05930305e-01,  1.21015683e-01, -4.11266625e-01, -3.56037796e-01,\n",
            "        4.73346829e-01, -1.52605474e-01,  1.49676278e-01,  7.64157400e-02,\n",
            "        2.25145862e-01, -2.80089676e-01,  6.63966000e-01, -1.01885110e-01,\n",
            "        8.83599520e-02,  2.81194031e-01,  2.66667366e-01, -5.93444467e-01,\n",
            "       -7.00649738e-01,  6.54341280e-02,  5.94968915e-01,  4.21379618e-02,\n",
            "       -4.43214148e-01, -2.04268992e-01, -2.07333148e-01,  2.30685389e-03,\n",
            "       -6.15872443e-02, -1.96405575e-01, -1.08674079e-01, -2.77527254e-02,\n",
            "        4.53984402e-02, -3.81787747e-01,  3.69067907e-01, -6.18952692e-01,\n",
            "        1.65001899e-01, -2.85648435e-01, -7.57725894e-01, -1.25455543e-01,\n",
            "        2.87921429e-01,  2.15908028e-02,  1.72438651e-01, -3.23352575e-01,\n",
            "        1.61562860e-01, -4.21116613e-02, -2.28651330e-01,  3.01178634e-01,\n",
            "        4.38930988e-01, -1.40313119e-01,  1.23672768e-01, -6.80108309e-01,\n",
            "       -2.58481801e-01,  6.57026917e-02,  1.83040828e-01, -1.72255024e-01,\n",
            "       -2.61660814e-02,  8.07560161e-02,  2.31240317e-01, -2.88217664e-01,\n",
            "        9.84611958e-02, -5.33063531e-01, -3.44268650e-01, -1.90759271e-01,\n",
            "       -2.74187386e-01, -2.46563002e-01,  7.10699856e-02, -1.24425828e-01,\n",
            "        4.12894845e-01, -1.96051443e+00,  1.25864325e-02,  5.79720736e-01,\n",
            "       -3.04370761e-01, -4.88180459e-01, -9.94370691e-03, -3.84550035e-01,\n",
            "        4.15143967e-01,  6.55202866e-02,  1.21667892e-01,  3.23095242e-03,\n",
            "        1.00497007e-01, -2.72283643e-01, -6.46069944e-02,  5.81144691e-01,\n",
            "        5.66342533e-01,  3.42038684e-02,  1.69944912e-01,  5.03551587e-02,\n",
            "       -2.89007634e-01,  3.90751660e-01, -6.03735670e-02,  4.64724869e-01,\n",
            "        7.04488903e-02, -1.10600982e-02, -5.35229921e-01,  1.32326558e-02,\n",
            "       -3.85945410e-01,  1.69542670e-01, -3.66360366e-01,  5.64911030e-02,\n",
            "       -5.10306478e-01,  8.31389725e-02, -2.03975827e-01,  1.03904200e+00,\n",
            "       -1.29335433e-01, -3.50194931e-01,  2.05032349e-01,  9.89461541e-02,\n",
            "       -3.21291655e-01, -1.85871646e-01,  2.77243629e-02, -4.40796673e-01,\n",
            "       -2.14450881e-01,  7.33110666e-01, -2.62919754e-01, -1.87972318e-02,\n",
            "       -1.27106845e-01, -1.57330886e-01,  1.16499901e-01, -4.31333005e-01,\n",
            "        1.42197773e-01,  3.82930756e-01, -1.39893889e-01, -2.63388425e-01,\n",
            "        2.60554366e-02,  2.35692114e-01, -5.37174106e-01, -3.33329827e-01,\n",
            "       -4.20105666e-01, -4.84428167e-01, -3.76387060e-01,  2.08102331e-01,\n",
            "        4.10684168e-01,  9.95790362e-02, -1.32325023e-01,  7.91878641e-01,\n",
            "        4.16715980e-01,  2.78676510e-01, -1.10573471e-02,  1.51078045e-01,\n",
            "        3.19097579e-01,  4.02427316e-01, -3.03831816e-01, -3.70178372e-03,\n",
            "        2.44997427e-01,  5.08060865e-02,  1.54633254e-01, -1.49432003e-01,\n",
            "       -2.96243191e-01,  2.03814670e-01,  3.04430038e-01,  4.12169307e-01,\n",
            "       -2.82846808e-01,  2.08593458e-01,  5.32331824e-01, -2.26687357e-01,\n",
            "       -1.81570172e-01, -8.79360735e-02,  8.34385082e-02,  5.11148572e-01,\n",
            "       -7.36056268e-01, -3.65253687e-01, -4.84942615e-01, -3.24138761e-01,\n",
            "       -5.34026265e-01,  2.66631655e-02, -7.42349625e-02, -3.62982929e-01,\n",
            "       -2.86231130e-01, -1.66131049e-01,  3.86375934e-02,  1.28315389e-01,\n",
            "       -4.45238173e-01,  3.73234749e-02, -2.64324933e-01,  3.01888764e-01,\n",
            "       -2.72308767e-01, -2.92973012e-01, -1.12924032e-01,  1.59693882e-01,\n",
            "       -2.55195126e-02, -1.54406920e-01,  3.55699509e-01,  1.58183575e-01,\n",
            "        4.36467737e-01, -4.01847512e-02,  4.24254648e-02, -2.91574925e-01,\n",
            "        3.96748364e-01,  1.07507400e-01,  3.12549591e-01, -2.46803492e-01,\n",
            "        1.30858600e-01,  2.82017767e-01,  2.35467821e-01,  1.39250755e-01,\n",
            "       -3.01537037e-01, -3.98708642e-01, -1.66164696e-01,  5.24218678e-01,\n",
            "       -1.99164495e-01, -3.18147898e-01, -5.48262224e-02, -5.88554263e-01,\n",
            "        1.19222492e-01,  2.46446103e-01, -3.88847828e-01, -2.70309061e-01,\n",
            "        1.08364984e-01,  5.70084751e-01, -4.68321562e-01, -1.47951972e-02,\n",
            "       -1.49453327e-01,  2.49704942e-01,  6.12155139e-01,  6.53186679e-01,\n",
            "       -3.72651041e-01, -3.84052575e-01,  6.81142807e-02, -4.82062668e-01,\n",
            "        3.18425447e-02,  3.74006480e-03,  4.67332125e-01,  9.39699411e-02,\n",
            "        2.51319826e-01,  5.49959168e-02, -6.66074157e-02,  1.08390473e-01,\n",
            "        3.88469875e-01,  1.08606681e-01, -5.38459539e-01,  1.35666192e-01,\n",
            "        4.95553426e-02,  7.14087114e-03,  2.26598024e-01, -4.23960805e-01,\n",
            "        1.03240073e-01,  2.62356967e-01,  3.21121812e-01,  5.59749529e-02,\n",
            "       -2.95298159e-01, -4.36642736e-01,  6.54033601e-01,  2.34819889e-01,\n",
            "       -1.10796049e-01,  1.54144727e-02,  5.40188789e-01,  4.82166708e-01,\n",
            "       -2.92946041e-01,  2.65587177e-02, -4.35061783e-01, -1.98278353e-01,\n",
            "        2.67434716e-01,  4.06048149e-01,  4.77799505e-01, -5.95480919e-01,\n",
            "       -1.03745520e-01, -6.65384233e-01,  3.40935320e-01, -2.66198099e-01,\n",
            "        1.05347976e-01,  9.52112079e-02,  9.03028399e-02,  3.22835967e-02,\n",
            "       -6.09479547e-01,  9.99834463e-02, -7.42459446e-02, -1.64439142e-01,\n",
            "        1.01296648e-01, -3.31508443e-02,  8.46395493e-02, -8.75661731e-01,\n",
            "        2.30889723e-01, -2.90309191e-01, -1.25343561e-01,  1.70938939e-01,\n",
            "        1.72687560e-01, -2.71916866e-01, -4.12603378e-01,  2.20171567e-02,\n",
            "        7.25049675e-02,  5.28761446e-01,  3.31236184e-01, -4.30730760e-01,\n",
            "        6.50442243e-01,  2.10866749e-01,  9.61202681e-02,  5.25832534e-01,\n",
            "       -4.19128239e-01,  2.81623542e-01, -4.97818701e-02,  1.91642672e-01,\n",
            "       -2.38338947e-01, -4.45311606e-01,  2.20795684e-02, -1.32032588e-01,\n",
            "       -4.25674878e-02, -9.88200456e-02, -6.44333124e-01,  6.27975643e-01,\n",
            "        1.89725026e-01, -6.71027005e-02, -5.12540936e-01,  3.66073614e-03,\n",
            "        1.03709307e-02,  8.74391124e-02, -3.55463982e-01, -3.90915513e-01,\n",
            "        5.30104160e-01,  1.40405908e-01,  1.98005721e-01,  2.36791134e-01,\n",
            "       -1.16167135e+01,  3.04354906e-01,  6.87925279e-01, -3.17160666e-01,\n",
            "        4.92733344e-02,  3.48334759e-03, -1.60090730e-01, -2.05280110e-01,\n",
            "        4.35730934e-01, -3.54191780e-01, -1.53372228e-01,  5.42607978e-02,\n",
            "       -2.87372530e-01, -2.28307903e-01, -4.28358525e-01,  3.67484987e-01,\n",
            "        6.08361140e-02, -4.37085599e-01,  1.23212241e-01, -6.50840163e-01,\n",
            "        4.77993459e-01, -4.37339872e-01, -1.13284171e-01,  2.88096130e-01,\n",
            "       -1.20031834e-01,  4.57728505e-02, -1.86829850e-01,  2.80381083e-01,\n",
            "        2.58535184e-02,  5.30652046e-01, -4.05168742e-01,  2.69450545e-01,\n",
            "       -8.49792957e-02,  2.55552232e-01, -3.23949903e-01, -1.71798021e-01,\n",
            "        3.18854779e-01, -2.28790030e-01,  2.22801089e-01, -1.99753702e-01,\n",
            "       -1.65453553e-01,  6.24768846e-02, -3.90380859e-01,  2.30445683e-01,\n",
            "        8.46816421e-01,  1.94829866e-01,  6.34423912e-01,  5.25448918e-01,\n",
            "        4.18982923e-01, -4.02462423e-01,  4.30058455e-03, -8.75011802e-01,\n",
            "        8.73857588e-02, -7.18944728e-01, -4.49381590e-01, -5.12938760e-02,\n",
            "        3.81800652e-01, -4.07094955e-01,  4.30914104e-01, -3.59740853e-01,\n",
            "        3.12099814e-01,  4.54434976e-02,  6.44790977e-02, -4.23492119e-03,\n",
            "       -6.99049652e-01,  5.38947225e-01, -1.27690718e-01, -2.50598460e-01,\n",
            "       -1.57559603e-01, -2.06945345e-01, -2.69937843e-01, -2.52695918e-01,\n",
            "        2.23457381e-01,  7.54890293e-02,  9.48057532e-01,  1.60032570e-01,\n",
            "       -6.25604033e-01, -3.46169882e-02,  1.15127349e+00, -2.24795982e-01,\n",
            "        6.76103979e-02,  1.05597340e-01,  7.52215385e-02,  1.62364066e-01,\n",
            "       -5.55325508e-01, -6.26082182e-01,  1.49465844e-01,  3.86173837e-02,\n",
            "       -6.76056743e-02,  3.72372836e-01, -1.52546316e-01, -7.32384175e-02,\n",
            "       -2.51550555e-01,  8.92036185e-02,  1.71676025e-01,  1.05863917e+00,\n",
            "       -7.83731937e-01,  3.06241632e-01,  4.63125035e-02,  3.26439515e-02,\n",
            "        1.82286069e-01, -3.83234769e-02, -2.52628624e-01, -5.35362661e-01,\n",
            "       -2.33096167e-01,  2.62792528e-01,  8.48311484e-02,  9.42592621e-02,\n",
            "        3.90076131e-01,  1.19876765e-01, -6.83320612e-02,  1.83687806e-01,\n",
            "        2.93067306e-01,  2.59012997e-01,  9.97398794e-03, -6.63179040e-01,\n",
            "        6.55622900e-01,  2.56804168e-01, -3.32461774e-01, -6.49739802e-01,\n",
            "        1.32032305e-01, -8.32983553e-02,  4.68591064e-01,  9.02416930e-02,\n",
            "       -1.97456330e-01,  3.25093806e-01, -4.97978300e-01, -1.61120355e-01,\n",
            "        1.07073128e-01, -4.76461183e-03,  6.04712188e-01,  4.62982386e-01,\n",
            "       -6.67392492e-01, -3.24596494e-01, -4.22271341e-01, -1.55105993e-01,\n",
            "       -8.29841137e-01, -8.73553306e-02,  3.79362762e-01,  2.74837520e-02,\n",
            "        2.11683959e-01, -2.86881447e-01, -1.69957317e-02,  1.32827610e-01,\n",
            "        7.68924356e-01, -1.92067083e-02,  5.88488698e-01,  3.19933668e-02,\n",
            "       -1.30760759e-01,  4.31760788e-01,  5.63502423e-02, -4.06412780e-01,\n",
            "       -2.94089526e-01, -1.66714400e-01, -3.33388805e-01, -5.80929935e-01,\n",
            "       -1.74464911e-01, -9.90839601e-02,  6.38341546e-01,  1.34557113e-01,\n",
            "        2.87142128e-01, -9.59127843e-02,  2.16665193e-01, -1.63306057e-01,\n",
            "        2.83424377e-01,  3.00063372e-01, -1.50052592e-01, -4.49727535e-01,\n",
            "       -3.55355024e-01, -4.97322977e-02,  4.28409338e-01, -2.45455548e-01,\n",
            "        1.20870575e-01, -2.04302669e-01, -2.54845023e-01, -1.19488418e-01,\n",
            "       -1.05587207e-01,  1.12288499e+00, -5.17248273e-01, -1.36806548e-01,\n",
            "        1.52271232e-02,  2.86691695e-01, -2.53071845e-01, -2.98951030e-01,\n",
            "        2.21870512e-01, -1.09832130e-01, -3.52084637e-03,  5.17510697e-02,\n",
            "        1.78382561e-01,  4.15569358e-02,  2.28840470e-01,  2.54508227e-01,\n",
            "        5.91975357e-03, -3.96000326e-01,  5.76206684e-01,  1.51840225e-03,\n",
            "       -5.73471561e-03,  6.66676974e-03,  1.52011458e-02, -1.15704606e-03,\n",
            "        5.81913352e-01, -4.67929214e-01,  4.05494571e-02, -3.22007835e-01,\n",
            "       -2.43098438e-02, -3.95749748e-01,  3.23690385e-01,  1.15941620e+00,\n",
            "       -3.46467972e-01,  3.12556550e-02,  1.05427456e+00, -1.37100024e-02,\n",
            "        3.84266555e-01,  2.26991534e-01,  2.40328848e-01,  4.39182203e-03,\n",
            "        4.13171977e-01, -1.50976598e-01, -7.64122427e-01, -2.11376846e-01,\n",
            "        2.27164745e-01,  1.52961016e-01, -4.81555074e-01, -4.15649116e-01,\n",
            "        1.60676807e-01,  1.12491570e-01,  3.51136684e-01,  5.94072044e-01,\n",
            "        1.38393044e-01,  2.27360308e-01,  3.12485993e-01, -6.58303440e-01,\n",
            "        4.28482383e-01,  7.31286556e-02,  3.10779929e-01, -2.48797499e-02,\n",
            "       -1.08025454e-01, -2.09284291e-01,  6.72844648e-02, -1.80895805e-01,\n",
            "        4.88289110e-02,  3.07975650e-01, -8.32309484e-01,  1.00254878e-01,\n",
            "        4.30355191e-01, -4.29323047e-01,  1.03314027e-01,  2.46996507e-01,\n",
            "        1.70268103e-01, -2.49049850e-02,  2.24240753e-03, -1.96117222e-01,\n",
            "        1.49527699e-01, -3.17978114e-01, -3.30535322e-01,  2.66497850e-01,\n",
            "        3.82653803e-01, -3.71142626e-02,  1.53534096e-02,  7.85485506e-02,\n",
            "       -3.23352098e-01, -3.05621594e-01,  2.33535290e-01, -3.96422446e-01,\n",
            "       -2.36499518e-01,  1.12634245e-02,  3.21364477e-02, -2.97408074e-01,\n",
            "        5.66738844e-01,  3.10178250e-01, -3.58690262e-01,  1.52758047e-01,\n",
            "        3.80552590e-01,  5.96607804e-01, -2.00560778e-01, -5.23290515e-01,\n",
            "        6.95045218e-02,  4.84836042e-01, -2.24804327e-01,  5.32167196e-01,\n",
            "        1.70078740e-01, -5.28024614e-01,  1.61894903e-01,  9.34094787e-02,\n",
            "       -3.78598452e-01,  3.97655427e-01, -1.80147767e-01, -3.26001614e-01,\n",
            "       -5.70279241e-01, -3.48032683e-01,  6.28809184e-02,  1.93639249e-01,\n",
            "        1.16513632e-01, -3.56013596e-01, -3.35720740e-02,  2.25991964e-01,\n",
            "        9.04152021e-02,  4.90761161e-01,  5.39780483e-02,  4.45907921e-01,\n",
            "       -7.24481940e-02,  1.38460562e-01, -1.20224886e-01, -6.13339655e-02,\n",
            "        1.54089063e-01, -3.27371240e-01, -6.27401710e-01, -6.83232993e-02,\n",
            "        1.57387301e-01,  4.25204188e-02,  3.11638772e-01,  5.35218656e-01,\n",
            "       -2.61862099e-01, -7.83675551e-01, -4.19634759e-01, -2.27061793e-01,\n",
            "       -2.62634009e-02,  1.19636349e-01,  1.71533525e-01, -1.04278319e-01,\n",
            "       -3.26051891e-01, -7.90335357e-01, -6.00791797e-02, -3.14915508e-01,\n",
            "        1.34335563e-01, -9.91414953e-03,  7.46364713e-01,  1.53812632e-01,\n",
            "        1.08079791e-01,  4.35996622e-01,  2.70670533e-01, -4.09006298e-01,\n",
            "        6.08317852e-01, -3.57861161e-01, -4.08563703e-01, -3.76625061e-02,\n",
            "        3.74080002e-01, -4.87689786e-02, -2.43296787e-01, -5.71609974e-01,\n",
            "        4.58807945e-01, -2.42464706e-01,  5.06161094e-01,  3.58844280e-01],\n",
            "      dtype=float32), 'Detailed_description': array([-2.43677855e-01, -4.07487392e-01, -7.97010660e-01, -4.47225690e-01,\n",
            "       -2.72199780e-01, -5.68818748e-01, -5.67676663e-01,  2.14835346e-01,\n",
            "        8.12483802e-02,  1.61060661e-01,  2.06085533e-01,  2.75595784e-01,\n",
            "       -7.93497264e-02, -3.74380529e-01, -5.28229237e-01,  6.10936172e-02,\n",
            "        3.37617218e-01,  2.83273309e-01,  2.60497212e-01,  3.82540643e-01,\n",
            "        3.16064209e-01,  3.41613680e-01, -3.07336636e-03,  1.15623213e-02,\n",
            "       -2.32455045e-01,  3.74276936e-02,  5.38014472e-02, -1.90051153e-01,\n",
            "        4.90605175e-01, -3.55049193e-01,  8.09759498e-02,  5.37312746e-01,\n",
            "       -4.01306719e-01,  7.21255064e-01,  3.50132585e-02, -1.75256327e-01,\n",
            "        1.24142699e-01,  1.53306857e-01,  8.85374010e-01,  3.61847021e-02,\n",
            "       -2.13944346e-01,  4.53515708e-01,  4.39232588e-01,  2.01455802e-01,\n",
            "        2.56182969e-01,  1.50641218e-01,  1.06751494e-01, -4.93716091e-01,\n",
            "        2.02916384e-01, -6.63447455e-02, -4.13475692e-01, -1.04772151e-01,\n",
            "        2.26935148e-01, -3.24052036e-01, -7.83322573e-01, -1.11323766e-01,\n",
            "       -4.69160676e-02,  1.90328375e-01, -2.68151555e-02,  2.64181018e-01,\n",
            "       -2.66668916e-01,  5.77364147e-01,  1.13530755e-02, -2.09678352e-01,\n",
            "       -2.17180550e-01, -3.96681502e-02, -5.82052469e-01, -2.88194150e-01,\n",
            "       -1.11450598e-01, -9.74421725e-02,  1.56195730e-01,  3.82090420e-01,\n",
            "        1.44889265e-01, -9.18555707e-02,  1.90454036e-01,  9.20252502e-03,\n",
            "        4.18182850e-01,  1.25677258e-01, -8.86888802e-02, -3.32425028e-01,\n",
            "        1.17845491e-01,  4.16485578e-01,  1.07569382e-01, -2.68488109e-01,\n",
            "       -2.01692402e-01,  1.42823830e-01,  4.35118049e-01, -7.33999684e-02,\n",
            "        2.66420841e-01,  1.17020682e-01,  7.91874707e-01,  3.89773697e-01,\n",
            "       -2.84076780e-01, -1.33154213e-01,  2.96318531e-01, -2.29452908e-01,\n",
            "        1.37542039e-02, -1.68123022e-01, -7.69719183e-02, -5.83315969e-01,\n",
            "       -1.27759993e-01, -5.27888536e-03,  1.70352831e-01, -2.73038089e-01,\n",
            "       -4.88936193e-02, -1.82892680e-01,  4.90218282e-01, -6.65478036e-03,\n",
            "       -1.97710365e-01, -2.43079811e-01, -1.54639572e-01, -4.01490748e-01,\n",
            "        1.66382179e-01, -4.53146130e-01, -2.01811254e-01,  1.02028549e-02,\n",
            "        1.83530152e-01, -1.78635597e-01, -1.50099754e-01,  2.03292117e-01,\n",
            "        1.08971424e-01, -2.16011286e-01, -1.33587122e-02, -1.30536892e-02,\n",
            "        1.65098310e-01, -2.52511919e-01,  4.43047583e-01, -1.34998158e-01,\n",
            "       -2.36976743e-02,  7.46081352e-01,  8.73448253e-02, -4.53690328e-02,\n",
            "       -4.18930411e-01, -3.31575125e-02,  2.99946815e-01,  2.81912774e-01,\n",
            "       -3.12479854e-01, -3.03040653e-01, -1.16839282e-01,  2.25539565e-01,\n",
            "       -1.78206772e-01,  5.68128787e-02, -1.90877080e-01, -1.68211848e-01,\n",
            "       -1.80139810e-01, -3.44850451e-01,  3.84058833e-01, -5.85301042e-01,\n",
            "       -3.43988955e-01, -3.81662428e-01, -4.86961216e-01, -3.07623237e-01,\n",
            "        2.66911864e-01,  1.89358354e-01,  1.13013923e-01, -2.14121670e-01,\n",
            "       -4.56932150e-02,  1.75489038e-01, -6.52868226e-02,  2.49221712e-01,\n",
            "        3.84192616e-01, -1.56826019e-01,  2.77619988e-01, -2.41507873e-01,\n",
            "       -8.53674561e-02,  2.57689536e-01, -5.22849321e-01, -1.75382033e-01,\n",
            "       -4.12311018e-01, -7.39140511e-02,  9.49325413e-02, -2.58208185e-01,\n",
            "        1.59977138e-01, -1.54866561e-01, -5.66275120e-02, -4.03816432e-01,\n",
            "        8.23044181e-02,  8.80485922e-02,  2.82399684e-01,  2.36779481e-01,\n",
            "        1.99150518e-01, -1.84575629e+00,  4.53883410e-01, -2.86872566e-01,\n",
            "        3.27016190e-02, -3.93664658e-01, -1.67730421e-01, -1.60214335e-01,\n",
            "        2.85332263e-01, -9.43527892e-02, -3.07540327e-01,  3.29773307e-01,\n",
            "       -2.50446737e-01,  1.12137981e-01, -2.91455030e-01,  5.63495398e-01,\n",
            "        5.66831112e-01,  2.43639052e-01,  4.31769341e-03,  4.86810952e-02,\n",
            "       -6.85782254e-01,  2.26989418e-01,  1.93003118e-02,  1.04291759e-01,\n",
            "        8.47349167e-02,  3.07231545e-01,  1.09448284e-01,  2.27707848e-01,\n",
            "        5.34014776e-02,  3.09410989e-02, -8.82781595e-02, -1.75580353e-01,\n",
            "       -2.23819807e-01,  5.66396564e-02, -1.02456920e-01,  8.25519636e-02,\n",
            "       -4.56328958e-01, -4.41167176e-01,  3.89982581e-01,  1.98367655e-01,\n",
            "       -2.15014860e-01, -1.20076954e-01,  2.83890609e-02, -2.35836029e-01,\n",
            "       -2.45104820e-01,  3.21651638e-01, -3.29676867e-01,  2.70034939e-01,\n",
            "       -1.38800964e-02, -1.93851084e-01, -5.33874109e-02, -4.63349819e-01,\n",
            "        3.96818101e-01,  5.63197210e-02,  4.63338420e-02, -8.07604715e-02,\n",
            "       -2.65248477e-01,  7.19239950e-01, -1.91963926e-01, -3.17693144e-01,\n",
            "       -6.57451302e-02, -4.61802602e-01, -5.12717009e-01,  1.33690432e-01,\n",
            "        4.70216453e-01,  1.64649218e-01,  6.49103224e-02,  8.36891353e-01,\n",
            "        7.79886782e-01,  4.34291959e-01,  2.42179096e-01,  2.39948958e-01,\n",
            "        5.56986213e-01,  3.39886010e-01, -7.24153519e-02,  1.44638538e-01,\n",
            "        2.77153283e-01, -1.16495127e-02,  4.29484278e-01, -1.15842000e-01,\n",
            "       -2.15006530e-01, -1.66918322e-01,  1.39191121e-01,  2.53253192e-01,\n",
            "        1.16530433e-02,  2.01782107e-01,  2.58189321e-01, -1.40571788e-01,\n",
            "       -2.71071136e-01,  2.17374220e-01, -2.66620964e-01,  1.12745002e-01,\n",
            "       -1.56862274e-01, -7.64038861e-02, -6.72316849e-01, -7.71594644e-01,\n",
            "       -2.22982362e-01, -2.84734070e-02,  4.59615774e-02, -3.67687047e-01,\n",
            "       -6.44932508e-01, -6.35634661e-02, -2.51466315e-02,  7.95950219e-02,\n",
            "        7.12691471e-02,  1.92576349e-01, -1.19371638e-01,  4.19100747e-02,\n",
            "       -2.05272645e-01, -4.65734124e-01, -2.65319377e-01,  1.85745746e-01,\n",
            "       -1.64942533e-01,  1.91878274e-01,  6.04796782e-02, -2.62036145e-01,\n",
            "        5.46766892e-02, -1.29150167e-01, -2.64448017e-01, -1.97276741e-01,\n",
            "       -2.79971436e-02, -8.00994635e-02, -1.77034467e-01,  3.56299058e-02,\n",
            "        3.83850932e-02,  3.90999131e-02,  4.04019020e-02,  4.04197991e-01,\n",
            "       -4.33055341e-01, -4.52160239e-01,  2.50693820e-02,  1.53202683e-01,\n",
            "        4.96459752e-03,  1.27562404e-01, -1.03879973e-01, -6.93168163e-01,\n",
            "        6.12359345e-02, -3.06235775e-02, -1.11561209e-01, -2.83739746e-01,\n",
            "        2.33173057e-01,  3.46670777e-01, -1.95283517e-01,  5.71387224e-02,\n",
            "       -1.45435140e-01,  2.24814385e-01,  3.33956629e-01,  2.88746059e-01,\n",
            "       -1.07315645e-01, -4.24650311e-01, -1.68632045e-01,  1.51322410e-02,\n",
            "       -2.38860890e-01, -1.00957125e-01,  4.08076435e-01,  1.18072450e-01,\n",
            "        2.43208170e-01, -4.44794893e-02, -3.74459438e-02, -6.79420978e-02,\n",
            "        4.59152639e-01, -1.05573058e-01, -2.87338614e-01,  3.79824460e-01,\n",
            "       -1.79604858e-01, -2.15410337e-01,  2.27445334e-01, -1.27797365e-01,\n",
            "       -1.01071641e-01,  2.88608313e-01,  4.09997821e-01,  8.94979089e-02,\n",
            "       -2.36379683e-01, -1.69958115e-01,  2.70184338e-01,  5.86400628e-01,\n",
            "        4.31532264e-02,  2.65158899e-02,  5.57174921e-01,  2.91623890e-01,\n",
            "       -1.81253374e-01,  1.23269252e-01, -6.13989115e-01, -1.45391256e-01,\n",
            "        1.62094802e-01,  1.91708684e-01,  3.72621596e-01, -2.64565706e-01,\n",
            "       -2.40526646e-01, -3.55421454e-02,  1.43280759e-01, -3.68494391e-01,\n",
            "        3.55685502e-02,  1.99867606e-01,  1.90510541e-01,  1.41139686e-01,\n",
            "       -5.63907385e-01,  1.86951265e-01, -1.81618154e-01, -4.33906674e-01,\n",
            "       -2.13030636e-01, -1.11337334e-01, -1.60369687e-02, -5.60145080e-01,\n",
            "       -4.53240097e-01, -8.92953202e-02, -2.59766698e-01,  3.03350717e-01,\n",
            "        2.42984936e-01,  1.37100667e-01, -6.83314353e-02, -7.40301013e-02,\n",
            "       -2.49249667e-01,  3.02050591e-01,  4.94325399e-01, -4.05945659e-01,\n",
            "        4.79204267e-01,  5.22268564e-02,  1.95291460e-01,  5.10044694e-01,\n",
            "       -3.84954751e-01, -4.84741926e-01, -1.04318276e-01,  9.64796543e-02,\n",
            "       -1.96636349e-01, -4.16345239e-01, -3.17959636e-02,  8.50793868e-02,\n",
            "       -3.89139652e-01,  2.83950800e-03, -5.76102734e-01,  3.45657647e-01,\n",
            "        4.60400358e-02,  2.81781316e-01, -4.88717854e-02,  1.56183213e-01,\n",
            "       -2.03082740e-01,  7.83266500e-03,  2.19513685e-01, -2.38393873e-01,\n",
            "        7.24402785e-01,  9.03436095e-02,  2.36158535e-01,  3.97657812e-01,\n",
            "       -1.11973057e+01,  2.28539750e-01,  7.76860476e-01, -2.89974332e-01,\n",
            "       -9.92409438e-02, -4.89866249e-02,  1.69115499e-01, -2.12047994e-03,\n",
            "        4.69166607e-01, -5.94343781e-01,  5.05911186e-02, -4.46916223e-02,\n",
            "       -1.96145475e-01, -2.71290898e-01, -5.18933892e-01, -1.92811042e-01,\n",
            "        9.13440883e-02, -5.40619731e-01,  9.08443630e-02, -9.56281722e-02,\n",
            "        4.56057847e-01, -4.59562778e-01, -1.31340846e-01,  3.43848854e-01,\n",
            "       -2.83731520e-03, -4.38227743e-01,  1.50186718e-01, -1.73125267e-01,\n",
            "        2.78923661e-02,  2.91552186e-01, -4.04790938e-02,  2.41390139e-01,\n",
            "        8.82017091e-02,  1.15915880e-01, -2.16303289e-01, -1.93454355e-01,\n",
            "        4.00866479e-01, -3.97710979e-01, -4.20497954e-02, -1.71012864e-01,\n",
            "       -2.21919358e-01, -1.41859487e-01, -1.49669200e-01,  2.48365760e-01,\n",
            "        2.88676172e-01,  6.77496672e-01,  1.37132621e+00,  1.00372148e+00,\n",
            "       -3.11731696e-02, -6.67113438e-02, -2.54357815e-01, -7.27760315e-01,\n",
            "        1.37848973e-01, -2.34429643e-01, -2.90493369e-01,  4.52032685e-03,\n",
            "        2.55112231e-01, -4.10666525e-01,  6.41017437e-01, -3.12404841e-01,\n",
            "        2.01240256e-01,  3.25316638e-01,  1.41559213e-01,  2.73984224e-01,\n",
            "       -6.06599450e-01,  2.87632853e-01,  1.93500698e-01, -2.05449536e-01,\n",
            "       -8.89550373e-02, -4.57637638e-01,  1.15658507e-01, -1.47000194e-01,\n",
            "        2.98285633e-01,  1.93737313e-01,  4.55435216e-01, -1.42662019e-01,\n",
            "       -5.44749200e-01, -3.34877335e-02,  1.22422743e+00,  1.21138096e-02,\n",
            "        9.41702947e-02, -6.59598634e-02,  1.84009224e-01,  6.61164224e-02,\n",
            "       -5.37392616e-01, -2.99712151e-01,  2.59097815e-02, -7.55904555e-01,\n",
            "        5.09556532e-01,  2.03809276e-01, -3.16092819e-01, -1.45487219e-01,\n",
            "        2.53299940e-02, -2.70530488e-03,  3.06197643e-01,  3.74175310e-01,\n",
            "       -7.02948570e-01, -1.86999440e-02, -4.04457599e-02,  1.13661423e-01,\n",
            "       -1.07438982e-01, -2.22564831e-01, -5.41363657e-02, -2.76620716e-01,\n",
            "        1.15953982e-01,  3.82659137e-01,  4.09748740e-02,  1.59965619e-01,\n",
            "       -5.38499504e-02,  1.72824144e-01, -2.07877129e-01,  1.00414723e-01,\n",
            "       -7.08871633e-02,  3.48613143e-01, -3.30077112e-01, -4.71169800e-01,\n",
            "        4.62123245e-01,  3.09669644e-01, -7.01111257e-01, -3.45310330e-01,\n",
            "       -4.44533646e-01,  8.73244181e-02,  1.88453153e-01,  1.38130710e-01,\n",
            "       -4.58002418e-01,  3.21818709e-01, -5.87272882e-01, -9.43751857e-02,\n",
            "        1.70438156e-01,  1.91733614e-01,  3.92224729e-01,  3.35130692e-01,\n",
            "       -2.64072120e-01,  2.83761084e-01, -2.56540686e-01, -5.23170173e-01,\n",
            "       -1.11446953e+00, -1.10150538e-01,  5.38615227e-01, -5.22079408e-01,\n",
            "        1.36871159e-01, -7.79358670e-02, -5.66953495e-02, -1.30376980e-01,\n",
            "        8.46915990e-02,  3.15905288e-02,  6.11605883e-01,  1.54263914e-01,\n",
            "       -3.67329001e-01,  5.39880693e-01, -1.04902372e-01, -2.74024189e-01,\n",
            "       -2.46240497e-01, -3.25825930e-01, -3.35052639e-01, -3.47081333e-01,\n",
            "       -4.23887432e-01,  8.32226425e-02, -1.07637212e-01,  3.36749762e-01,\n",
            "       -5.78706712e-02, -3.01074028e-01, -2.16249079e-01, -2.04954177e-01,\n",
            "        5.36837399e-01,  3.06112438e-01,  1.54809654e-01, -1.83217257e-01,\n",
            "       -1.87006414e-01, -1.93240598e-01,  3.13197941e-01,  3.27909261e-01,\n",
            "        1.96657181e-01, -4.56063688e-01, -6.25081837e-01,  3.98058817e-02,\n",
            "       -1.72713757e-01,  6.89676762e-01, -7.75907516e-01, -2.59337872e-01,\n",
            "        2.76230484e-01,  6.77382946e-02, -4.82348800e-02, -3.59472960e-01,\n",
            "        2.20629573e-01,  2.03144670e-01,  7.36820847e-02, -2.02250749e-01,\n",
            "       -9.52784568e-02, -1.14685014e-01,  1.61686599e-01,  4.00187850e-01,\n",
            "        2.64717460e-01, -3.80175769e-01,  4.70692925e-02, -2.46135429e-01,\n",
            "       -2.73490906e-01,  1.55220255e-02,  6.53953701e-02,  2.28749678e-01,\n",
            "       -1.36309415e-02, -2.90700555e-01,  3.60250950e-01, -3.66955042e-01,\n",
            "        1.00367315e-01, -1.20005503e-01,  2.77700782e-01,  1.29642534e+00,\n",
            "       -1.27232254e-01, -4.77910712e-02,  1.06056225e+00,  2.07297713e-01,\n",
            "        9.79478583e-02,  3.05453353e-02, -4.31902707e-03,  2.48142362e-01,\n",
            "        3.93357456e-01, -2.93041766e-01, -2.92641968e-01,  2.04443261e-01,\n",
            "        5.21726131e-01,  1.99270114e-01, -4.01426584e-01, -3.02760810e-01,\n",
            "       -1.20593250e-01,  2.44740337e-01, -2.29770154e-01,  1.47049785e-01,\n",
            "       -2.54173696e-01,  1.93758339e-01,  1.28734604e-01, -8.49681139e-01,\n",
            "        1.37953699e-01, -5.35562336e-01,  8.35494876e-01, -4.45637107e-03,\n",
            "       -2.19887733e-01, -1.57673627e-01, -1.75665438e-01, -1.45234197e-01,\n",
            "        2.21945032e-01,  4.53758001e-01, -8.14359307e-01,  1.09774545e-01,\n",
            "        5.51080048e-01, -1.12427607e-01,  2.43454516e-01, -2.46106744e-01,\n",
            "        3.87034088e-01,  1.99275762e-01,  4.26532812e-02,  3.07009444e-02,\n",
            "        1.28938273e-01, -3.97720963e-01, -4.69610929e-01,  1.47632748e-01,\n",
            "        3.43705177e-01,  1.51815191e-02,  4.07934040e-02,  2.07058579e-01,\n",
            "       -3.15281987e-01, -1.56910464e-01,  7.95401111e-02, -2.80013144e-01,\n",
            "        7.73755461e-02, -2.88748622e-01, -1.13630027e-01, -1.90876901e-01,\n",
            "        6.43987805e-02,  2.04913452e-01, -1.18301958e-02,  6.96053356e-02,\n",
            "        1.34347260e-01,  7.57343292e-01, -4.62216586e-02, -2.44494364e-01,\n",
            "        2.81228393e-01,  8.41105342e-01,  5.31541780e-02,  6.48983240e-01,\n",
            "       -1.78619236e-01, -2.54485607e-01,  3.29791009e-01,  8.30513462e-02,\n",
            "       -3.41463894e-01,  3.75072956e-01, -1.28951237e-01, -1.70074314e-01,\n",
            "       -3.27132642e-01, -3.99864793e-01,  3.82352561e-01,  2.75475681e-01,\n",
            "        2.08853573e-01, -5.02994120e-01, -8.97417739e-02,  2.24328279e-01,\n",
            "       -1.57621264e-01,  2.13795394e-01, -9.92743522e-02,  5.18176138e-01,\n",
            "       -7.30764031e-01, -6.87427819e-02, -4.64205325e-01,  9.44816694e-02,\n",
            "        1.62857741e-01, -2.08342299e-02, -3.16958070e-01, -2.23320588e-01,\n",
            "        3.12789798e-01,  2.37185601e-02,  6.63480163e-01,  6.11043751e-01,\n",
            "       -2.09037513e-01, -1.98292315e-01, -3.02017123e-01, -8.27004984e-02,\n",
            "       -2.12169737e-01, -2.17022777e-01,  2.54167289e-01,  1.44431561e-01,\n",
            "       -3.92413735e-01, -5.86995006e-01, -4.54292387e-01, -2.89888866e-02,\n",
            "        5.41313529e-01, -1.19144663e-01,  4.13807034e-01, -5.21490015e-02,\n",
            "        4.44653392e-01,  3.25311065e-01,  3.15887183e-01,  8.32532644e-02,\n",
            "        3.11679006e-01, -3.93499076e-01, -4.03399646e-01,  6.17887415e-02,\n",
            "        4.78581823e-02, -2.04310596e-01, -3.88444811e-02, -1.01947999e+00,\n",
            "        3.68219167e-01,  1.47258252e-01,  4.36007977e-01,  5.23311853e-01],\n",
            "      dtype=float32), 'Biological_structure': array([-2.90327668e-01, -3.34549963e-01, -6.47480667e-01, -5.37075818e-01,\n",
            "       -2.10533112e-01,  6.20298088e-02, -8.94713342e-01,  4.50375348e-01,\n",
            "        2.16747135e-01,  3.16687375e-01,  2.35607058e-01,  2.57525742e-01,\n",
            "       -1.75452083e-02, -7.04821885e-01, -2.98282087e-01, -1.09204203e-02,\n",
            "        7.91448429e-02,  3.20218205e-01, -1.17863193e-02,  1.58199042e-01,\n",
            "        9.67107058e-01,  2.71392465e-01, -3.30712646e-02, -7.30457529e-02,\n",
            "        5.76063395e-02,  1.66030541e-01,  9.57161635e-02,  9.29138586e-02,\n",
            "        4.73590523e-01, -7.09220767e-01,  1.47753134e-01,  2.28197008e-01,\n",
            "       -1.79332867e-01,  7.87472844e-01, -2.39903539e-01, -5.01235485e-01,\n",
            "       -3.21591422e-02, -9.84479636e-02,  5.96409202e-01,  1.65293798e-01,\n",
            "        4.67739224e-01,  3.63701969e-01,  1.37159109e-01,  1.87964454e-01,\n",
            "        1.98330164e-01,  3.28112125e-01,  1.66396618e-01, -6.22334182e-01,\n",
            "        4.00196850e-01, -3.70723754e-01, -1.82089344e-01,  9.68147367e-02,\n",
            "        1.52044520e-01,  2.00132996e-01, -6.90460801e-01,  2.26602256e-02,\n",
            "       -7.25669414e-02,  1.39371008e-01, -3.12680125e-01,  6.14432335e-01,\n",
            "       -1.03735715e-01,  5.55051446e-01, -1.13325693e-01, -1.73460729e-02,\n",
            "        4.56198379e-02, -8.06889832e-02, -3.75457704e-01, -4.73585166e-02,\n",
            "       -4.01525110e-01,  2.03801841e-02, -1.58801645e-01, -1.34149641e-01,\n",
            "       -2.37255424e-01,  1.75952896e-01,  2.10730374e-01, -4.11981076e-01,\n",
            "        2.08625734e-01, -3.65040958e-01, -1.70716375e-01, -2.19575286e-01,\n",
            "        1.29547849e-01,  3.84646982e-01,  1.46327168e-03, -2.81055808e-01,\n",
            "       -2.75359333e-01,  6.66399896e-02,  2.02416509e-01,  3.54862124e-01,\n",
            "       -3.69459465e-02, -8.02685767e-02,  7.83886760e-02,  1.84193492e-01,\n",
            "       -8.79291892e-01, -6.98174834e-02,  1.86890364e-01, -1.79467797e-01,\n",
            "       -3.85750458e-02, -3.15420508e-01, -2.49707043e-01, -2.34214351e-01,\n",
            "        1.55732960e-01, -2.31323108e-01,  1.36947781e-01,  2.19956134e-02,\n",
            "       -6.33789778e-01, -3.31490040e-01,  2.12072015e-01, -3.30212712e-02,\n",
            "       -7.56594062e-01, -1.36485308e-01, -2.74372578e-01, -2.74849206e-01,\n",
            "        5.56061566e-02, -6.09742820e-01, -2.76090447e-02, -2.03311928e-02,\n",
            "        2.96135932e-01,  1.20574161e-01, -2.41398156e-01,  2.82822788e-01,\n",
            "        1.11320406e-01, -3.45047116e-01,  1.33795291e-02, -2.53346622e-01,\n",
            "        2.42699012e-01, -1.97195724e-01,  4.30275679e-01, -1.74415603e-01,\n",
            "        9.07838568e-02,  6.72326446e-01, -2.66578887e-02, -2.71830022e-01,\n",
            "       -7.80221403e-01, -5.16555130e-01, -4.54894453e-02,  5.38422763e-02,\n",
            "        5.60145304e-02, -4.53499019e-01,  5.97186387e-03,  1.06059283e-01,\n",
            "        1.73377037e-01, -5.96121177e-02, -6.61096156e-01,  4.77902628e-02,\n",
            "       -2.55418897e-01, -3.54849756e-01,  6.27869368e-02, -5.39129674e-01,\n",
            "       -5.93460858e-01, -6.26763552e-02, -6.50889397e-01, -4.96923894e-01,\n",
            "        6.34160399e-01, -7.18559846e-02,  3.87965068e-02, -1.81347102e-01,\n",
            "        5.71131051e-01, -6.57984763e-02, -2.37150043e-02,  2.96346277e-01,\n",
            "        3.76741827e-01, -3.13094825e-01,  9.44920927e-02, -4.21347737e-01,\n",
            "        1.13809913e-01,  3.34001184e-01, -3.34716350e-01, -6.74367994e-02,\n",
            "        1.22897752e-01,  1.17374361e-01,  2.19638690e-01, -2.69498318e-01,\n",
            "       -1.21945485e-01, -3.44936252e-01,  5.77417836e-02, -3.16116810e-01,\n",
            "        8.77785832e-02, -2.16566935e-01, -1.01395428e-01,  2.57587641e-01,\n",
            "        1.64247811e-01, -2.10495281e+00, -1.38538465e-01, -4.63150918e-01,\n",
            "        1.23386562e-01, -7.79896975e-02,  2.84955055e-02,  2.38169521e-01,\n",
            "        4.95968878e-01, -2.18650103e-01,  6.62387908e-03,  1.65712565e-01,\n",
            "        8.26410949e-04,  1.29082248e-01, -5.48645630e-02,  5.59337854e-01,\n",
            "        3.12714338e-01,  4.21794027e-01, -9.60402340e-02,  1.19155563e-01,\n",
            "       -2.16508225e-01,  3.51394743e-01,  1.34617537e-01, -1.99026406e-01,\n",
            "        7.96794295e-02,  1.16723571e-02, -1.21231042e-01, -4.88936678e-02,\n",
            "       -1.26974314e-01,  8.16167071e-02, -1.09563936e-02, -2.10581347e-01,\n",
            "       -4.79074091e-01,  1.08977079e-01, -1.81154907e-03,  5.38822651e-01,\n",
            "       -1.98446855e-01, -5.91922283e-01,  3.79956007e-01,  3.21638823e-01,\n",
            "       -3.18630934e-02,  2.51980834e-02,  1.46198459e-02, -3.40920925e-01,\n",
            "       -4.33871448e-02,  4.33082402e-01, -4.16864276e-01,  3.52384262e-02,\n",
            "       -2.18276769e-01, -2.04133779e-01, -1.97418779e-03, -2.53479004e-01,\n",
            "        4.28933620e-01,  2.43998021e-02,  5.70611730e-02, -8.90590250e-04,\n",
            "       -1.56539947e-01,  3.26402247e-01, -3.35125446e-01,  2.90661722e-01,\n",
            "        4.64691147e-02, -2.87075013e-01, -4.33580101e-01,  7.29916990e-02,\n",
            "        4.68130112e-01,  1.69685692e-01, -5.42547405e-02,  4.69930172e-01,\n",
            "        6.32712603e-01,  5.84978342e-01,  2.64272746e-02,  5.46615794e-02,\n",
            "        4.74322736e-01,  3.84450734e-01, -2.14114964e-01, -7.96527490e-02,\n",
            "        1.73532784e-01,  9.85327587e-02,  5.61393350e-02, -4.88163084e-01,\n",
            "       -3.01301330e-01, -2.01591905e-02,  2.22200051e-01,  7.93712616e-01,\n",
            "       -2.54211307e-01, -1.74007326e-01,  3.61181587e-01, -4.31771934e-01,\n",
            "       -2.28753924e-01, -3.55966896e-01, -2.53872812e-01,  2.54321963e-01,\n",
            "       -4.26910073e-01, -1.55529827e-01, -5.60308814e-01, -7.53468513e-01,\n",
            "       -4.86522526e-01, -2.81455010e-01,  2.44444907e-01, -4.60794508e-01,\n",
            "       -1.45888895e-01, -2.12548122e-01,  7.48869926e-02,  4.43694964e-02,\n",
            "       -1.56302303e-01, -4.50337976e-02, -2.96709895e-01, -6.49488568e-02,\n",
            "       -6.93804622e-02, -2.27032751e-01,  9.56543982e-02,  2.23654509e-01,\n",
            "       -5.02218127e-01,  8.31921771e-02,  1.71346411e-01,  1.04876295e-01,\n",
            "        3.28713477e-01,  2.68781513e-01, -8.98119956e-02, -2.64543891e-02,\n",
            "       -9.56992805e-02,  4.69244212e-01,  1.53043419e-01, -5.29272966e-02,\n",
            "        2.49034390e-02,  2.09901929e-01,  4.52290773e-02,  3.19114387e-01,\n",
            "       -3.64330947e-01, -2.46927947e-01, -1.04408771e-01,  5.29215455e-01,\n",
            "       -1.56788737e-01, -1.08810484e-01,  1.40438601e-02, -4.02273595e-01,\n",
            "       -1.36022717e-02,  1.39393345e-01, -3.56577486e-01,  1.66616082e-01,\n",
            "        6.53264523e-02,  1.41916901e-01, -1.46985948e-01,  8.62634629e-02,\n",
            "       -4.19997603e-01,  3.66119713e-01,  5.89340329e-01,  1.15724623e-01,\n",
            "       -1.83794841e-01, -4.43174422e-01, -1.61105931e-01,  9.76827219e-02,\n",
            "       -4.48124319e-01, -2.53701627e-01, -8.65333825e-02,  3.42336088e-01,\n",
            "        3.47516716e-01, -4.11389880e-02, -7.25111440e-02, -6.27348572e-03,\n",
            "        3.60774279e-01, -8.53703246e-02, -2.51683146e-01,  3.97882730e-01,\n",
            "        1.65368468e-01, -3.80831987e-01,  2.54335105e-01, -2.52097428e-01,\n",
            "       -3.50334525e-01,  3.77342522e-01,  2.04148591e-01, -3.60746980e-01,\n",
            "       -2.26295561e-01, -4.22048181e-01,  9.06558514e-01,  4.11407173e-01,\n",
            "        2.87195116e-01, -2.90730417e-01,  4.97784615e-01,  4.17850614e-01,\n",
            "       -3.22292715e-01,  1.35004848e-01,  2.00275183e-01, -1.37391344e-01,\n",
            "        1.23258457e-01,  4.57151443e-01,  2.53789157e-01, -3.26175779e-01,\n",
            "       -4.03992161e-02, -9.82746184e-02,  3.47693831e-01, -2.44068146e-01,\n",
            "       -1.84609875e-01,  2.32245117e-01,  7.83832192e-01, -2.38025740e-01,\n",
            "       -5.62695503e-01, -1.14371069e-02, -3.23070198e-01, -5.37807047e-01,\n",
            "       -1.05482966e-01, -1.40370309e-01,  4.70999703e-02, -4.59417671e-01,\n",
            "       -1.80935010e-01, -2.91614026e-01, -6.70274049e-02,  2.26361617e-01,\n",
            "        6.00232556e-03,  3.57279554e-03,  5.25331274e-02,  8.11701044e-02,\n",
            "       -1.18519992e-01,  5.87996840e-01,  9.51927826e-02, -1.15651637e-01,\n",
            "        3.66916060e-01, -7.76025951e-02,  2.33057559e-01,  1.32518858e-01,\n",
            "       -4.57832843e-01, -4.18311656e-01, -2.74911106e-01,  3.42868090e-01,\n",
            "        1.52331948e-01, -1.01439536e-01,  2.14936689e-01,  3.26916695e-01,\n",
            "       -9.69076455e-02, -6.22797310e-02, -5.79252899e-01,  2.64071316e-01,\n",
            "        6.31294101e-02, -2.32789338e-01, -3.13677549e-01,  3.81352723e-01,\n",
            "       -3.08161885e-01, -3.07365060e-01,  2.26317823e-01, -2.00717166e-01,\n",
            "        6.76991105e-01, -1.51515841e-01,  1.28997356e-01,  4.85629916e-01,\n",
            "       -1.13681936e+01,  5.57235181e-01,  4.66548264e-01, -1.10967480e-01,\n",
            "        2.23778605e-01, -1.12511124e-02,  1.48364246e-01, -2.05593497e-01,\n",
            "        2.23010004e-01, -2.87496865e-01,  5.49932234e-02, -9.22185555e-02,\n",
            "       -1.68972343e-01, -6.28872514e-02, -3.61351550e-01, -6.42987266e-02,\n",
            "       -1.28785595e-02, -8.61278534e-01,  8.86474699e-02, -1.95779860e-01,\n",
            "        4.22800481e-01, -4.72503901e-01, -2.63715804e-01,  1.63183913e-01,\n",
            "       -1.56477153e-01, -1.70913160e-01,  3.14716578e-01, -6.48569167e-02,\n",
            "       -1.93700284e-01,  2.81609684e-01, -6.98851198e-02,  6.55532107e-02,\n",
            "       -2.49984950e-01,  1.02942385e-01, -3.03779304e-01, -3.89479518e-01,\n",
            "        1.04967035e-01, -2.17505991e-02,  9.45636630e-03,  3.34054828e-01,\n",
            "       -4.47493136e-01, -1.60697639e-01, -5.28548956e-01,  3.15850914e-01,\n",
            "        6.68568611e-01,  4.26515669e-01,  9.48651731e-01,  6.42513633e-01,\n",
            "        1.53780252e-01, -6.59525692e-02, -1.38254389e-01, -5.60153306e-01,\n",
            "        1.51382893e-01, -7.29544401e-01,  2.72882700e-01, -3.23213637e-02,\n",
            "        6.85941279e-01, -7.08648562e-01,  1.02030717e-01, -1.69793934e-01,\n",
            "       -4.20772284e-02,  5.01943350e-01,  1.63107097e-01,  1.40268043e-01,\n",
            "       -9.47681144e-02, -3.37531567e-01,  4.36585128e-01,  3.38200107e-02,\n",
            "        2.21521277e-02, -5.84421158e-01, -1.56881690e-01, -1.47716030e-01,\n",
            "        6.36180818e-01,  9.31520909e-02, -4.07624207e-02,  5.39061308e-01,\n",
            "        2.87643164e-01,  1.28970370e-02,  1.07935786e+00,  7.14419857e-02,\n",
            "        2.05324009e-01,  2.55815446e-01,  8.82537514e-02, -5.22932827e-01,\n",
            "       -4.32626635e-01, -6.48369268e-02, -7.89970607e-02,  3.77328023e-02,\n",
            "        1.65219203e-01,  1.32823735e-01,  3.88019308e-02, -2.33148873e-01,\n",
            "       -4.71370757e-01,  2.46029198e-01,  2.67309219e-01,  5.14705300e-01,\n",
            "       -7.36818254e-01,  3.16470027e-01, -1.00906849e-01,  1.95637912e-01,\n",
            "        2.03487650e-03, -5.46688177e-02, -1.72166184e-01, -3.54676783e-01,\n",
            "       -1.63396150e-02, -3.56968284e-01,  1.25959754e-01,  1.45605039e-02,\n",
            "        1.50669470e-01,  9.60396752e-02, -7.36998320e-02,  3.18883777e-01,\n",
            "        9.79755446e-02,  1.44990146e-01, -1.28174171e-01, -4.06085730e-01,\n",
            "        2.26163894e-01,  5.61289191e-02, -6.89210296e-01, -1.65693730e-01,\n",
            "       -1.76786944e-01,  3.92951369e-02,  2.71482050e-01,  1.28598571e-01,\n",
            "       -5.08207381e-01,  1.26004010e-01, -9.97572064e-01,  1.39635503e-01,\n",
            "       -3.33624706e-02,  2.20888734e-01,  3.38237435e-01,  4.13342595e-01,\n",
            "        1.04889706e-01,  9.47692767e-02, -4.19056803e-01, -2.46584773e-01,\n",
            "       -1.13652730e+00,  1.67951956e-01,  3.67121279e-01, -1.85851961e-01,\n",
            "        9.12722722e-02, -1.32304460e-01, -1.44257203e-01, -1.49876252e-01,\n",
            "        3.22558999e-01, -9.20984596e-02,  6.06397986e-01,  7.52237439e-02,\n",
            "       -5.54417014e-01, -2.58382410e-01, -5.24549745e-02, -3.30540299e-01,\n",
            "       -4.71824139e-01,  3.05000618e-02, -7.42294192e-02, -6.32528186e-01,\n",
            "       -4.51214314e-02, -3.35239619e-02,  4.62836236e-01, -2.46612370e-01,\n",
            "       -1.08985715e-01, -2.11986955e-02, -1.23198792e-01,  9.58652049e-02,\n",
            "        2.95356452e-01,  4.11789566e-01,  1.44319862e-01, -2.52654850e-01,\n",
            "       -2.75600970e-01,  1.08349383e-01,  5.96393496e-02,  9.18779969e-02,\n",
            "        3.08262765e-01,  5.77794090e-02, -3.54323715e-01, -4.25713509e-01,\n",
            "       -3.49418700e-01,  1.02728486e+00, -4.74255264e-01, -3.86559844e-01,\n",
            "       -2.11406261e-01,  3.08464319e-01, -1.13649353e-01, -2.58764535e-01,\n",
            "        3.25154364e-02,  3.60395491e-01, -3.24177518e-02, -2.39517868e-01,\n",
            "        1.63187295e-01,  1.55815467e-01,  2.89315224e-01,  2.56930083e-01,\n",
            "        4.04348701e-01, -4.81758177e-01,  4.92047340e-01,  2.76657134e-01,\n",
            "       -1.63120642e-01, -3.27631757e-02, -3.92489105e-01,  3.78556475e-02,\n",
            "        4.83170807e-01,  1.66870803e-01,  2.52261341e-01, -2.51677603e-01,\n",
            "       -1.73762832e-02, -1.24549493e-02,  5.35248876e-01,  1.20761585e+00,\n",
            "       -4.04690206e-03,  2.22249508e-01,  8.17512035e-01,  2.71687984e-01,\n",
            "       -2.36541517e-02,  2.29188755e-01,  2.91238785e-01,  3.90476920e-02,\n",
            "        6.43450141e-01, -1.26447529e-01, -3.92576814e-01,  2.04549700e-01,\n",
            "        2.56569803e-01,  1.77441984e-01, -3.54985714e-01, -5.46739042e-01,\n",
            "       -1.32595032e-01,  3.84399533e-01,  4.64695990e-02,  2.90117323e-01,\n",
            "       -2.48053521e-01, -1.87435597e-02,  3.07002783e-01, -5.78027725e-01,\n",
            "        6.06933758e-02,  3.51981819e-03,  1.95588708e-01, -7.34816790e-02,\n",
            "       -1.88134879e-01, -1.94258884e-01, -6.92477673e-02, -2.43875891e-01,\n",
            "        6.54446781e-02,  3.87450755e-01, -1.08293104e+00, -2.60348246e-02,\n",
            "        5.13030767e-01, -6.02900743e-01, -1.19028896e-01, -1.98288798e-01,\n",
            "        8.60594958e-02,  1.03036106e-01,  2.25977749e-01, -2.05106810e-01,\n",
            "        1.73362195e-01, -1.95158750e-01, -4.96920556e-01,  2.89002001e-01,\n",
            "        5.65597177e-01, -4.49214876e-02,  8.03121328e-02, -1.05568253e-01,\n",
            "       -4.11024928e-01, -8.85112062e-02, -2.54981667e-01, -2.15280890e-01,\n",
            "       -2.18416080e-01,  8.88391584e-02, -2.29665369e-01, -4.42178130e-01,\n",
            "        4.84739207e-02,  1.35436833e-01,  2.79557794e-01,  3.99436355e-01,\n",
            "        6.58979177e-01,  4.76326168e-01, -2.01241612e-01, -2.42941707e-01,\n",
            "       -9.48705971e-02,  4.20545638e-01,  4.84054536e-02,  4.20484185e-01,\n",
            "       -8.97224918e-02, -3.39373350e-01,  4.59673762e-01,  1.64222553e-01,\n",
            "       -3.19263428e-01,  5.01855016e-01, -4.51898038e-01,  8.46818686e-02,\n",
            "       -5.40986478e-01, -3.58156741e-01, -6.42182231e-02,  2.23142892e-01,\n",
            "        1.62118413e-02, -2.00593606e-01,  9.20651928e-02,  3.73646796e-01,\n",
            "        8.65699574e-02,  2.86715388e-01,  1.95852995e-01,  5.52561402e-01,\n",
            "       -8.37918743e-02,  1.04937226e-01, -5.77121615e-01, -4.43592407e-02,\n",
            "       -2.64392257e-01,  3.08970749e-01, -2.21299320e-01, -1.58532575e-01,\n",
            "        3.24933499e-01,  2.57003248e-01,  7.04582810e-01,  4.70509887e-01,\n",
            "       -2.69530207e-01, -4.70027208e-01, -4.98402625e-01, -1.76505938e-01,\n",
            "       -8.06422681e-02,  4.69139293e-02,  3.19031417e-01,  1.11953586e-01,\n",
            "        1.62382513e-01, -6.13446891e-01, -2.10166812e-01,  9.99043882e-03,\n",
            "        1.60478950e-01, -1.59267217e-01,  4.89141375e-01, -4.40967083e-03,\n",
            "        2.09746301e-01, -4.77699637e-02,  4.12931710e-01, -1.60866231e-01,\n",
            "        4.78706002e-01,  2.38923460e-01, -2.10127681e-01,  1.37225240e-01,\n",
            "        1.02220505e-01, -9.13258567e-02,  2.52216667e-01, -5.56917787e-01,\n",
            "        3.39246571e-01,  9.82420892e-02,  8.70075524e-01,  2.52973080e-01],\n",
            "      dtype=float32), 'Sign_symptom': array([-1.85592636e-01, -4.50773865e-01, -3.89081150e-01, -3.17601174e-01,\n",
            "       -3.43265116e-01, -1.31990105e-01, -6.89878345e-01,  4.04904276e-01,\n",
            "        5.57781023e-04, -4.51051584e-03,  1.80752918e-01, -3.31873506e-01,\n",
            "        3.69114690e-02, -2.35198423e-01, -4.88082409e-01, -1.14717777e-03,\n",
            "       -6.88299313e-02,  1.45396680e-01,  2.49760941e-01,  2.80633271e-01,\n",
            "        2.09907696e-01,  2.35533327e-01, -7.07316026e-02,  5.02441898e-02,\n",
            "       -1.47870749e-01,  1.31514981e-01, -3.96372765e-01, -1.85034648e-01,\n",
            "        5.08636475e-01, -2.09491760e-01,  1.25811547e-01,  5.12708664e-01,\n",
            "       -4.23039526e-01,  5.22368789e-01, -2.16323901e-02, -4.45018351e-01,\n",
            "       -8.80307108e-02, -8.18327218e-02,  9.06842828e-01,  1.67608887e-01,\n",
            "        2.91875184e-01,  1.37201101e-01,  1.57331198e-01,  2.52616376e-01,\n",
            "        4.45787720e-02,  4.87936407e-01,  1.45837396e-01, -4.87422526e-01,\n",
            "        3.68936062e-01, -3.26685488e-01, -1.77576572e-01, -1.61167551e-02,\n",
            "       -1.37760952e-01, -1.68670684e-01, -2.89497495e-01, -5.47176879e-03,\n",
            "       -6.02535680e-02,  1.90695375e-01, -3.22090596e-01,  1.66888997e-01,\n",
            "        1.40553713e-01,  6.42567992e-01, -2.11684495e-01, -1.58013016e-01,\n",
            "       -6.76241666e-02,  2.80881766e-02, -4.35832888e-01, -1.45260304e-01,\n",
            "       -4.40694809e-01,  5.49837807e-03, -2.39135742e-01, -8.91046226e-02,\n",
            "        2.68706620e-01, -1.94800377e-01,  1.21487655e-01,  5.78169450e-02,\n",
            "        1.68770760e-01, -2.09055707e-01, -8.56078267e-02, -3.50745618e-01,\n",
            "        2.26870865e-01,  1.32085294e-01,  1.29721195e-01, -6.14137165e-02,\n",
            "       -3.11095774e-01,  4.07691300e-02,  4.33479369e-01,  2.03683779e-01,\n",
            "        2.27653533e-01,  2.16245264e-01,  3.98589075e-01,  2.93689400e-01,\n",
            "       -2.40390807e-01, -2.76079744e-01,  3.20985883e-01,  4.48437817e-02,\n",
            "        3.12054399e-02, -9.74036679e-02, -3.26320087e-03, -2.22425729e-01,\n",
            "        1.76127311e-02, -1.03310533e-02,  3.70967180e-01,  1.23216156e-02,\n",
            "       -3.83464307e-01, -1.89651385e-01,  2.96292007e-01,  1.60003781e-01,\n",
            "       -3.82310718e-01, -2.13805482e-01, -4.31394093e-02, -2.08711579e-01,\n",
            "        2.05839917e-01, -5.22885323e-01,  7.37283602e-02,  1.21191300e-01,\n",
            "        2.92445064e-01, -1.57111242e-01, -2.65178144e-01,  7.44046718e-02,\n",
            "        4.12508100e-01, -3.15106004e-01, -1.22497842e-01, -3.18984687e-02,\n",
            "        3.49278152e-01, -2.17795327e-01,  3.38903069e-01, -2.24529892e-01,\n",
            "        3.54372561e-01,  4.34237480e-01, -1.12033270e-01,  3.85092646e-01,\n",
            "       -4.97632027e-01, -2.89707989e-01,  1.78347543e-01,  2.84170057e-03,\n",
            "       -1.59760743e-01, -1.67928725e-01,  6.25537634e-02,  1.80156750e-03,\n",
            "       -2.43067428e-01,  1.06175639e-01, -2.85796702e-01,  2.29299076e-05,\n",
            "       -7.09975585e-02, -2.34098762e-01,  5.73895685e-02, -2.12452263e-01,\n",
            "       -3.88693601e-01, -3.90803665e-01, -8.42519760e-01, -4.44793522e-01,\n",
            "        2.51909554e-01,  1.92626402e-01, -7.12698698e-03, -3.75738800e-01,\n",
            "        3.30893546e-01, -9.19760093e-02,  2.01624542e-01,  4.23786156e-02,\n",
            "        3.91407579e-01, -1.07394621e-01,  1.51564211e-01, -3.25543314e-01,\n",
            "       -1.90940663e-01,  3.79386544e-01, -1.29555734e-02, -1.18509136e-01,\n",
            "       -1.61002934e-01,  3.70343506e-01,  5.49200773e-02, -1.73757911e-01,\n",
            "        8.39360431e-02, -1.79298952e-01, -2.53170934e-02,  6.71334118e-02,\n",
            "       -3.17332824e-03,  1.08710453e-02, -1.98298603e-01, -6.24430887e-02,\n",
            "       -4.51804809e-02, -2.05693817e+00,  3.07244390e-01, -1.23772308e-01,\n",
            "       -1.49277419e-01, -4.76741314e-01, -2.55172968e-01, -1.05609216e-01,\n",
            "        2.92563349e-01,  1.28150195e-01, -4.04877663e-01, -1.29376143e-01,\n",
            "        6.10948214e-03,  6.81405216e-02, -2.77359068e-01,  3.65020931e-01,\n",
            "        6.21768355e-01,  2.94387400e-01, -1.80821538e-01, -1.52484607e-02,\n",
            "       -4.92038071e-01,  4.08428818e-01,  2.59371698e-01, -6.03857823e-02,\n",
            "        5.01938350e-02,  5.20219684e-01, -1.82357114e-02,  1.17603503e-01,\n",
            "        2.23734424e-01,  6.82105571e-02,  1.26949877e-01, -1.82930991e-01,\n",
            "       -2.61471868e-01, -1.21110342e-01,  1.95540875e-01,  3.22826326e-01,\n",
            "       -3.12545151e-01, -2.13819146e-01,  3.73871624e-01,  1.32141978e-01,\n",
            "       -1.18923187e-01, -9.79299843e-03, -2.50108354e-02, -4.19177115e-01,\n",
            "        5.17054573e-02,  5.14782012e-01, -4.00349200e-01,  2.28297323e-01,\n",
            "       -4.27145332e-01, -2.29946941e-01, -3.95489018e-03, -1.94291279e-01,\n",
            "        1.20232716e-01,  2.77394116e-01, -2.26985455e-01, -3.53804260e-01,\n",
            "       -3.58933151e-01,  5.02725482e-01, -6.52259588e-02, -1.77782863e-01,\n",
            "        3.99993099e-02, -2.81707436e-01, -4.17149484e-01,  1.45730764e-01,\n",
            "        2.90686190e-01,  2.88780004e-01, -1.24360351e-02,  6.11695468e-01,\n",
            "        5.38868189e-01,  3.79266083e-01,  2.21641615e-01,  2.48408437e-01,\n",
            "        4.31848377e-01,  6.11064792e-01, -1.59094244e-01, -2.64921606e-01,\n",
            "        3.37457120e-01,  2.72565335e-03,  3.48383665e-01, -2.27861255e-01,\n",
            "       -1.05429627e-01, -1.59870952e-01,  1.27812013e-01,  4.84340906e-01,\n",
            "       -1.57632023e-01,  5.31412587e-02,  3.34237337e-01, -1.72727108e-01,\n",
            "       -2.29078293e-01, -2.81887911e-02, -5.42700469e-01,  4.72546101e-01,\n",
            "       -2.24693820e-01, -2.11232975e-01, -5.81923306e-01, -5.47755897e-01,\n",
            "       -3.51347983e-01, -8.86026490e-03,  3.77689749e-01, -6.73785090e-01,\n",
            "       -1.91060305e-01, -1.94716662e-01,  9.72279161e-02, -1.83378190e-01,\n",
            "       -8.10534731e-02,  1.28927678e-01,  2.70914137e-01, -1.82072315e-02,\n",
            "       -1.52288288e-01, -1.60235673e-01, -1.14464983e-01,  3.97449210e-02,\n",
            "        7.27864588e-03,  6.55980855e-02, -6.19538315e-02, -1.64528012e-01,\n",
            "       -1.09196842e-01,  9.03838500e-02,  1.81288242e-01, -2.08865404e-01,\n",
            "       -1.72772527e-01, -1.66233271e-01,  9.12941247e-02, -1.89949065e-01,\n",
            "       -5.40310852e-02,  2.33542278e-01, -2.19008122e-02,  3.33734334e-01,\n",
            "       -3.68953288e-01, -4.33023535e-02, -6.00733161e-02,  5.26179612e-01,\n",
            "       -1.33907005e-01, -8.08427110e-02, -1.17928237e-02, -1.08303964e+00,\n",
            "       -8.74074828e-03, -8.48468915e-02, -1.69483393e-01, -6.66577816e-02,\n",
            "        2.01856136e-01, -7.64594832e-03, -3.88565324e-02,  1.84927940e-01,\n",
            "       -4.43864577e-02,  5.01416564e-01,  7.07008004e-01,  3.49092066e-01,\n",
            "       -1.87680691e-01, -3.76779258e-01, -2.99474180e-01,  1.14747487e-01,\n",
            "       -3.98824245e-01, -1.56972528e-01,  4.87125695e-01,  3.91765505e-01,\n",
            "        1.37301579e-01, -2.13293642e-01,  8.11432377e-02, -3.65165055e-01,\n",
            "        6.53558195e-01, -3.23998749e-01, -5.98611012e-02,  3.79935652e-01,\n",
            "       -5.92706725e-02,  4.10381034e-02,  4.37959522e-01, -2.35584497e-01,\n",
            "       -1.86193705e-01,  1.61827207e-01,  1.81022197e-01,  6.50758296e-02,\n",
            "       -2.50168413e-01, -2.31619209e-01,  4.15573895e-01,  5.75001717e-01,\n",
            "        3.98500621e-01, -2.17353076e-01,  3.15798700e-01,  5.30234635e-01,\n",
            "       -3.70511353e-01,  4.39581811e-01, -2.98417926e-01, -3.82727861e-01,\n",
            "        4.18991633e-02,  3.70208681e-01,  1.17016003e-01, -2.74426997e-01,\n",
            "       -2.55571716e-02, -1.39051095e-01,  3.23871285e-01, -2.88130581e-01,\n",
            "        1.30176814e-02,  1.10431053e-01,  5.04030943e-01, -9.21387076e-02,\n",
            "       -4.94833797e-01,  5.30462861e-02, -8.22642326e-01, -4.22406971e-01,\n",
            "       -2.15231217e-02, -3.01205128e-01,  8.96846727e-02, -4.06768978e-01,\n",
            "       -2.20776632e-01, -2.49730870e-01,  7.12484270e-02,  2.90019482e-01,\n",
            "        3.16607207e-01,  3.90965454e-02,  1.62071854e-01, -5.72276413e-02,\n",
            "       -1.91661820e-01,  6.10885680e-01,  8.98829773e-02, -2.89115608e-01,\n",
            "        1.80214807e-01, -4.75136712e-02,  1.89283729e-01,  5.56403279e-01,\n",
            "       -4.32418108e-01, -2.29352906e-01, -3.01578879e-01, -2.21951649e-01,\n",
            "       -3.82579565e-02, -5.68908080e-02,  1.34470254e-01,  1.73340350e-01,\n",
            "       -3.19173217e-01, -2.64084756e-01, -2.91540384e-01,  4.25738811e-01,\n",
            "       -6.25484362e-02,  2.98714727e-01, -8.20100978e-02,  2.68984735e-01,\n",
            "       -2.17450306e-01,  1.33421704e-01,  1.29305139e-01, -2.73606956e-01,\n",
            "        5.90749562e-01,  1.11975752e-01, -8.20397586e-02,  3.00305009e-01,\n",
            "       -1.16231537e+01,  2.05375239e-01,  7.26109684e-01, -1.57123089e-01,\n",
            "       -2.26656925e-02, -2.38813207e-01,  3.02125990e-01, -2.73439080e-01,\n",
            "        1.94679633e-01, -4.21843708e-01, -3.44100967e-02,  2.57849451e-02,\n",
            "       -1.30752474e-01,  1.01098046e-02, -6.47681773e-01,  1.49056822e-01,\n",
            "        2.41422653e-02, -6.36218786e-01,  1.11622334e-01, -1.46989912e-01,\n",
            "        1.02613725e-01, -6.51341677e-02, -1.24197446e-01,  2.08968326e-01,\n",
            "       -2.24032685e-01, -1.91114649e-01,  1.18104694e-02,  4.54579443e-02,\n",
            "        2.02443555e-01,  3.20126832e-01, -6.39277175e-02, -1.96078628e-01,\n",
            "        9.32748057e-03,  9.06989723e-02, -2.55464852e-01, -2.19744921e-01,\n",
            "       -5.95789179e-02, -2.60901213e-01,  1.37730405e-01, -1.39699295e-01,\n",
            "       -2.32177109e-01, -1.24656335e-01, -3.77222449e-01,  1.35677814e-01,\n",
            "        4.07765388e-01,  4.27699178e-01,  9.35403943e-01,  5.11828601e-01,\n",
            "        8.51125419e-02,  1.90671414e-01,  5.39745018e-02, -7.24447072e-01,\n",
            "        2.93233812e-01, -2.97555149e-01,  1.96804017e-01, -2.86733210e-01,\n",
            "        3.51514190e-01, -3.54076087e-01,  1.94853619e-01, -2.32229665e-01,\n",
            "       -4.32378314e-02,  1.89745188e-01,  3.05889070e-01,  1.14993766e-01,\n",
            "       -1.45774052e-01, -5.83306663e-02,  2.55942672e-01, -4.98558357e-02,\n",
            "       -3.82496789e-02, -2.21440226e-01,  5.22524491e-02, -4.70442846e-02,\n",
            "        2.97713757e-01,  7.23616257e-02, -7.36560822e-02, -3.49260564e-03,\n",
            "       -2.30305389e-01,  7.20980912e-02,  9.18580055e-01, -1.52924210e-01,\n",
            "        1.45375192e-01, -1.12250946e-01, -9.34370905e-02,  1.09971918e-01,\n",
            "       -4.62624639e-01, -9.79755223e-02,  8.80576074e-02, -1.89000815e-01,\n",
            "        5.58158994e-01,  3.11609596e-01, -3.33155006e-01, -7.85075873e-02,\n",
            "       -2.45476663e-02,  1.70065105e-01,  2.87501514e-01,  7.40086496e-01,\n",
            "       -4.77181137e-01,  1.43451560e-02,  6.66290075e-02,  1.00292102e-01,\n",
            "       -2.15130806e-01, -8.64491910e-02, -3.39575112e-02, -7.80980736e-02,\n",
            "       -6.41832575e-02, -1.32855177e-01,  1.22358397e-01, -1.83265194e-01,\n",
            "        1.77415326e-01,  1.75678328e-01, -5.14537156e-01,  9.88321975e-02,\n",
            "       -2.31739715e-01,  1.63161635e-01, -2.44799539e-01, -3.06938112e-01,\n",
            "        3.93079907e-01,  3.03261101e-01, -5.85286021e-01, -1.72169209e-01,\n",
            "       -2.49210615e-02, -2.25901037e-01,  3.62858325e-01,  1.64573580e-01,\n",
            "       -3.44552428e-01, -1.80260167e-02, -5.73121905e-01,  9.79496762e-02,\n",
            "        3.30062956e-01,  5.94325773e-02,  3.55278283e-01,  4.02893394e-01,\n",
            "       -2.32515499e-01,  2.36846775e-01, -4.51137483e-01, -5.29182076e-01,\n",
            "       -9.68833745e-01,  6.37458339e-02,  2.95260102e-01, -2.66449869e-01,\n",
            "        3.85051407e-02,  4.68572900e-02, -1.48735911e-01, -2.16839030e-01,\n",
            "        3.53971660e-01, -1.17659703e-01,  2.69344211e-01,  1.87070787e-01,\n",
            "       -1.76999345e-01,  2.40590692e-01, -1.88818231e-01, -4.81331736e-01,\n",
            "       -2.97876924e-01, -1.70115143e-01, -6.39487058e-02, -7.18533695e-01,\n",
            "       -2.56302297e-01,  9.24717709e-02,  2.49717552e-02, -2.51850456e-01,\n",
            "        1.81534320e-01, -2.16593057e-01, -2.91433483e-01, -2.06505150e-01,\n",
            "        4.46848646e-02,  2.15866044e-01,  6.71687722e-02, -4.07463461e-01,\n",
            "        4.77522425e-02,  1.13788113e-01,  3.30656357e-02,  3.23450893e-01,\n",
            "        2.89927483e-01, -3.77076536e-01, -3.73465806e-01, -2.74962839e-02,\n",
            "       -4.11842775e-04,  8.21932435e-01, -5.13609111e-01, -1.83940858e-01,\n",
            "        3.02744538e-01,  2.05620408e-01, -1.25702173e-01, -1.69096023e-01,\n",
            "       -5.08994870e-02,  3.09906036e-01, -5.00388816e-02, -2.67515749e-01,\n",
            "        1.07974768e-01, -3.50055110e-04,  1.00420393e-01,  2.07142830e-01,\n",
            "       -4.46801558e-02, -2.93448895e-01,  6.50774598e-01, -5.46151176e-02,\n",
            "       -1.60841912e-01,  7.16442838e-02,  6.86610788e-02,  1.07857421e-01,\n",
            "        1.59843400e-01, -2.42879316e-01,  7.90065676e-02, -1.50284797e-01,\n",
            "        7.30009228e-02, -1.43924236e-01,  5.27659655e-01,  1.54139447e+00,\n",
            "        6.02225736e-02, -4.76563498e-02,  7.17432022e-01,  3.61285180e-01,\n",
            "        9.61531997e-02,  1.35467619e-01, -2.00792789e-01,  1.77574560e-01,\n",
            "        2.57682621e-01, -1.70673013e-01, -3.00843865e-01,  1.48964792e-01,\n",
            "        3.62512022e-01,  2.96936333e-01, -5.21505535e-01, -4.61939186e-01,\n",
            "       -1.86872467e-01,  2.52632886e-01, -2.78078467e-01,  4.99409765e-01,\n",
            "       -3.51137295e-02,  1.50835827e-01,  4.93964627e-02, -6.20089412e-01,\n",
            "       -3.90710905e-02, -9.90437567e-02,  3.22640598e-01, -8.09961483e-02,\n",
            "       -4.71775681e-01, -3.66437912e-01, -5.90512827e-02,  5.89101315e-02,\n",
            "        2.97372639e-01,  4.02782619e-01, -5.55410445e-01,  3.04224074e-01,\n",
            "        2.76397198e-01, -2.63620615e-01,  4.63744812e-02, -1.78026065e-01,\n",
            "        2.39705995e-01, -1.02625296e-01, -1.40210241e-01, -1.55470952e-01,\n",
            "        1.66713715e-01, -5.13090730e-01, -4.76794809e-01,  3.53822559e-01,\n",
            "        4.45371777e-01, -2.82555874e-02,  6.01390973e-02, -1.40347034e-01,\n",
            "       -2.52964288e-01, -2.83025861e-01, -3.40003639e-01, -1.90631479e-01,\n",
            "        3.46511364e-01, -2.69539893e-01, -1.76703498e-01, -4.42460060e-01,\n",
            "       -3.78678553e-02,  3.53030562e-01,  3.55935574e-01,  1.12456560e-01,\n",
            "        5.62008739e-01,  6.26000345e-01, -1.35511637e-01, -2.08103612e-01,\n",
            "        1.37550279e-01,  4.03788507e-01, -1.15638539e-01,  5.82844138e-01,\n",
            "        4.39586490e-02, -3.02327514e-01,  1.99640512e-01, -5.45586571e-02,\n",
            "       -1.02041587e-01,  3.06375593e-01, -6.16609640e-02, -8.89282767e-03,\n",
            "       -2.60980189e-01, -9.11057144e-02,  2.80788839e-01,  3.34845364e-01,\n",
            "        2.27198452e-01, -4.87169594e-01, -1.68228120e-01,  2.56353438e-01,\n",
            "        1.10710319e-02,  3.13099235e-01, -9.09896046e-02,  5.68421245e-01,\n",
            "       -4.39848244e-01,  1.82296723e-01, -6.84226990e-01, -9.57551822e-02,\n",
            "       -1.18139550e-01, -2.73567349e-01, -5.72101593e-01, -3.13956320e-01,\n",
            "        3.80820185e-01, -6.89397752e-03,  4.87170368e-01,  6.80620849e-01,\n",
            "       -2.90270895e-01, -5.54547608e-01, -1.99519917e-01,  1.00486591e-01,\n",
            "        4.05445546e-02, -2.57192463e-01,  2.24870831e-01, -1.11873029e-02,\n",
            "       -6.62571788e-02, -7.11029649e-01, -2.79282868e-01, -2.90187951e-02,\n",
            "        6.81795478e-02, -3.40443075e-01,  6.04789972e-01,  1.00911140e-01,\n",
            "        1.60032183e-01,  2.17778251e-01,  2.79304653e-01,  8.12835642e-04,\n",
            "        2.82894015e-01,  1.22600123e-01, -2.71902412e-01, -4.32144776e-02,\n",
            "        2.90701002e-01, -3.08597714e-01, -1.20595120e-01, -6.51078880e-01,\n",
            "        4.26514149e-01,  2.96251684e-01,  8.34475696e-01,  2.64429837e-01],\n",
            "      dtype=float32), 'Date': array([-4.10826683e-01, -4.27570432e-01, -6.90165579e-01, -3.44063759e-01,\n",
            "       -9.57496688e-02, -3.12941045e-01, -4.42201495e-01,  5.14671445e-01,\n",
            "        4.69632261e-02,  3.11383247e-01,  4.56298627e-02,  2.39567474e-01,\n",
            "        1.56290814e-01, -3.05969119e-01, -6.08482063e-01, -2.34648734e-02,\n",
            "       -1.81723475e-01,  2.18037188e-01,  1.24836219e-02,  2.91157123e-02,\n",
            "        4.28133011e-01,  3.48170519e-01, -8.43597054e-02, -7.92862996e-02,\n",
            "       -8.59762356e-02,  6.63133562e-02, -6.53545081e-04, -1.08348459e-01,\n",
            "        4.49601412e-01, -2.56155103e-01,  1.68636307e-01,  3.24341327e-01,\n",
            "       -2.00243354e-01,  5.70784867e-01, -1.28484992e-02, -2.38416135e-01,\n",
            "       -1.56351134e-01, -1.38392448e-01,  7.29088306e-01,  3.08351666e-01,\n",
            "        2.12788105e-01,  1.98877230e-01,  3.66856962e-01,  9.37012732e-02,\n",
            "        1.45396829e-01,  3.49298686e-01,  1.78502560e-01, -3.86529356e-01,\n",
            "        1.69197485e-01, -3.92940313e-01, -4.31169420e-01, -1.90059841e-01,\n",
            "        2.66836047e-01, -1.26068795e-03, -4.70148772e-01, -1.56510800e-01,\n",
            "        2.23571584e-02,  5.45281172e-02, -2.33110383e-01,  7.51449987e-02,\n",
            "       -2.39476144e-01,  4.29599762e-01, -1.68321609e-01, -2.58004338e-01,\n",
            "        2.17899099e-01, -1.48812175e-01, -5.24676263e-01, -1.34063393e-01,\n",
            "       -4.00868446e-01,  3.33530456e-03, -2.47540902e-02,  1.29830077e-01,\n",
            "        1.02761395e-01, -1.16589963e-02,  3.31911981e-01, -1.34460106e-01,\n",
            "        1.73994258e-01,  9.73387435e-03, -1.57933667e-01, -3.88367027e-02,\n",
            "        2.60708600e-01,  3.52635115e-01, -1.55355603e-01, -1.11798733e-01,\n",
            "       -1.52588427e-01, -5.54018132e-02,  1.73743889e-01,  1.57086074e-01,\n",
            "        4.55560386e-02, -1.50929093e-02,  1.74401045e-01,  2.45188788e-01,\n",
            "       -2.92111278e-01, -1.48989573e-01,  1.31180167e-01, -2.98516154e-01,\n",
            "        2.01153114e-01, -1.74113825e-01,  2.15023220e-01, -3.27523500e-01,\n",
            "       -1.81770921e-01, -8.04893300e-02,  1.02258168e-01, -1.07217371e-01,\n",
            "       -2.30581835e-01, -4.06878114e-01,  2.99152195e-01,  5.08560687e-02,\n",
            "       -2.05763161e-01, -7.68007040e-02, -1.32186040e-01, -3.98946762e-01,\n",
            "        8.56741145e-02, -3.63641113e-01, -1.85496554e-01,  2.81612068e-01,\n",
            "        3.31946194e-01,  8.15350115e-02, -2.67156601e-01,  7.13516846e-02,\n",
            "        2.98239410e-01, -3.32673043e-01,  1.68454193e-03,  1.25660822e-01,\n",
            "        2.89834887e-01, -9.60902199e-02,  4.46247786e-01, -4.10094708e-02,\n",
            "        1.93279549e-01,  5.74302852e-01, -9.31567773e-02, -3.71272974e-02,\n",
            "       -1.88610241e-01, -1.29139036e-01, -4.30688858e-02,  1.57751903e-01,\n",
            "       -7.83444643e-02, -3.06051373e-01,  9.16874632e-02,  2.01320469e-01,\n",
            "       -5.36238961e-02, -8.84057358e-02, -5.80089092e-01, -1.84404850e-01,\n",
            "       -2.46850657e-03, -2.48024940e-01, -3.79030257e-02, -3.04695070e-01,\n",
            "       -4.98575062e-01, -3.28063846e-01, -7.54277408e-01, -3.98509622e-01,\n",
            "        4.77721572e-01,  7.61122257e-02,  1.15020066e-01, -3.30966383e-01,\n",
            "        2.02614143e-01, -2.43720841e-02,  1.56855747e-01,  1.30855665e-01,\n",
            "        6.84003592e-01, -1.70177504e-01,  3.51561427e-01, -1.05996825e-01,\n",
            "       -2.60829125e-02,  3.24473865e-02, -4.54086393e-01, -2.54414529e-01,\n",
            "        9.71840248e-02,  2.81388909e-02,  5.54916374e-02, -2.48145282e-01,\n",
            "        2.70988792e-01, -3.45976800e-01,  1.08438283e-01, -3.78866553e-01,\n",
            "        1.30167067e-01, -1.03752404e-01, -9.33524370e-02,  1.08899139e-01,\n",
            "        2.01360300e-01, -2.17704391e+00,  8.30196440e-02, -2.46586695e-01,\n",
            "        2.18278408e-01, -3.09882671e-01, -3.13014239e-02, -1.16257630e-02,\n",
            "        3.53219122e-01, -4.37010266e-02, -1.68388411e-01,  1.53606355e-01,\n",
            "       -3.19283903e-01,  3.95984501e-02, -1.17492527e-01,  2.64186949e-01,\n",
            "        3.34194392e-01,  3.41061383e-01,  3.31513047e-01,  5.41017540e-02,\n",
            "       -2.32237116e-01,  3.35907727e-01,  8.26428551e-03,  9.71112326e-02,\n",
            "        8.58119726e-02,  3.27484876e-01, -3.95648293e-02,  1.96881756e-01,\n",
            "       -9.13287178e-02,  6.46181181e-02, -1.52340099e-01, -8.53199661e-02,\n",
            "       -2.67120451e-01,  9.30609107e-02, -1.21357322e-01,  1.75072432e-01,\n",
            "       -2.92256027e-01, -2.33436689e-01,  4.20413464e-01,  2.74808466e-01,\n",
            "       -2.55115423e-02, -3.27906422e-02, -1.05271854e-01, -2.73835778e-01,\n",
            "        2.24483252e-01,  4.96499389e-01, -3.13523114e-01,  2.96921074e-01,\n",
            "        4.32576723e-02, -2.48127818e-01,  1.32504001e-01, -2.35642835e-01,\n",
            "        5.06545484e-01,  8.16764757e-02,  4.35871594e-02, -1.41754195e-01,\n",
            "       -3.83513778e-01,  2.67123878e-01, -2.84700036e-01, -7.66525269e-02,\n",
            "       -3.22161704e-01, -3.06862921e-01, -2.15112150e-01,  3.72732170e-02,\n",
            "        3.30387801e-01,  2.28555918e-01, -8.70184135e-03,  4.74649757e-01,\n",
            "        3.67244363e-01,  5.28296471e-01,  6.52414486e-02,  7.50981718e-02,\n",
            "        2.97120661e-01,  3.14663380e-01, -3.11141998e-01,  1.07986443e-01,\n",
            "        1.82515740e-01, -1.14772685e-01,  1.85227856e-01,  1.42740175e-01,\n",
            "       -2.34168589e-01, -1.85489848e-01,  8.26872960e-02,  4.62958187e-01,\n",
            "       -8.67285207e-02, -6.76613972e-02,  2.82099634e-01, -1.02493189e-01,\n",
            "       -1.51488140e-01,  5.59556037e-02, -3.38407546e-01,  4.29476708e-01,\n",
            "       -2.22597405e-01, -2.34920919e-01, -6.07973397e-01, -6.76221788e-01,\n",
            "       -2.55479425e-01, -4.04055901e-02,  1.04334682e-01, -6.56377375e-01,\n",
            "       -2.99747497e-01, -6.87556326e-01, -2.89505739e-02, -9.88172516e-02,\n",
            "       -1.10777877e-01,  1.04606740e-01, -6.53660893e-02,  5.66311479e-02,\n",
            "       -4.49000806e-01, -3.22942883e-01, -3.14254820e-01,  7.82424137e-02,\n",
            "       -3.93637091e-01, -3.73295918e-02,  1.57691747e-01,  6.30520210e-02,\n",
            "        1.93454131e-01,  1.47199169e-01, -1.62696838e-01, -2.77170509e-01,\n",
            "       -3.14086825e-02,  1.28651723e-01,  2.44439170e-02,  3.81065160e-02,\n",
            "        7.29053169e-02,  1.27172455e-01,  7.54978508e-02,  4.68097776e-01,\n",
            "       -2.08752856e-01, -3.30361128e-01, -2.33429536e-01,  4.22952414e-01,\n",
            "       -3.04668918e-02, -7.90654197e-02,  4.80075665e-02, -7.64484107e-01,\n",
            "       -4.18694913e-02, -6.35451078e-02, -2.58666962e-01, -1.76215813e-01,\n",
            "        1.87775016e-01,  3.23367447e-01, -1.80595055e-01, -6.91386545e-03,\n",
            "       -1.22412466e-01,  4.42343146e-01,  5.50859153e-01,  6.08750731e-02,\n",
            "        5.76835535e-02, -5.34581661e-01,  4.81740832e-02, -1.18316032e-01,\n",
            "       -7.20692053e-02, -3.36145550e-01,  4.82674360e-01,  2.50871807e-01,\n",
            "        1.27804950e-01, -8.40474740e-02,  1.04841150e-01, -4.27187942e-02,\n",
            "        2.69553781e-01, -7.95620903e-02, -1.45854741e-01,  3.72206211e-01,\n",
            "        2.19975542e-02,  2.73972005e-01,  3.80569309e-01, -2.05895856e-01,\n",
            "       -7.99876973e-02,  1.42639413e-01,  1.17450655e-01, -2.78934687e-01,\n",
            "       -1.23979330e-01, -3.08640927e-01,  4.10916537e-01,  4.74515915e-01,\n",
            "        2.39618674e-01, -4.85328957e-03,  3.16114336e-01,  2.18311295e-01,\n",
            "       -1.43971503e-01,  2.57007271e-01, -3.60668212e-01, -9.03003812e-02,\n",
            "        1.20574266e-01,  4.25241560e-01,  6.15906179e-01, -3.19367558e-01,\n",
            "       -7.05799758e-02,  5.51441044e-04, -2.21088767e-01, -9.15801898e-02,\n",
            "       -1.15008272e-01,  1.26858160e-01,  4.09575701e-01, -4.35905784e-01,\n",
            "       -2.71092206e-01,  3.60729806e-02, -3.69114488e-01, -3.65447849e-01,\n",
            "        1.40797809e-01, -3.02295312e-02, -4.88031022e-02, -3.35292727e-01,\n",
            "       -2.26869836e-01, -2.20401391e-01, -4.25872318e-02,  4.35459524e-01,\n",
            "        2.65961558e-01,  6.76310956e-02, -9.95642915e-02, -1.41418979e-01,\n",
            "        3.81567441e-02,  6.39591753e-01,  4.51027192e-02, -3.13207924e-01,\n",
            "        2.81997889e-01, -1.07920110e-01,  1.26886860e-01,  2.96797574e-01,\n",
            "       -2.93685198e-01, -4.40855265e-01, -1.68177322e-01,  2.57551402e-01,\n",
            "        2.44206086e-01, -3.14242989e-01, -6.80931807e-02,  1.70793101e-01,\n",
            "       -8.50430205e-02, -6.70533106e-02, -2.58245528e-01,  1.22993983e-01,\n",
            "       -2.47158930e-01, -6.83510378e-02,  1.45071968e-01,  4.50581789e-01,\n",
            "       -2.82231599e-01, -3.05823032e-02,  2.38095179e-01, -2.10806862e-01,\n",
            "        4.65226024e-01,  1.52116731e-01,  6.55332133e-02,  3.75539184e-01,\n",
            "       -1.15579481e+01,  4.81360912e-01,  4.49458569e-01, -2.69743651e-01,\n",
            "        1.06475361e-01, -1.97616871e-02,  2.11732268e-01, -2.49516681e-01,\n",
            "        1.12978548e-01, -3.10671180e-01, -1.13853984e-01, -4.09557633e-02,\n",
            "       -2.48385802e-01, -1.85470164e-01, -4.59079027e-01,  3.61964107e-03,\n",
            "       -2.41959020e-02, -5.19157052e-01,  1.80780828e-01, -2.73135360e-02,\n",
            "        2.60823011e-01, -4.24128957e-02, -2.74276346e-01,  4.41752404e-01,\n",
            "       -2.46130750e-01, -2.50536293e-01,  2.01178137e-02,  8.45288113e-02,\n",
            "        1.61777541e-01,  1.25927150e-01, -7.40356073e-02,  1.00363612e-01,\n",
            "       -8.03368911e-02,  6.67430460e-02, -1.49543568e-01, -4.74577159e-01,\n",
            "        6.29320219e-02, -3.02571267e-01,  6.43886114e-03, -9.64049920e-02,\n",
            "       -3.17900956e-01, -2.33964976e-02, -3.17595005e-01,  1.85133874e-01,\n",
            "        4.54174370e-01,  4.18234259e-01,  7.35434294e-01,  8.66102159e-01,\n",
            "       -1.76231548e-01, -6.56840578e-02, -2.54294425e-01, -5.34977019e-01,\n",
            "        1.93734601e-01, -3.91218752e-01, -2.59936392e-01, -1.81251049e-01,\n",
            "        2.94347942e-01, -3.36606354e-01,  3.53470653e-01, -2.15459570e-01,\n",
            "       -3.16602774e-02,  2.39923730e-01,  1.10355251e-01, -8.09813365e-02,\n",
            "       -4.82509941e-01,  1.28217995e-01,  2.87891030e-01,  1.83523253e-01,\n",
            "       -1.76614169e-02, -5.59503496e-01,  1.22115575e-01,  5.31312376e-02,\n",
            "        4.31186587e-01,  2.94969618e-01,  1.26839802e-01,  2.38990739e-01,\n",
            "       -1.06306978e-01, -8.29087272e-02,  1.20473826e+00, -8.93180370e-02,\n",
            "        2.05545679e-01, -3.45975272e-02,  1.00069702e-01, -3.06330085e-01,\n",
            "       -1.76575959e-01, -3.18488657e-01,  4.67643924e-02, -4.01600748e-01,\n",
            "        1.07988827e-01,  3.31871569e-01, -1.29982665e-01, -2.44476631e-01,\n",
            "       -1.03152335e-01, -8.90805349e-02,  1.28157601e-01,  7.00882137e-01,\n",
            "       -6.43258691e-01,  1.56840682e-03, -1.14165805e-01,  3.96925241e-01,\n",
            "        1.93023626e-02,  6.14022501e-02, -8.87846276e-02, -5.80439985e-01,\n",
            "        4.89159226e-02, -1.46160619e-02, -4.67731208e-02,  1.67900488e-01,\n",
            "        2.03388669e-02,  2.74846584e-01, -1.16170704e-01, -5.38335852e-02,\n",
            "       -1.21310502e-02,  3.28740627e-01, -3.96524042e-01, -3.79150629e-01,\n",
            "        9.84652713e-02,  2.55675316e-01, -4.97489184e-01, -3.25977057e-01,\n",
            "       -1.71103999e-01,  3.32985409e-02,  2.55889803e-01,  1.22080117e-01,\n",
            "       -3.57434392e-01,  2.46748179e-02, -6.44892275e-01, -1.76604226e-01,\n",
            "       -8.78488347e-02,  1.57265842e-01,  4.42663521e-01,  4.73968625e-01,\n",
            "       -1.90677643e-01,  2.04469040e-01, -2.82398939e-01, -3.28710347e-01,\n",
            "       -9.10755217e-01,  2.04981323e-02,  4.13799047e-01, -1.52323127e-01,\n",
            "        1.24449812e-01, -1.78824231e-01, -2.30928306e-02,  1.52178453e-02,\n",
            "        3.17347050e-01,  1.97198272e-01,  4.80578691e-01,  1.98228583e-01,\n",
            "       -3.99598002e-01,  3.37216742e-02, -8.34081769e-02, -4.17580813e-01,\n",
            "        1.07911550e-01, -1.11487500e-01, -6.41863421e-02, -2.90589541e-01,\n",
            "       -2.12522864e-01, -5.77584058e-02,  5.89038283e-02, -2.69004256e-01,\n",
            "        1.02332555e-01,  1.15111686e-01, -1.76297426e-01, -1.34995475e-01,\n",
            "        2.96204716e-01,  3.28027815e-01,  9.24907625e-04, -3.61228824e-01,\n",
            "       -5.99651873e-01, -1.73658356e-01,  2.19301298e-01,  4.35359478e-01,\n",
            "        1.45011336e-01, -2.64688492e-01, -3.70375007e-01, -1.09331878e-02,\n",
            "       -1.16865069e-01,  7.09992349e-01, -4.58808303e-01, -3.58284712e-01,\n",
            "        1.13178633e-01,  6.47923872e-02, -1.26529440e-01, -2.63036519e-01,\n",
            "        7.15506598e-02,  3.02551270e-01, -1.15376495e-01, -1.80048704e-01,\n",
            "        3.23687166e-01,  1.50487483e-01,  2.22412050e-01,  5.29131830e-01,\n",
            "        3.44329745e-01, -9.84878764e-02,  1.64631888e-01,  4.80844676e-02,\n",
            "       -2.85855055e-01, -1.07183225e-01, -3.64525169e-01,  1.22401156e-01,\n",
            "        3.44933987e-01, -2.93428689e-01,  2.05714479e-01, -2.13129416e-01,\n",
            "        9.50211287e-02,  8.32157806e-02,  2.05274656e-01,  1.42857134e+00,\n",
            "       -3.14655811e-01, -1.38562605e-01,  5.64810216e-01,  4.85359021e-02,\n",
            "       -1.92001760e-01,  1.48346975e-01, -3.18916664e-02,  9.08442959e-02,\n",
            "        2.68087417e-01, -2.56802410e-01, -2.80234486e-01,  8.33060127e-03,\n",
            "        3.16900909e-01,  4.29460078e-01, -4.49639589e-01, -3.94067168e-01,\n",
            "       -2.15669751e-01,  4.42997068e-01, -7.42784441e-02,  3.99129421e-01,\n",
            "       -1.96628883e-01,  2.31750056e-01,  1.99798226e-01, -7.05643237e-01,\n",
            "        9.69092175e-02,  2.36131940e-02,  2.45860502e-01, -9.74833891e-02,\n",
            "       -2.65247047e-01, -2.07701400e-01, -2.34198406e-01, -1.66667029e-01,\n",
            "        1.28692612e-01,  1.71237409e-01, -6.25047266e-01,  1.33160755e-01,\n",
            "        4.45004553e-01, -1.77474692e-01,  2.55192816e-01,  7.17905983e-02,\n",
            "        1.30876556e-01,  2.47643530e-01,  7.22455680e-02,  1.65747721e-02,\n",
            "        1.48365200e-01, -2.89230764e-01, -5.14988363e-01,  2.50688165e-01,\n",
            "        4.01663661e-01,  1.46828704e-02,  2.49306243e-02, -1.34893656e-01,\n",
            "       -1.62049532e-01, -2.26920664e-01, -5.45780547e-02, -1.61702428e-02,\n",
            "        9.51640010e-02, -7.01853111e-02, -5.44757545e-02, -3.14663979e-03,\n",
            "        1.15434051e-01,  3.42205167e-01, -1.94718778e-01,  2.93373883e-01,\n",
            "        4.38443184e-01,  7.90580094e-01, -1.72882512e-01, -1.93090737e-01,\n",
            "        3.46579659e-03,  4.18870896e-01, -2.31171269e-02,  5.18545449e-01,\n",
            "       -2.73264676e-01, -2.93511838e-01,  1.33823112e-01,  1.43622652e-01,\n",
            "       -2.81095952e-01,  4.13680464e-01, -1.54515013e-01,  3.95090915e-02,\n",
            "       -4.42999721e-01, -3.00640434e-01,  2.79415637e-01,  1.00426249e-01,\n",
            "        1.77166268e-01, -4.43203598e-01,  2.41366420e-02,  4.18472737e-01,\n",
            "       -3.98025475e-02,  3.93481612e-01,  7.84352124e-02,  4.09974843e-01,\n",
            "       -8.16186443e-02,  7.22499117e-02, -6.71950758e-01, -7.46381581e-02,\n",
            "        7.27765039e-02, -6.20765798e-02, -2.84276694e-01, -3.27577978e-01,\n",
            "        2.52577215e-01, -4.25563641e-02,  4.76242453e-01,  6.48981392e-01,\n",
            "       -1.02302141e-01, -4.95877594e-01, -3.93541604e-01, -2.69888163e-01,\n",
            "       -2.98354715e-01, -1.58319492e-02, -1.21815234e-01,  2.28784308e-01,\n",
            "        5.79891307e-03, -6.06966436e-01, -1.18882619e-01,  1.76041499e-02,\n",
            "        2.32352912e-02, -1.76305875e-01,  5.33451855e-01, -1.49227828e-02,\n",
            "        1.20301686e-01,  5.54296732e-01,  4.06387329e-01,  9.77221131e-02,\n",
            "        4.41520005e-01,  1.24123186e-01, -2.53539205e-01,  5.84606193e-02,\n",
            "        9.89798829e-02, -3.52578998e-01, -8.81462470e-02, -7.42154300e-01,\n",
            "        2.25166082e-01,  1.29533723e-01,  6.77750051e-01,  3.36688519e-01],\n",
            "      dtype=float32), 'Duration': array([-2.17499405e-01, -6.27695918e-01, -7.84299314e-01, -1.77336574e-01,\n",
            "       -4.65439372e-02, -3.36409748e-01, -1.51162341e-01,  4.51155543e-01,\n",
            "       -1.62495568e-01,  4.79487896e-01,  2.83759058e-01,  5.47379494e-01,\n",
            "       -3.97811979e-01, -2.63801754e-01, -5.36908507e-01, -4.96205866e-01,\n",
            "        5.58810890e-01,  2.76147097e-01, -1.39662966e-01,  4.89739239e-01,\n",
            "        5.73166907e-01,  4.79795188e-01, -3.63767743e-02, -1.99862763e-01,\n",
            "        7.80641362e-02,  3.16288024e-01, -8.00061226e-02, -5.25534563e-02,\n",
            "        5.43158650e-01,  1.58106729e-01,  2.82571372e-02,  1.38946891e-01,\n",
            "       -2.34309763e-01,  5.83422661e-01, -5.34089923e-01, -5.93853712e-01,\n",
            "        7.41425753e-02,  3.45357731e-02,  7.55453765e-01,  3.58126253e-01,\n",
            "        4.87976551e-01, -1.71465781e-02,  9.97854620e-02,  2.87541598e-01,\n",
            "        8.84645134e-02,  5.19180059e-01, -2.84642354e-02, -8.68459404e-01,\n",
            "        3.25676292e-01, -1.96065724e-01, -1.08992100e-01, -5.42933159e-02,\n",
            "        1.17995620e-01, -9.48025137e-02, -7.30502486e-01,  6.53982386e-02,\n",
            "       -2.47406170e-01,  1.44967288e-01, -4.08712953e-01, -9.88233089e-02,\n",
            "       -3.86309326e-01,  5.55622101e-01, -1.32809386e-01, -4.92988192e-02,\n",
            "        3.44143391e-01, -2.52338201e-01, -2.88713694e-01, -5.42804413e-02,\n",
            "       -2.03663498e-01, -5.32083064e-02, -2.63048530e-01,  3.39686513e-01,\n",
            "        2.98430115e-01,  9.90113541e-02,  5.40048063e-01, -3.13784689e-01,\n",
            "        4.39579308e-04,  9.55189988e-02, -1.51577473e-01, -1.67957470e-01,\n",
            "        2.07647115e-01,  3.54698360e-01,  8.63917097e-02, -1.78651839e-01,\n",
            "        1.54898632e-02,  8.10013115e-02, -2.61359990e-01,  4.26444858e-01,\n",
            "       -1.13467991e-01, -2.47385129e-01, -1.86905980e-01,  7.18456581e-02,\n",
            "       -3.16203237e-01,  1.66755080e-01,  1.58533966e-03, -1.96596503e-01,\n",
            "        4.79007572e-01,  6.80430532e-02,  8.88974071e-02, -1.99126765e-01,\n",
            "       -1.65994868e-01,  9.16339681e-02, -8.60096440e-02, -1.35938730e-02,\n",
            "       -8.56152624e-02, -5.20353436e-01,  5.51663637e-01,  8.82276297e-02,\n",
            "       -4.60864663e-01, -6.27568662e-01,  4.33592349e-02, -1.61997139e-01,\n",
            "        1.59050986e-01, -2.07406551e-01,  1.62023276e-01,  2.79249310e-01,\n",
            "        4.29538101e-01,  3.29348780e-02, -2.24170566e-01,  2.28924558e-01,\n",
            "        1.94262803e-01, -3.00146699e-01, -8.06225464e-02,  2.00960279e-01,\n",
            "        5.41288480e-02, -3.22691023e-01,  6.04936481e-01, -4.64981675e-01,\n",
            "        4.76447582e-01,  5.63275218e-01,  1.94609091e-01,  1.87908992e-01,\n",
            "       -5.05336337e-02, -1.13457888e-01,  4.14290637e-01,  1.39708459e-01,\n",
            "       -4.29138869e-01, -3.01593959e-01,  1.52418539e-01,  3.76509689e-02,\n",
            "       -1.18594095e-02, -1.70411110e-01, -2.59291351e-01, -1.71453863e-01,\n",
            "       -3.36772278e-02, -3.04320544e-01, -4.08834428e-01, -3.04683983e-01,\n",
            "       -4.31426823e-01, -3.78454775e-01, -1.17955196e+00, -7.94680476e-01,\n",
            "        7.38568008e-02,  2.36890063e-01,  2.31609821e-01, -4.59428787e-01,\n",
            "        1.76717043e-01,  1.33995935e-01, -9.60707068e-02,  1.36226997e-01,\n",
            "        5.40317893e-01, -3.38448852e-01,  1.48263633e-01, -4.56433356e-01,\n",
            "       -1.13715045e-02, -3.03769317e-02, -4.72910553e-02, -3.27593029e-01,\n",
            "       -1.00715920e-01, -6.66529834e-02,  1.01512343e-01, -1.23762749e-01,\n",
            "        2.71562099e-01, -4.54141468e-01,  1.01515576e-01, -4.20192719e-01,\n",
            "       -4.24060822e-02, -3.58488917e-01, -1.70171306e-01,  2.12410495e-01,\n",
            "        2.22392842e-01, -2.14759445e+00, -1.02764152e-01, -4.20266762e-02,\n",
            "        2.55469263e-01, -5.26363969e-01,  3.64532709e-01, -7.27469772e-02,\n",
            "        3.41655225e-01,  1.34859025e-01, -5.62950850e-01,  2.23684400e-01,\n",
            "       -5.26187778e-01, -5.73148369e-04, -4.73841541e-02,  4.44525242e-01,\n",
            "        2.86567241e-01, -9.39146280e-02, -8.68655741e-02,  9.51617509e-02,\n",
            "       -2.61846811e-01,  5.23840189e-01, -1.70886844e-01, -2.01830089e-01,\n",
            "        5.61635420e-02,  4.75708455e-01,  4.44830656e-02,  4.29258436e-01,\n",
            "        5.59511892e-02, -1.83036719e-02, -1.24463169e-02, -4.90079634e-02,\n",
            "        8.09868872e-02,  1.17636070e-01,  4.53511029e-02,  4.48609114e-01,\n",
            "       -3.58655423e-01, -4.06816691e-01,  2.96640456e-01, -1.43726155e-01,\n",
            "       -2.26108313e-01, -2.20582634e-01, -3.06091934e-01, -2.30709553e-01,\n",
            "        3.85691732e-01,  5.88942051e-01,  1.62061453e-01,  5.77306151e-01,\n",
            "       -1.48208767e-01, -3.36662233e-01, -4.51690927e-02,  9.78273749e-02,\n",
            "        7.12225854e-01,  7.76523054e-02,  1.27770662e-01, -4.98292595e-02,\n",
            "       -1.86004147e-01,  4.87801790e-01, -4.87551093e-02, -4.03000563e-02,\n",
            "       -4.23193187e-01, -6.61028862e-01, -1.33031920e-01,  2.93900728e-01,\n",
            "        2.51448989e-01,  3.37143958e-01,  5.60658835e-02,  6.17633998e-01,\n",
            "        1.61725104e-01,  4.08060312e-01, -7.09731644e-03, -1.14271119e-02,\n",
            "        4.87488151e-01,  2.00413883e-01, -5.75792432e-01,  1.82713401e-02,\n",
            "        1.53062744e-02, -1.36892796e-01,  2.69358218e-01,  3.70800018e-01,\n",
            "        2.94777840e-01, -2.98391432e-01,  2.85840958e-01,  2.93761075e-01,\n",
            "       -2.90499091e-01,  9.42912549e-02,  4.80063558e-01,  1.61311328e-01,\n",
            "       -4.12581921e-01,  5.56041226e-02,  2.33473885e-03,  5.40473402e-01,\n",
            "        3.42874646e-01, -3.50744188e-01, -5.06402373e-01, -8.18157554e-01,\n",
            "       -1.82750493e-01, -5.28003946e-02,  1.20279931e-01, -7.78219104e-01,\n",
            "       -7.08120048e-01, -6.74695611e-01,  3.37379500e-02, -5.70554510e-02,\n",
            "       -1.23163179e-01,  2.82679796e-01,  3.19311678e-01, -3.75538617e-02,\n",
            "       -1.38470322e-01, -2.05589145e-01, -1.21249557e-01, -3.73369642e-02,\n",
            "        2.95964837e-01,  1.23204023e-01,  1.85785443e-02, -2.13566154e-01,\n",
            "        3.66764605e-01,  4.29338455e-01,  1.91813290e-01, -3.75285268e-01,\n",
            "       -8.38282034e-02,  8.14687759e-02,  1.48199946e-01, -1.23949870e-02,\n",
            "        1.71426699e-01,  9.06455368e-02,  3.75413984e-01,  5.61414480e-01,\n",
            "       -4.41751659e-01, -5.77300251e-01, -2.91070104e-01, -1.57010004e-01,\n",
            "       -8.53008032e-02,  4.35408205e-01,  1.96450353e-01, -6.83377862e-01,\n",
            "        1.57278791e-01, -1.17269561e-01, -2.68682659e-01, -5.31943679e-01,\n",
            "        2.55078018e-01,  2.53095686e-01,  3.43001150e-02,  5.55710644e-02,\n",
            "        2.46935308e-01,  2.61425495e-01,  2.51680613e-04, -9.36141890e-03,\n",
            "        2.69836690e-02, -8.48350465e-01,  3.17165107e-01, -3.18796724e-01,\n",
            "       -5.27702272e-04, -5.71112335e-01,  4.59800482e-01,  2.07203835e-01,\n",
            "        2.36047551e-01, -3.67973328e-01, -1.58264279e-01, -5.02412319e-01,\n",
            "        4.14453000e-01, -4.68315691e-01, -1.78633898e-01,  1.53974697e-01,\n",
            "        5.65201938e-02, -1.08200870e-01,  4.22884345e-01, -3.69357347e-01,\n",
            "       -2.35269472e-01,  2.34704122e-01,  1.85185134e-01,  5.62954471e-02,\n",
            "       -3.83436739e-01, -2.25235254e-01,  3.25097978e-01, -6.03180937e-02,\n",
            "        2.32849285e-01, -4.67618763e-01,  3.43926489e-01,  2.46929049e-01,\n",
            "       -9.39977467e-02, -7.34830871e-02, -1.48253560e-01, -2.90309995e-01,\n",
            "        3.03394079e-01,  3.27379346e-01,  3.65310669e-01, -7.45830610e-02,\n",
            "        2.68191211e-02, -2.27828264e-01,  2.49155909e-01,  2.33585507e-01,\n",
            "       -1.14257902e-01, -2.06110775e-02,  4.88459766e-01, -2.58635938e-01,\n",
            "       -4.31078851e-01,  1.01741225e-01, -4.99552965e-01, -4.19536412e-01,\n",
            "        1.10096656e-01, -2.52501369e-02, -1.19205620e-02, -5.01066923e-01,\n",
            "       -6.07126467e-02, -2.31863350e-01, -1.16536081e-01,  4.67838049e-01,\n",
            "        1.14691138e-01, -1.01321740e-02, -2.26688355e-01,  1.05874091e-01,\n",
            "       -1.98122002e-02,  7.24832237e-01, -1.25440091e-01, -3.52368593e-01,\n",
            "        5.78032076e-01, -2.73269743e-01, -6.22498803e-02,  5.85134923e-01,\n",
            "       -3.34990233e-01, -6.18253760e-02, -1.29727840e-01,  1.50684386e-01,\n",
            "       -1.76483896e-02, -1.95606753e-01,  1.45081878e-01,  3.17638554e-02,\n",
            "       -2.34433532e-01, -1.08959794e-01, -1.44247591e-01,  3.78095716e-01,\n",
            "       -3.87800708e-02, -8.46702605e-02,  8.89319330e-02,  2.85688043e-01,\n",
            "       -2.38038883e-01,  6.37082532e-02, -1.41625151e-01, -3.19017708e-01,\n",
            "        5.26524663e-01,  1.54023409e-01, -1.36117637e-01,  2.89014757e-01,\n",
            "       -1.17902937e+01,  4.67754304e-01,  4.03823227e-01, -3.31486873e-02,\n",
            "       -2.34166503e-01, -1.48944423e-01,  3.26050252e-01, -2.22318739e-01,\n",
            "        3.11691254e-01, -4.50134367e-01, -1.83408290e-01,  4.85181898e-01,\n",
            "       -1.63648829e-01, -3.74437571e-01, -7.18062639e-01, -6.32221550e-02,\n",
            "       -9.99411568e-03, -6.05374694e-01,  1.48656629e-02,  1.36654034e-01,\n",
            "       -2.87591785e-01, -4.85210828e-02, -1.02260709e-02,  3.60913575e-01,\n",
            "        1.24126393e-02, -4.84618038e-01, -2.81109840e-01, -2.29066372e-01,\n",
            "       -8.00465867e-02,  3.22896898e-01, -4.49919224e-01,  4.05101389e-01,\n",
            "        1.15614772e-01,  4.04465139e-01, -2.76691079e-01, -1.29790545e-01,\n",
            "       -6.51166365e-02, -3.70187044e-01, -3.38088155e-01, -2.72897422e-01,\n",
            "       -2.58011609e-01,  3.83846718e-03, -6.91269338e-01,  5.09440541e-01,\n",
            "        1.55239880e-01,  2.19789237e-01,  6.66496277e-01,  3.17710519e-01,\n",
            "       -7.55395144e-02,  1.76942557e-01, -5.25411889e-02, -3.93505543e-01,\n",
            "       -3.90274897e-02, -4.77185220e-01, -7.90437222e-01, -5.03123283e-01,\n",
            "        5.23213029e-01, -6.37890518e-01,  1.53495371e-01, -5.87001666e-02,\n",
            "       -6.76472392e-03, -4.97578345e-02,  2.34682918e-01, -4.76981103e-02,\n",
            "       -7.71330118e-01, -4.99363840e-02, -8.07631761e-02, -2.82896757e-01,\n",
            "       -1.14164636e-01, -5.46122611e-01, -4.37952355e-02,  5.88197887e-01,\n",
            "        5.18877268e-01, -2.74062734e-02, -2.82948017e-01, -2.37225965e-02,\n",
            "       -2.43750244e-01, -3.43562737e-02,  1.50188100e+00,  4.30110320e-02,\n",
            "       -3.72987166e-02, -1.19037382e-01, -6.13519102e-02,  1.95296854e-01,\n",
            "       -3.39121938e-01, -3.54223371e-01,  3.17210779e-02, -9.33319092e-01,\n",
            "        3.89721394e-01,  7.49046504e-02,  1.12863496e-01, -1.36990279e-01,\n",
            "        6.81852847e-02,  3.40697616e-01,  3.16926956e-01,  5.84878147e-01,\n",
            "       -7.50084102e-01,  3.07065517e-01,  1.01619422e-01,  2.77692407e-01,\n",
            "        3.76282632e-01, -1.94130421e-01, -2.24826619e-01, -6.96583033e-01,\n",
            "       -1.86808676e-01, -4.16590452e-01,  1.78287357e-01,  4.67711151e-01,\n",
            "       -1.26335472e-01,  3.33938062e-01, -1.93869442e-01,  1.48364991e-01,\n",
            "        2.75701642e-01,  1.96603224e-01, -5.85737079e-02, -1.47809938e-01,\n",
            "       -9.92770121e-02,  1.69694781e-01, -4.09099340e-01, -2.44474053e-01,\n",
            "       -1.69614702e-01, -2.09033459e-01,  2.60912806e-01, -1.37223955e-02,\n",
            "       -2.29314268e-02, -1.93358690e-01, -4.11424339e-01,  1.94807649e-01,\n",
            "        3.36316288e-01,  1.63836166e-01,  6.29224539e-01,  4.48254675e-01,\n",
            "       -2.33312786e-01, -4.79509942e-02, -2.10987747e-01, -6.84222400e-01,\n",
            "       -7.39653111e-01, -2.74795771e-01,  4.69249576e-01, -3.26145858e-01,\n",
            "        1.16382524e-01,  1.46668196e-01, -1.45185795e-02,  1.28752381e-01,\n",
            "       -1.57911666e-02,  1.20309383e-01,  5.00736356e-01,  4.89375800e-01,\n",
            "       -3.18538398e-01, -3.21329646e-02, -3.78535539e-02, -1.40598029e-01,\n",
            "        7.14471564e-03, -2.11077258e-01, -3.19063276e-01, -3.38025212e-01,\n",
            "       -1.40173614e-01, -2.28578508e-01, -1.35317504e-01, -3.27540249e-01,\n",
            "        2.84245223e-01,  1.20126888e-01, -2.13857532e-01, -2.12971181e-01,\n",
            "       -8.75878260e-02,  5.52231789e-01, -5.12483865e-02, -4.43840474e-01,\n",
            "       -6.21543407e-01, -9.68247354e-02,  1.02125760e-02,  2.92697251e-01,\n",
            "        3.18701193e-02, -3.44540030e-02, -3.91786098e-01,  1.56479061e-01,\n",
            "       -2.62401283e-01,  7.01819837e-01, -1.35639071e-01, -5.65195203e-01,\n",
            "        1.01019904e-01,  5.01063585e-01,  5.18691167e-02, -1.75274909e-01,\n",
            "       -1.56686097e-01, -9.25868079e-02, -2.56823659e-01, -6.22739553e-01,\n",
            "       -9.81656462e-02, -4.15731311e-01,  3.74876231e-01,  7.34210432e-01,\n",
            "        4.30201054e-01,  2.22004056e-01,  4.42095518e-01,  9.59071964e-02,\n",
            "       -2.23714590e-01,  4.32644695e-01, -2.34594047e-01,  1.14471160e-01,\n",
            "       -4.80526760e-02,  2.17028186e-01, -2.16648191e-01, -1.97106004e-01,\n",
            "       -1.98978007e-01, -2.84950845e-02,  5.71476817e-01,  1.63335228e+00,\n",
            "       -1.07695580e-01, -2.32712492e-01,  2.49847412e-01, -7.83581361e-02,\n",
            "       -6.97171688e-02,  1.01532124e-01, -1.02059886e-01, -1.24671094e-01,\n",
            "        7.38270879e-01,  1.52262628e-01, -1.88251406e-01, -3.46049368e-02,\n",
            "        2.91794062e-01,  4.14272904e-01, -2.60835528e-01, -2.59549856e-01,\n",
            "       -8.19700360e-02,  3.28286469e-01, -1.47784591e-01,  7.13470697e-01,\n",
            "        2.54276156e-01,  3.83582801e-01, -5.61212539e-04, -1.12181261e-01,\n",
            "        6.97364509e-01,  2.49515682e-01,  6.28122211e-01,  9.59174857e-02,\n",
            "       -1.76787362e-01, -6.99346960e-02,  1.81893021e-01, -2.45097071e-01,\n",
            "        3.82982977e-02,  3.34801435e-01, -5.78845978e-01,  6.47558793e-02,\n",
            "        1.97711095e-01,  3.60434055e-01,  4.59296703e-01,  2.01626308e-02,\n",
            "        1.56755298e-01, -1.02197707e-01, -2.30187520e-01, -3.81650925e-01,\n",
            "        4.12365407e-01, -2.62018561e-01, -3.70225787e-01,  3.80693525e-01,\n",
            "        6.17261767e-01, -1.56905204e-01,  4.36630249e-02, -1.47226781e-01,\n",
            "        3.99102382e-02, -3.61286968e-01, -3.50092277e-02, -2.53035188e-01,\n",
            "        2.67682523e-01, -1.56733811e-01, -1.99798465e-01, -2.67041445e-01,\n",
            "        1.38549924e-01,  4.46216226e-01, -1.94820851e-01,  8.76841173e-02,\n",
            "        7.52413720e-02,  4.58372116e-01, -2.29469031e-01,  1.02983370e-01,\n",
            "        9.16424021e-02,  2.46671245e-01, -1.24100208e-01,  6.42452002e-01,\n",
            "       -2.24720150e-01, -1.58153474e-01, -4.64750342e-02,  1.90550461e-02,\n",
            "       -1.39139473e-01,  4.12172258e-01,  2.14556511e-02, -9.32642221e-02,\n",
            "       -2.93180585e-01, -8.58252496e-02,  4.10129279e-01,  2.17630625e-01,\n",
            "        1.10935271e-01, -4.02880400e-01,  2.10356995e-01,  2.62760699e-01,\n",
            "        3.63480508e-01,  2.67487824e-01, -7.14094788e-02,  6.63196981e-01,\n",
            "       -1.84695601e-01,  1.15172945e-01, -4.32986736e-01,  3.14703248e-02,\n",
            "        8.25850815e-02, -2.67137557e-01, -1.78107321e-01, -3.53270531e-01,\n",
            "        2.41117522e-01, -4.85576570e-01,  8.79696965e-01,  8.08177888e-01,\n",
            "       -1.52296782e-01, -5.06857812e-01, -4.90987271e-01, -2.19779015e-01,\n",
            "       -4.18172479e-01,  2.51530647e-01, -2.51653373e-01,  3.76680225e-01,\n",
            "        1.56896725e-01, -4.62071955e-01, -1.31002769e-01,  3.92175794e-01,\n",
            "        1.35769099e-01, -1.38939932e-01,  6.50218487e-01,  1.91468764e-02,\n",
            "       -1.57205626e-01,  5.02914548e-01,  1.01783916e-01, -1.73110887e-01,\n",
            "        3.30924869e-01, -2.77300209e-01, -3.14380139e-01,  1.12178825e-01,\n",
            "       -2.05382869e-01, -4.62628365e-01, -1.46568999e-01, -4.72949445e-01,\n",
            "        2.34359622e-01,  3.99237722e-01,  6.22922778e-01,  2.76735276e-01],\n",
            "      dtype=float32), 'Severity': array([-1.47833437e-01, -7.52884388e-01, -4.41521257e-01, -6.27388358e-01,\n",
            "       -2.56706446e-01, -6.96212947e-02, -7.65527606e-01,  1.60834074e-01,\n",
            "        1.02583110e-01,  2.61672854e-01, -1.59665883e-01,  2.03257591e-01,\n",
            "        7.82750174e-02, -5.72069287e-01, -9.22227185e-03,  2.38892540e-01,\n",
            "       -3.27194512e-01,  3.84240717e-01,  7.81760439e-02,  3.36681008e-01,\n",
            "        6.87314719e-02,  7.42861480e-02,  9.50190350e-02, -1.02891803e-01,\n",
            "       -2.16008544e-01,  1.89988300e-01,  1.41868263e-01, -2.27006793e-01,\n",
            "        3.62307280e-01, -4.35011476e-01,  1.94335550e-01,  5.08722186e-01,\n",
            "       -3.18554282e-01,  2.89419591e-01, -8.15677494e-02, -1.15205124e-01,\n",
            "       -2.32936203e-01, -3.17947179e-01,  1.01619899e+00, -5.32802567e-02,\n",
            "       -2.21998334e-01,  8.76255870e-01,  2.12915838e-02,  2.56173313e-01,\n",
            "       -3.98878828e-02,  5.06833196e-01,  1.52528569e-01, -5.72017670e-01,\n",
            "        3.34509671e-01, -1.30818933e-01, -8.90063494e-02, -4.14519422e-02,\n",
            "        2.65285939e-01,  6.32861778e-02, -1.23079431e+00,  4.75081317e-02,\n",
            "       -9.00522172e-02,  1.15450136e-01, -5.70593834e-01,  2.02136114e-02,\n",
            "        2.64263656e-02,  8.62217844e-01,  2.88134217e-01,  7.72293657e-02,\n",
            "        2.30899617e-01, -2.69111753e-01, -3.15768838e-01, -6.43873513e-02,\n",
            "       -3.94470811e-01, -2.56300628e-01,  4.27619636e-01,  3.06277722e-01,\n",
            "        3.19096893e-01, -5.46716824e-02,  2.37501532e-01, -1.74843401e-01,\n",
            "        2.65138835e-01,  5.32159805e-02, -1.75172575e-02, -8.32875729e-01,\n",
            "        1.58371925e-01,  1.70850322e-01,  2.23741919e-01, -1.38717061e-02,\n",
            "       -1.24688350e-01,  9.78101790e-02,  4.26855266e-01,  2.31695294e-01,\n",
            "        5.51874451e-02,  7.57828876e-02,  3.97529602e-01,  5.18574845e-03,\n",
            "       -4.79539931e-01, -3.70470613e-01,  2.78938174e-01, -7.69264698e-01,\n",
            "        3.15112442e-01, -1.99945092e-01,  1.05573935e-02, -7.00607240e-01,\n",
            "       -5.18350527e-02, -2.07053095e-01,  3.82880419e-02,  3.34771156e-01,\n",
            "       -4.60325718e-01, -2.05290794e-01,  9.49279293e-02, -2.31564686e-01,\n",
            "       -3.25215757e-01, -1.85288280e-01, -4.30836603e-02, -8.02833438e-01,\n",
            "       -1.44106317e-02, -5.62948942e-01, -2.39610121e-01,  1.95400119e-01,\n",
            "        3.67546320e-01, -2.68574715e-01, -7.57010654e-02,  4.35333014e-01,\n",
            "        4.25457597e-01, -3.64026248e-01, -1.18885756e-01, -5.79979084e-02,\n",
            "        3.10870379e-01, -2.87557811e-01, -6.49174750e-02,  2.71656156e-01,\n",
            "        3.92183363e-01,  7.16611087e-01, -5.11779636e-03,  3.17621008e-02,\n",
            "       -4.41132069e-01,  3.92358601e-02, -2.41278380e-01, -6.07555434e-02,\n",
            "       -2.63553917e-01, -3.16183507e-01, -4.42605000e-04,  2.61515439e-01,\n",
            "       -3.30177397e-02, -6.28873482e-02, -4.33531940e-01,  2.74751604e-01,\n",
            "        5.57554185e-01, -1.65128186e-01,  2.30298266e-01, -2.03307912e-01,\n",
            "        3.51307094e-02, -3.26329112e-01, -7.87774444e-01, -3.29386175e-01,\n",
            "        3.07585478e-01, -2.08453625e-01, -1.98420078e-01, -2.91184872e-01,\n",
            "        4.62518454e-01,  3.26630294e-01,  7.03293905e-02,  4.18108076e-01,\n",
            "        7.62015700e-01, -1.48259729e-01,  2.45840371e-01, -3.94282043e-01,\n",
            "        7.68327266e-02,  3.96989197e-01, -5.04449487e-01, -2.05441713e-01,\n",
            "        1.10200278e-01, -1.15346685e-02, -8.26953053e-02, -3.39810789e-01,\n",
            "        3.98728102e-01,  3.99707481e-02,  3.03094745e-01, -2.45533079e-01,\n",
            "        4.08335656e-01,  1.15228146e-01,  2.39462964e-02,  4.23428833e-01,\n",
            "       -5.27968049e-01, -2.18435407e+00,  2.58599639e-01,  5.45877576e-01,\n",
            "        2.02067010e-02, -3.29644084e-01, -9.96541828e-02, -8.59111361e-03,\n",
            "        5.93227625e-01, -3.68616939e-01,  1.16807841e-01,  5.06106988e-02,\n",
            "        1.51028976e-01,  3.60109180e-01, -2.28033423e-01,  5.89480996e-01,\n",
            "        4.43826079e-01,  2.28367001e-01,  4.94544618e-02,  3.53089631e-01,\n",
            "       -7.14998186e-01, -1.16296187e-02,  6.56449020e-01, -8.92724283e-03,\n",
            "        1.57659322e-01,  4.20390099e-01, -2.77350485e-01, -1.25777155e-01,\n",
            "       -6.01640642e-02,  6.55631483e-01, -3.57428074e-01, -9.31188613e-02,\n",
            "       -2.47750372e-01,  1.71198875e-01,  2.04661489e-01,  1.79557592e-01,\n",
            "       -3.39218020e-01, -3.41110557e-01,  2.48301655e-01,  3.24099600e-01,\n",
            "       -2.02781186e-01, -2.11550772e-01, -2.58372754e-01, -3.74114573e-01,\n",
            "        9.74065065e-02,  3.96489590e-01, -2.10126206e-01,  1.91159308e-01,\n",
            "       -4.47737873e-01, -5.54197729e-01, -1.94158763e-01,  5.73365331e-01,\n",
            "        9.54414308e-02,  4.87497270e-01, -2.25319862e-01,  1.33616298e-01,\n",
            "       -5.93879938e-01,  4.82565284e-01, -5.10283113e-01, -4.95247185e-01,\n",
            "       -2.31105208e-01, -3.16208661e-01, -4.37414050e-01,  1.30244672e-01,\n",
            "        3.72510076e-01,  2.30699420e-01,  1.59448460e-02,  6.75210595e-01,\n",
            "        6.41465902e-01,  1.67718321e-01,  1.03195652e-01,  1.85990259e-02,\n",
            "        5.08192420e-01,  5.31172454e-01, -2.37413555e-01, -1.38918862e-01,\n",
            "        4.36900318e-01, -7.09487572e-02,  4.27793443e-01, -2.79343426e-01,\n",
            "       -4.19463962e-01,  1.16669849e-01,  9.91673917e-02,  6.86623931e-01,\n",
            "       -1.79527313e-01, -4.10245955e-01, -8.98387134e-02, -3.34265053e-01,\n",
            "       -2.99802534e-02,  3.63856018e-01, -4.97153372e-01,  6.87949836e-01,\n",
            "       -3.97365510e-01, -6.62236512e-02, -6.40661418e-01, -6.52590692e-01,\n",
            "       -2.64355719e-01, -1.73100531e-01,  4.84259501e-02, -7.83355713e-01,\n",
            "       -2.06575423e-01, -1.48645639e-01,  1.09815314e-01,  2.50588834e-01,\n",
            "        8.14087130e-03,  1.59430847e-01, -3.54754210e-01,  1.44937038e-01,\n",
            "       -1.62099168e-01, -3.35668981e-01,  5.49017405e-03, -7.33435899e-02,\n",
            "       -3.37087005e-01, -1.36706755e-01, -8.52972865e-02, -1.43107623e-01,\n",
            "       -1.46030784e-01,  5.04798703e-02, -8.12081322e-02,  1.06524140e-01,\n",
            "       -1.18348762e-01,  4.24662888e-01, -1.90306589e-01, -2.16882117e-02,\n",
            "       -3.13396975e-02,  1.72653109e-01,  1.64610714e-01,  6.44025266e-01,\n",
            "       -5.26651382e-01, -5.01227081e-01, -6.06255829e-02,  1.55053332e-01,\n",
            "       -7.22394288e-02, -2.16601118e-01, -1.02458499e-01, -9.66225863e-01,\n",
            "        1.47767946e-01,  5.90782687e-02, -4.67944235e-01, -5.16543686e-01,\n",
            "        1.93483993e-01, -1.77612811e-01, -6.71453327e-02, -5.80285043e-02,\n",
            "       -3.52114916e-01,  6.72979712e-01,  1.86078548e-01,  3.40910763e-01,\n",
            "       -2.17509530e-02, -4.75751400e-01, -2.79394793e-03, -3.76288474e-01,\n",
            "       -4.10307020e-01, -5.02839312e-02, -1.54929802e-01,  1.92100450e-01,\n",
            "        1.30791795e-02, -2.78348386e-01, -6.40868172e-02, -1.45661263e-02,\n",
            "        3.89203966e-01, -3.42907548e-01, -5.76763004e-02,  2.69040883e-01,\n",
            "       -1.54756382e-02,  1.16870463e-01,  3.23481619e-01, -6.83496654e-01,\n",
            "       -3.50514650e-01,  4.41205740e-01,  1.97125778e-01,  1.41183510e-01,\n",
            "       -1.96993038e-01, -1.76669523e-01,  2.23292157e-01,  5.22354364e-01,\n",
            "        1.48414578e-02, -2.69239068e-01,  2.91155040e-01,  2.05285192e-01,\n",
            "       -2.90590934e-02,  2.04314142e-01, -4.26474899e-01, -3.73404980e-01,\n",
            "       -1.16228372e-01,  5.69458008e-01,  4.50811565e-01, -3.51224780e-01,\n",
            "        2.00774491e-01, -1.70106202e-01, -9.16421711e-02, -6.32534862e-01,\n",
            "       -1.05614260e-01,  2.25122929e-01,  9.60450396e-02, -7.55615115e-01,\n",
            "       -3.05419147e-01, -3.97687614e-01, -5.81127048e-01, -4.17917430e-01,\n",
            "        2.06294298e-01, -2.98141271e-01, -3.31892192e-01, -4.08525109e-01,\n",
            "        1.31014019e-01, -5.08088589e-01, -2.26238608e-01,  5.30044258e-01,\n",
            "        1.22207820e-01, -1.47146001e-01, -1.05106652e-01, -4.73704636e-01,\n",
            "       -3.67334783e-01,  5.27493000e-01, -1.83092505e-01, -1.53597355e-01,\n",
            "        2.36451596e-01, -4.33320940e-01, -8.64813179e-02,  9.89545509e-02,\n",
            "       -1.61478117e-01, -2.22954243e-01, -9.33186635e-02,  1.60104305e-01,\n",
            "        3.25984478e-01, -6.03713334e-01, -1.10287562e-01,  7.47035146e-02,\n",
            "        8.54123235e-02, -1.31213889e-01,  5.42922854e-01, -3.79496813e-01,\n",
            "        1.80229004e-02, -3.11273001e-02, -4.86261360e-02,  3.16336960e-01,\n",
            "       -4.44063451e-03,  9.46906731e-02,  1.11549258e-01, -5.33652067e-01,\n",
            "        6.17359042e-01,  4.33009952e-01,  2.61997819e-01,  9.31838512e-01,\n",
            "       -1.11578617e+01,  3.54074419e-01,  2.68187433e-01, -2.21110761e-01,\n",
            "       -8.02858174e-02,  4.59606275e-02,  7.54585788e-02, -1.26842484e-01,\n",
            "        7.72997290e-02, -3.44754271e-02,  4.51281965e-02, -4.68150020e-01,\n",
            "       -2.78902829e-01, -9.76709127e-02, -5.46137571e-01, -3.63320440e-01,\n",
            "       -2.27501199e-01, -7.02184618e-01,  3.72828782e-01,  1.20881021e-01,\n",
            "        3.51159990e-01,  1.60220021e-03, -2.30084047e-01,  6.96316123e-01,\n",
            "       -1.12560175e-01, -5.33490591e-02,  3.89834404e-01, -1.62586905e-02,\n",
            "        4.29502964e-01,  3.61681208e-02, -2.94084638e-01,  4.07047719e-01,\n",
            "        6.10654615e-02,  4.00551736e-01, -1.02285624e-01, -1.85602546e-01,\n",
            "        2.55084455e-01, -2.84332689e-02, -2.96408892e-01, -2.55285054e-01,\n",
            "       -1.13343552e-01,  6.57936782e-02, -4.12160717e-02,  3.20714593e-01,\n",
            "        5.96671820e-01,  7.03010738e-01,  9.71775234e-01,  6.63452685e-01,\n",
            "        3.97626236e-02,  4.81336936e-03, -2.33720273e-01, -4.72825408e-01,\n",
            "        5.25783658e-01, -4.00446415e-01, -9.33307335e-02,  8.81820917e-03,\n",
            "        1.67465910e-01, -3.72624218e-01,  3.53626490e-01, -1.40882701e-01,\n",
            "       -1.55086845e-01,  4.23037052e-01,  3.14366668e-01,  1.48387074e-01,\n",
            "       -4.66785610e-01,  1.98026508e-01,  4.44294155e-01, -1.26996428e-01,\n",
            "       -1.67994946e-01, -4.56292480e-01, -3.79354537e-01, -2.14754820e-01,\n",
            "        4.82860923e-01, -1.58967495e-01,  3.85220855e-01, -3.46520036e-01,\n",
            "       -3.60640019e-01,  1.45505831e-01,  1.53540802e+00,  2.64927685e-01,\n",
            "        4.76820096e-02,  1.08764842e-01, -9.13996845e-02,  9.67240781e-02,\n",
            "       -4.19243038e-01,  1.48008630e-01, -3.63401249e-02, -3.09368819e-01,\n",
            "        3.45960677e-01,  2.93708861e-01, -3.00151825e-01, -1.13237381e-01,\n",
            "       -1.54980391e-01, -8.80486369e-02,  2.46299848e-01,  6.94109321e-01,\n",
            "       -1.05767393e+00, -5.77312373e-02, -5.36912344e-02,  5.28738499e-01,\n",
            "        2.79900312e-01, -5.15772067e-02, -3.33641499e-01, -9.92650837e-02,\n",
            "       -3.15852053e-02,  1.98682532e-01,  2.88649313e-02, -6.68711960e-02,\n",
            "        1.34881586e-01,  4.85455811e-01, -2.53871202e-01,  3.24564248e-01,\n",
            "       -2.46195361e-01,  4.48093832e-01, -5.10934949e-01, -6.91693187e-01,\n",
            "       -6.25762492e-02,  2.47917384e-01, -6.93422258e-01, -1.27115175e-01,\n",
            "       -3.76462936e-02, -1.63486339e-02,  2.82342285e-01,  4.13764060e-01,\n",
            "       -3.52650404e-01,  1.02279890e+00, -7.91976213e-01,  3.12090158e-01,\n",
            "       -2.45363280e-01, -3.29078168e-01,  2.42743403e-01,  3.67526770e-01,\n",
            "       -6.49159729e-01,  3.82828593e-01, -4.57010448e-01, -1.02433181e+00,\n",
            "       -9.57104564e-01,  1.36959061e-01,  4.76826012e-01, -4.69626606e-01,\n",
            "        1.23421714e-01, -4.83978987e-01, -4.75367904e-02,  2.13560864e-01,\n",
            "        1.50404066e-01,  2.38883942e-01,  6.80856705e-01,  5.34159988e-02,\n",
            "       -3.59394163e-01,  1.47535592e-01, -2.58657157e-01, -6.76970363e-01,\n",
            "       -1.46611363e-01,  2.80005276e-01,  1.64528519e-01, -3.03953946e-01,\n",
            "       -2.60552377e-01, -3.05635989e-01, -3.73981297e-01,  4.97334659e-01,\n",
            "       -4.44961004e-02,  4.75546688e-01, -1.30688533e-01,  7.31556341e-02,\n",
            "        3.22512656e-01,  7.08590001e-02,  1.31529078e-01, -3.08585852e-01,\n",
            "       -7.15257764e-01, -3.21025215e-03,  1.57094643e-01,  3.21459681e-01,\n",
            "        4.58113641e-01,  1.29131630e-01, -2.14079663e-01,  1.58786625e-01,\n",
            "       -1.15019366e-01,  8.63543868e-01, -5.38816214e-01, -4.93984222e-01,\n",
            "       -2.89778829e-01, -3.11659634e-01, -5.58824502e-02, -6.94733500e-01,\n",
            "       -9.93553698e-02,  1.88688919e-01, -1.91891938e-01,  1.03898868e-02,\n",
            "        2.14215249e-01,  6.76327497e-02,  1.12187147e-01,  9.39904451e-02,\n",
            "        6.38202667e-01, -1.27182662e-01,  9.69661713e-01, -1.45877004e-01,\n",
            "       -4.54737127e-01, -3.77791852e-01, -3.51353496e-01, -6.43918067e-02,\n",
            "        4.29442048e-01, -4.70053524e-01,  4.22298670e-01, -2.49666482e-04,\n",
            "       -1.22531220e-01,  3.43473315e-01,  5.49311280e-01,  1.40442395e+00,\n",
            "       -3.38059723e-01, -8.60599428e-02,  1.20373881e+00,  4.31142896e-01,\n",
            "        3.69412363e-01,  2.52589911e-01, -1.16787486e-01, -1.64498493e-01,\n",
            "        5.94915032e-01, -3.32884878e-01, -1.62393481e-01,  8.13442692e-02,\n",
            "        7.80859739e-02,  4.17640924e-01, -6.04210317e-01, -2.93108076e-01,\n",
            "        2.71831483e-01,  6.31671309e-01, -5.86160421e-01,  9.65469107e-02,\n",
            "       -4.08796728e-01,  1.23819366e-01,  4.87531513e-01, -6.02266550e-01,\n",
            "       -1.55516431e-01,  3.24239731e-01,  4.40448612e-01, -2.94773310e-01,\n",
            "       -4.83254284e-01, -6.96450621e-02, -1.32545665e-01, -1.32830560e-01,\n",
            "        4.92971748e-01, -3.60485353e-02, -8.00793350e-01,  5.03365755e-01,\n",
            "        6.27346396e-01, -3.00658941e-01,  3.50742936e-01,  1.09083563e-01,\n",
            "        1.13998558e-02,  1.20230518e-01,  3.30244601e-01,  6.88750446e-01,\n",
            "        5.44916503e-02, -6.91307724e-01, -4.50234175e-01,  5.58029294e-01,\n",
            "        6.36349440e-01,  3.01437899e-02,  3.08558699e-02,  3.46918672e-01,\n",
            "       -1.65134534e-01, -1.72044486e-01, -1.31908715e-01, -1.35647833e-01,\n",
            "       -3.07675600e-01, -6.22652099e-02, -7.41281500e-03, -3.13956499e-01,\n",
            "       -2.64799535e-01,  4.46451873e-01, -3.91202047e-02,  1.62960827e-01,\n",
            "        5.07041872e-01,  8.98987234e-01, -2.65569180e-01, -2.26992927e-02,\n",
            "        8.01169991e-01,  3.87431383e-01, -1.43289611e-01,  2.29872346e-01,\n",
            "       -7.36857504e-02, -4.51031655e-01, -1.51152313e-01,  3.23003620e-01,\n",
            "       -4.71475385e-02,  3.45879614e-01, -1.99559420e-01,  4.27744463e-02,\n",
            "       -2.61897683e-01, -5.60836315e-01,  2.14638323e-01,  2.85801888e-01,\n",
            "       -4.75050472e-02, -5.89574575e-01,  3.53602134e-02,  6.75177097e-01,\n",
            "       -2.54014969e-01,  1.06528625e-01, -1.27609000e-01,  7.93039322e-01,\n",
            "        2.86389768e-01,  6.98446389e-03, -8.94244671e-01, -2.50642113e-02,\n",
            "       -5.35060346e-01,  2.08204627e-01, -6.31858468e-01, -7.46623650e-02,\n",
            "       -4.30212989e-02, -2.21384808e-01,  3.29815686e-01,  9.03938413e-01,\n",
            "       -2.86415577e-01, -5.99659562e-01, -6.05315089e-01, -2.07765281e-01,\n",
            "       -2.25819856e-01, -3.95919345e-02, -5.87493479e-02,  3.73560667e-01,\n",
            "       -7.21752364e-03, -5.04391551e-01, -3.33668292e-01, -1.76086634e-01,\n",
            "        5.26073158e-01, -5.17524362e-01,  9.99089420e-01,  1.22513972e-01,\n",
            "        1.93254694e-01,  5.63942909e-01,  3.30515862e-01,  1.96384877e-01,\n",
            "        2.15691030e-01, -5.97069263e-02, -3.63155067e-01,  3.56131196e-01,\n",
            "       -2.04512961e-02, -3.03791523e-01, -2.28461176e-01, -8.77003551e-01,\n",
            "        3.51867139e-01,  5.35829842e-01,  4.16366309e-01,  5.14631987e-01],\n",
            "      dtype=float32), 'Time': array([-5.23246765e-01, -4.74128723e-01, -9.47024286e-01, -3.77950430e-01,\n",
            "       -1.90291017e-01, -2.59641528e-01, -4.55109686e-01,  3.77969950e-01,\n",
            "        2.32692122e-01, -5.69703011e-03,  3.06739539e-01,  2.44172186e-01,\n",
            "       -6.12198524e-02, -9.61663961e-01, -4.62236255e-01, -5.26705794e-02,\n",
            "        3.95072922e-02,  2.53370941e-01,  2.90382922e-01,  4.52641964e-01,\n",
            "        7.82336116e-01,  4.06737089e-01,  1.58570081e-01, -3.34215373e-01,\n",
            "        1.86223611e-01,  3.34440246e-02, -3.58369872e-02, -3.26281905e-01,\n",
            "        7.34169006e-01,  1.27736300e-01,  4.54031169e-01,  2.30376050e-01,\n",
            "       -3.82291764e-01,  4.08008188e-01, -6.11714602e-01, -6.55285954e-01,\n",
            "        3.91907901e-01, -7.97384024e-01,  9.38716650e-01,  3.14393580e-01,\n",
            "        3.86043966e-01,  4.25092787e-01,  1.25805542e-01,  2.20239878e-01,\n",
            "        2.15081617e-01,  3.20177734e-01,  2.83546448e-01, -8.12945366e-01,\n",
            "        3.51704657e-01,  2.63654944e-02, -1.50473163e-01, -6.24727309e-02,\n",
            "       -9.78208333e-02, -2.07144424e-01, -6.93842828e-01,  3.28410387e-01,\n",
            "       -8.33698064e-02,  2.27226317e-01, -1.75442383e-01,  2.66615510e-01,\n",
            "       -1.40164346e-01,  2.91095346e-01, -1.50344178e-01, -2.78292477e-01,\n",
            "        1.87692150e-01, -3.32262486e-01, -8.58007133e-01, -2.24996321e-02,\n",
            "       -6.24591947e-01,  5.78018539e-02, -4.34576273e-02,  4.00233597e-01,\n",
            "        1.34002715e-01, -9.14789662e-02,  5.62547386e-01, -3.01084578e-01,\n",
            "        2.03438491e-01, -5.81540108e-01, -1.62243303e-02, -3.00393790e-01,\n",
            "        8.15371126e-02,  4.23943460e-01, -1.97553381e-01, -4.57011282e-01,\n",
            "       -8.07241350e-02, -1.75921842e-01, -2.24929795e-01,  7.36017153e-02,\n",
            "        6.72359094e-02, -8.47360343e-02,  5.19791804e-02,  2.37898514e-01,\n",
            "       -5.03563523e-01,  4.20854568e-01,  1.47927850e-01, -3.39067101e-01,\n",
            "        5.41748181e-02,  2.22181007e-01,  2.19911560e-01, -4.09346402e-01,\n",
            "       -3.21489036e-01,  4.57678363e-02,  2.05683574e-01, -2.13785887e-01,\n",
            "        8.03672448e-02, -1.38625622e-01,  3.07091534e-01,  1.38074026e-01,\n",
            "       -3.36122736e-02, -3.15452576e-01,  7.85413682e-02, -3.71780217e-01,\n",
            "        2.47867078e-01, -2.98162311e-01,  1.68277785e-01,  1.82686329e-01,\n",
            "        3.71075243e-01, -5.73753417e-02, -8.42674971e-02,  7.30959252e-02,\n",
            "       -6.34281337e-02,  1.81900367e-01,  2.16062367e-01,  1.33038005e-02,\n",
            "        1.49312660e-01, -4.24102843e-01,  3.45113695e-01, -3.55607033e-01,\n",
            "        4.70105976e-01,  3.08247685e-01,  2.02825829e-01, -2.37623498e-01,\n",
            "       -1.49634361e-01, -3.63600031e-02,  1.90672576e-01,  3.59411359e-01,\n",
            "       -2.44548962e-01, -1.06260762e-01,  3.44226003e-01,  2.09616512e-01,\n",
            "       -9.56423506e-02, -3.14037502e-01, -8.94561037e-02,  3.21176872e-02,\n",
            "       -6.12487733e-01, -1.09319210e-01,  3.02151442e-01, -2.03096271e-01,\n",
            "       -7.05790997e-01, -5.20231307e-01, -1.16508079e+00, -3.30994010e-01,\n",
            "        4.42250252e-01,  2.14237660e-01,  3.14716309e-01, -6.81079626e-01,\n",
            "       -1.32164866e-01, -1.83767988e-03,  4.69444990e-02,  1.08261690e-01,\n",
            "        3.92336130e-01, -9.34535190e-02, -5.43901622e-02, -5.31538606e-01,\n",
            "       -1.54907554e-01, -3.94728556e-02,  3.32884192e-02, -1.94765627e-01,\n",
            "       -1.00298878e-02, -3.25536489e-01, -1.29753947e-01, -4.37123440e-02,\n",
            "        3.38771999e-01, -4.54844654e-01, -2.19976362e-02, -5.27691722e-01,\n",
            "        1.01987742e-01, -2.63073802e-01,  1.88893706e-01,  1.72589779e-01,\n",
            "        5.85973501e-01, -2.43261862e+00,  1.82862282e-01,  6.65152818e-02,\n",
            "       -1.44877583e-01, -3.40323150e-01,  2.54487824e-02, -2.91186482e-01,\n",
            "        3.71943593e-01,  1.41969606e-01, -8.44439194e-02,  6.14494324e-01,\n",
            "       -5.11583805e-01,  3.71432118e-02, -2.57050365e-01,  5.50402403e-01,\n",
            "        1.56044438e-01,  5.32271802e-01,  9.69075188e-02,  1.09761462e-01,\n",
            "       -7.35917568e-01,  4.77819890e-01,  1.82830095e-02,  4.68242131e-02,\n",
            "        1.28801554e-01,  2.02476736e-02,  4.66802493e-02,  2.90344536e-01,\n",
            "       -2.38218531e-01, -7.95042291e-02, -1.04423419e-01, -4.20391522e-02,\n",
            "       -1.82059050e-01, -1.53535400e-02, -1.49944454e-01,  2.83682406e-01,\n",
            "       -3.14580977e-01, -5.03646851e-01,  4.38211769e-01, -9.48635489e-02,\n",
            "        1.73745990e-01, -2.75969028e-01, -1.01842478e-01, -1.27158925e-01,\n",
            "        1.01563022e-01,  2.98389137e-01, -1.25987709e-01,  2.45399415e-01,\n",
            "        8.21581334e-02, -1.17754892e-01,  3.14296216e-01, -9.99162421e-02,\n",
            "        8.09189260e-01,  6.24953151e-01,  3.37781131e-01, -1.39770821e-01,\n",
            "        5.98787516e-02,  5.76572895e-01, -5.96313000e-01, -2.81797945e-01,\n",
            "       -4.47626822e-02, -6.61079645e-01, -3.81787270e-01,  1.25395790e-01,\n",
            "        1.04698334e-02,  2.95312762e-01,  3.14571597e-02,  8.40357542e-01,\n",
            "        7.04396218e-02,  4.10676211e-01,  2.12274477e-01,  1.82728469e-01,\n",
            "        5.80106139e-01,  5.72940223e-02, -6.09313428e-01,  1.36852130e-01,\n",
            "       -8.93934369e-02,  1.60661340e-01,  3.50797772e-01,  2.64736652e-01,\n",
            "       -2.45096594e-01, -3.89990658e-01, -4.39456031e-02,  3.38168263e-01,\n",
            "       -3.49194258e-01,  4.41717431e-02,  5.52579045e-01, -2.09337205e-01,\n",
            "       -2.24020287e-01,  2.73906440e-01, -2.49883041e-01,  9.25179422e-02,\n",
            "       -9.98724252e-02, -1.62387908e-01, -6.62091672e-01, -1.00803018e+00,\n",
            "       -2.51944959e-01, -4.73586321e-01,  8.96268710e-02, -7.40932345e-01,\n",
            "       -6.42174780e-01, -3.14482093e-01,  9.09535736e-02,  2.00884230e-02,\n",
            "       -1.40559245e-02,  4.98457015e-01,  7.14692380e-03,  9.32894051e-02,\n",
            "       -3.78851593e-01, -2.81079523e-02,  1.50818691e-01, -2.09535938e-02,\n",
            "       -3.48541081e-01, -1.18708566e-01,  1.12134270e-01, -3.83488268e-01,\n",
            "        5.53007945e-02,  1.92166299e-01,  6.70798197e-02, -1.26639962e-01,\n",
            "       -1.78608894e-01,  5.76250970e-01,  3.85330200e-01,  3.37898731e-04,\n",
            "        1.24451116e-01,  2.84584552e-01,  5.16036987e-01,  8.33157897e-01,\n",
            "       -7.28120267e-01, -3.56108129e-01, -1.16879523e-01,  2.51383394e-01,\n",
            "       -3.40250075e-01,  5.21975994e-01,  1.82278365e-01, -1.02460182e+00,\n",
            "        8.27952940e-03, -2.31557831e-01, -7.00779319e-01, -3.41690034e-02,\n",
            "        1.43908188e-01,  6.11268520e-01,  1.46331295e-01,  8.59237164e-02,\n",
            "        1.57126963e-01,  2.07397088e-01,  4.41036910e-01, -1.52388394e-01,\n",
            "       -1.36148229e-01, -4.52456355e-01,  2.30963200e-01,  8.59939903e-02,\n",
            "       -1.11600962e-02, -3.21670294e-01,  7.40187526e-01,  3.93766820e-01,\n",
            "        4.80594859e-02, -5.05180120e-01, -3.06569934e-01, -2.03527600e-01,\n",
            "        4.80703086e-01, -1.11628592e-01, -3.21040094e-01,  3.34836692e-01,\n",
            "        4.18141894e-02, -2.72457063e-01,  3.24603647e-01, -3.04140329e-01,\n",
            "       -1.14651486e-01,  3.60557437e-01,  3.94381702e-01, -2.89636314e-01,\n",
            "       -2.91104019e-01, -1.41629204e-01,  2.55115658e-01,  2.41417348e-01,\n",
            "        3.99541855e-01, -5.72848581e-02,  8.16605568e-01,  1.65600644e-03,\n",
            "        2.19780266e-01,  1.87351614e-01, -2.57489771e-01, -6.50570542e-02,\n",
            "        2.64346659e-01,  2.95171320e-01,  6.76840246e-01, -3.35477114e-01,\n",
            "        4.88808267e-02, -2.57753134e-01,  1.06005326e-01, -1.83460042e-01,\n",
            "        1.68679520e-01,  2.21254915e-01,  2.46994287e-01, -2.63584971e-01,\n",
            "       -6.85541630e-01,  4.53720033e-01, -2.56195188e-01, -6.36754394e-01,\n",
            "        2.58142352e-01, -6.19020760e-02,  1.76611133e-02, -3.75476182e-01,\n",
            "       -1.03045665e-01, -1.91430092e-01,  9.12986845e-02,  2.06263423e-01,\n",
            "       -1.86035186e-01,  1.25867337e-01, -3.54060173e-01, -3.47630307e-02,\n",
            "       -1.17324054e-01,  8.11377883e-01,  1.57332763e-01, -5.94952464e-01,\n",
            "        5.78110695e-01,  1.10773459e-01, -4.35768478e-02,  3.36794347e-01,\n",
            "       -2.45634377e-01, -2.24935085e-01, -6.94446592e-03, -1.71430692e-01,\n",
            "       -2.52658390e-02, -2.67058939e-01,  1.99190646e-01, -1.14021748e-01,\n",
            "       -2.25075424e-01, -2.79811680e-01, -6.88823879e-01,  3.12174052e-01,\n",
            "        6.46335557e-02, -4.67764854e-01,  2.29302630e-01,  2.55438745e-01,\n",
            "       -2.53646314e-01,  4.21992868e-01,  2.73408175e-01, -3.00851583e-01,\n",
            "        4.03540134e-01,  2.98794031e-01,  2.52506763e-01,  3.88234824e-01,\n",
            "       -1.12688503e+01,  3.88615012e-01,  3.62697244e-01, -9.02078748e-02,\n",
            "        4.26216722e-02, -8.88126045e-02,  2.56865084e-01, -2.27343261e-01,\n",
            "        5.32155156e-01, -4.28172201e-01,  3.98852974e-02,  2.81324953e-01,\n",
            "       -2.74356663e-01, -4.28422451e-01, -7.95359135e-01,  1.61137328e-01,\n",
            "        7.79293403e-02, -5.40183544e-01, -6.25614673e-02, -4.68694568e-01,\n",
            "        1.69863552e-01, -1.67159468e-01,  3.82783860e-02,  1.04543939e-01,\n",
            "       -6.18041605e-02, -4.07841533e-01, -2.71214247e-01, -4.52308744e-01,\n",
            "       -3.46741825e-02,  4.52295035e-01, -7.26447627e-02,  1.66431367e-01,\n",
            "       -2.33230263e-01,  3.03480119e-01, -2.16532633e-01, -2.01628447e-01,\n",
            "        1.51787013e-01, -8.12518597e-01, -6.59012020e-01, -2.47401342e-01,\n",
            "       -3.30527633e-01, -1.86519727e-01, -7.27811337e-01,  4.49884415e-01,\n",
            "        5.46086058e-02,  4.07130569e-01,  1.16777635e+00,  5.37597895e-01,\n",
            "       -4.99213152e-02,  4.79333214e-02,  2.57285535e-01, -6.83716178e-01,\n",
            "       -1.07272051e-01, -5.83916426e-01, -2.86043793e-01, -1.77092642e-01,\n",
            "        5.78467607e-01, -6.34278893e-01,  1.97248861e-01, -3.83519053e-01,\n",
            "        2.82034010e-01,  2.86025614e-01,  9.81574953e-02,  2.13342279e-01,\n",
            "       -9.59215641e-01,  2.58432925e-01,  1.73451275e-01, -4.66722339e-01,\n",
            "       -2.23141700e-01, -2.35760212e-01,  1.14786610e-01,  5.87681770e-01,\n",
            "       -3.12955342e-02,  1.83311030e-01, -4.16753292e-01,  2.47663707e-01,\n",
            "       -4.85523939e-01, -1.18879415e-02,  1.32789290e+00,  3.84134233e-01,\n",
            "       -2.90985815e-02, -2.41573989e-01,  3.67962182e-01,  2.75124758e-02,\n",
            "       -3.01963389e-01, -2.56415278e-01, -8.36618990e-02, -6.96468711e-01,\n",
            "        2.21732259e-03,  1.51295960e-01, -1.74445271e-01,  1.15535185e-01,\n",
            "        1.48110807e-01,  1.73557341e-01,  3.23764920e-01,  2.71566480e-01,\n",
            "       -6.72056913e-01,  3.46561521e-01, -2.90447950e-01,  3.32921565e-01,\n",
            "        2.35135794e-01,  1.54461553e-02,  5.68464324e-02, -7.08789527e-01,\n",
            "       -3.25151265e-01, -2.09041145e-02,  1.37574822e-01,  3.74210536e-01,\n",
            "       -1.54555008e-01,  3.66992116e-01, -5.38854375e-02,  1.86658859e-01,\n",
            "        3.29651177e-01,  4.51382756e-01, -5.62321357e-02, -7.49394417e-01,\n",
            "        2.67783165e-01,  1.47215098e-01, -3.94376636e-01, -2.79878557e-01,\n",
            "       -2.58190989e-01,  2.16792926e-01,  3.57285380e-01,  1.38190448e-01,\n",
            "       -6.30804002e-01, -1.23762405e-02, -2.95182228e-01, -3.70110348e-02,\n",
            "        3.83893937e-01, -2.55405903e-02,  3.16598862e-01, -9.16864797e-02,\n",
            "       -2.04311892e-01,  1.79644316e-01, -8.33223388e-02, -2.65230775e-01,\n",
            "       -8.54677796e-01, -1.35326982e-01,  4.69424456e-01, -1.67642504e-01,\n",
            "        6.82292819e-01, -1.25058874e-01,  9.71166603e-03,  2.68101335e-01,\n",
            "       -2.21353754e-01,  2.86959767e-01,  7.78340280e-01,  2.86215007e-01,\n",
            "       -4.52309132e-01, -1.07197225e-01,  1.31957352e-01, -2.26948500e-01,\n",
            "        7.60480314e-02, -3.72481674e-01, -1.80803925e-01, -5.66940367e-01,\n",
            "        2.51910627e-01, -1.39659026e-03, -3.82681340e-01,  8.42666440e-03,\n",
            "        6.49640709e-02, -2.81425178e-01,  7.10232630e-02, -4.07731861e-01,\n",
            "        3.19446295e-01,  3.24393600e-01, -9.99305695e-02, -5.72571993e-01,\n",
            "       -6.28803670e-01, -5.17447948e-01,  4.79923300e-02,  4.64969724e-01,\n",
            "        3.27698067e-02,  1.17419567e-02, -7.00369716e-01,  1.49664491e-01,\n",
            "       -3.14154446e-01,  4.85997438e-01, -4.07342553e-01, -3.13196123e-01,\n",
            "        2.82778144e-01,  3.62967730e-01,  2.23224610e-01, -3.88656229e-01,\n",
            "        2.46561527e-01,  5.51339149e-01, -3.89403343e-01, -4.15904760e-01,\n",
            "       -1.13069855e-01, -1.24548875e-01,  4.62470174e-01,  3.71846527e-01,\n",
            "        4.48216379e-01,  2.41546743e-02,  5.46407513e-02,  2.76334465e-01,\n",
            "       -3.59173268e-01,  1.53594077e-01,  2.20511794e-01,  3.21454406e-01,\n",
            "        5.62974572e-01,  2.72907734e-01,  2.55285323e-01, -4.09475788e-02,\n",
            "       -1.15875840e-01, -3.24521840e-01,  4.29270744e-01,  1.56975639e+00,\n",
            "       -2.95319378e-01, -1.81350380e-01,  5.69915235e-01, -3.81281495e-01,\n",
            "       -4.12242234e-01, -1.47973061e-01,  2.54707068e-01, -4.78660285e-01,\n",
            "        5.00026107e-01,  1.82093188e-01, -2.83958763e-02, -4.65564393e-02,\n",
            "        6.86480880e-01,  2.67647207e-01, -3.83297890e-01, -5.97820505e-02,\n",
            "       -3.31074119e-01, -2.02300295e-01, -3.52165997e-01,  4.29352850e-01,\n",
            "        1.43927127e-01,  1.21388435e-01, -4.42560911e-01, -6.70706630e-01,\n",
            "        4.70877588e-01,  1.67607814e-01,  9.96985793e-01,  4.48223129e-02,\n",
            "       -2.99651027e-02, -6.39593154e-02,  2.58407176e-01, -5.35358131e-01,\n",
            "        1.67135775e-01,  4.93452251e-01, -3.59504223e-01,  2.13919282e-01,\n",
            "        4.75374281e-01,  2.19284862e-01,  2.67444342e-01,  5.21576405e-02,\n",
            "       -1.50017634e-01,  7.28977025e-02, -8.57912302e-02, -4.18354958e-01,\n",
            "        2.67015725e-01, -5.37672520e-01, -3.53312492e-02, -2.45248037e-03,\n",
            "        6.44669294e-01, -2.66426325e-01, -2.43826389e-01, -2.68845171e-01,\n",
            "       -1.19432412e-01, -1.67642698e-01, -1.62583172e-01, -3.32867235e-01,\n",
            "        2.16924533e-01,  2.19849557e-01, -1.59479231e-01, -4.37839717e-01,\n",
            "        3.17537308e-01,  7.25957096e-01, -5.09643316e-01, -6.07584268e-02,\n",
            "        8.53196979e-02,  3.94948304e-01, -2.98745573e-01, -1.54074654e-01,\n",
            "        4.24329862e-02,  3.04054976e-01, -1.37847513e-01,  9.03059423e-01,\n",
            "       -3.35474432e-01, -3.48478794e-01, -7.58310407e-02,  5.12762219e-02,\n",
            "       -4.93691504e-01,  2.83424646e-01,  4.42042351e-02,  3.42194796e-01,\n",
            "       -5.12779474e-01, -1.07560337e-01,  5.36146641e-01,  2.38867432e-01,\n",
            "        3.77621233e-01, -4.66068238e-01,  9.15605053e-02,  2.41342872e-01,\n",
            "        1.64526269e-01,  4.71817225e-01,  3.01832706e-02,  5.82281351e-01,\n",
            "       -5.36589324e-02, -1.35839328e-01, -4.19729918e-01,  2.28401646e-01,\n",
            "        3.82038683e-01,  1.50760338e-01, -4.89935040e-01, -4.30617273e-01,\n",
            "        4.53794539e-01, -3.95365298e-01,  5.08215785e-01,  8.65175128e-01,\n",
            "       -3.70003730e-02, -2.43169993e-01, -4.70734298e-01, -3.06810319e-01,\n",
            "       -1.70804471e-01,  2.32175495e-02,  1.01247035e-01,  1.21906154e-01,\n",
            "        6.53688163e-02, -3.39381635e-01,  1.44651681e-01,  2.75286376e-01,\n",
            "        2.38031268e-01, -2.32518718e-01,  3.96796227e-01, -2.68934309e-01,\n",
            "        2.98427641e-01,  8.61676633e-02, -6.24180362e-02,  1.32171601e-01,\n",
            "        7.25146949e-01,  3.69049966e-01, -4.43322420e-01,  2.29186982e-01,\n",
            "       -3.92228737e-02, -3.74853700e-01, -3.35213661e-01, -8.12594831e-01,\n",
            "        4.65530872e-01, -9.67144594e-02,  5.73383451e-01,  7.56207645e-01],\n",
            "      dtype=float32), 'History': array([-2.75918663e-01, -4.32962060e-01, -5.62820911e-01, -4.97414231e-01,\n",
            "       -1.26394600e-01, -2.02250406e-01, -4.23404843e-01,  4.47860777e-01,\n",
            "       -8.99241269e-02,  1.46215186e-01,  2.79626071e-01,  3.25835824e-01,\n",
            "        1.11530714e-01, -3.66519928e-01, -7.04444706e-01,  7.37811998e-03,\n",
            "        4.21085298e-01,  3.29962730e-01,  1.24328256e-01,  4.13265735e-01,\n",
            "        5.47275543e-01,  2.39001691e-01, -2.02738702e-01, -8.54978338e-04,\n",
            "       -1.90521955e-01,  1.91930830e-02,  1.14066854e-01, -5.34333102e-02,\n",
            "        4.49873149e-01, -3.54231060e-01,  1.15781426e-01,  2.99415857e-01,\n",
            "       -2.43079126e-01,  8.95815969e-01,  3.55760641e-02, -3.29481483e-01,\n",
            "        1.16086774e-01,  4.57904190e-02,  6.97621882e-01, -1.08289674e-01,\n",
            "        1.81090668e-01,  2.23860089e-02,  3.27208519e-01, -7.80151784e-02,\n",
            "        2.21816450e-02,  1.40871435e-01,  5.61643578e-02, -5.66294789e-01,\n",
            "       -2.10649520e-03, -4.38927799e-01, -1.25229552e-01, -3.40381622e-01,\n",
            "        1.30646646e-01, -9.84131172e-02, -3.48580599e-01, -2.32923433e-01,\n",
            "       -6.19459264e-02,  2.11995423e-01, -3.49882454e-01,  3.78901005e-01,\n",
            "       -2.93337554e-01,  5.27764082e-01, -6.94965124e-02, -2.47040167e-01,\n",
            "        3.92664224e-03, -5.20968847e-02, -3.44400316e-01, -2.50562042e-01,\n",
            "       -3.49182159e-01,  2.90704500e-02,  3.23364809e-02,  5.91725111e-05,\n",
            "        5.30351773e-02, -4.76262718e-03,  1.51930720e-01, -2.96315670e-01,\n",
            "        2.05164388e-01, -5.77631742e-02, -1.47324026e-01, -3.23409289e-01,\n",
            "        3.30949605e-01,  3.99507403e-01, -1.59017846e-01, -5.47060110e-02,\n",
            "       -3.20566952e-01, -1.98770419e-01,  3.12967211e-01,  9.94841382e-02,\n",
            "       -5.16244024e-02, -4.47349474e-02,  3.02698344e-01,  1.48345530e-01,\n",
            "       -1.13656625e-01, -2.53232062e-01,  1.40633419e-01, -1.92382842e-01,\n",
            "       -2.11416129e-02, -2.26900935e-01,  2.18778908e-01, -2.47278169e-01,\n",
            "       -8.00257400e-02, -2.71902531e-02,  4.08308774e-01, -1.55873984e-01,\n",
            "       -1.40898108e-01, -2.47644171e-01,  2.28347927e-01, -2.31551547e-02,\n",
            "       -5.48226655e-01,  1.09562777e-01, -7.07451403e-02, -2.65926868e-01,\n",
            "        3.49017322e-01, -2.45094940e-01, -7.80954584e-02,  2.10156620e-01,\n",
            "        3.93724740e-01, -4.88624983e-02, -2.37054795e-01,  3.07114124e-02,\n",
            "        6.99179322e-02, -2.34221488e-01,  4.50499691e-02,  1.62562616e-02,\n",
            "        3.30739558e-01, -2.34939963e-01,  5.54688692e-01, -1.38077646e-01,\n",
            "        1.39998451e-01,  4.65905815e-01,  2.57085636e-02,  5.17022386e-02,\n",
            "       -3.58528554e-01, -3.39864850e-01, -1.91238988e-02,  1.05803050e-01,\n",
            "       -9.94897336e-02, -3.24698299e-01,  5.68431951e-02,  1.13082632e-01,\n",
            "       -1.49586231e-01, -9.70075279e-02, -2.05451667e-01, -3.34321588e-01,\n",
            "       -2.63817728e-01, -2.84063756e-01,  2.60943145e-01, -1.53072938e-01,\n",
            "       -3.51691544e-01, -1.81915566e-01, -7.37165749e-01, -5.55601120e-01,\n",
            "        6.49225235e-01,  2.77142256e-01,  9.98721570e-02, -3.23028564e-01,\n",
            "        1.00609049e-01,  1.08996786e-01,  1.67239502e-01,  2.63412058e-01,\n",
            "        5.10330081e-01, -1.83420539e-01,  3.26205432e-01, -1.38261110e-01,\n",
            "        7.03474879e-02, -7.80873979e-03, -3.06223154e-01, -1.25634655e-01,\n",
            "        1.07570909e-01,  9.07508209e-02,  1.59760535e-01, -2.57111818e-01,\n",
            "       -9.39959288e-02, -4.44352776e-01,  2.83096805e-02, -3.09337676e-01,\n",
            "        1.99206561e-01, -2.34715119e-02, -1.93010360e-01,  2.75302082e-02,\n",
            "        9.69518796e-02, -2.18225121e+00,  3.00696403e-01, -2.20752269e-01,\n",
            "       -6.60679340e-02, -5.47793388e-01,  7.65621439e-02, -2.04645842e-02,\n",
            "        2.43238539e-01,  3.75333801e-03, -1.59637898e-01,  8.43078643e-02,\n",
            "       -7.11398348e-02, -7.87883997e-04, -4.15620565e-01,  2.96460271e-01,\n",
            "        3.32038879e-01,  2.48056531e-01,  1.82125896e-01,  7.93183073e-02,\n",
            "       -2.42312431e-01,  3.35781723e-01, -2.57380307e-03, -2.62484644e-02,\n",
            "       -5.80437258e-02,  2.54673749e-01, -6.23609386e-02,  1.84329510e-01,\n",
            "        3.61170620e-01,  2.44592488e-01,  1.07773319e-01, -2.47794136e-01,\n",
            "       -4.32933688e-01, -2.97537416e-01,  9.47697237e-02,  3.05901468e-01,\n",
            "       -2.62926787e-01, -2.71455437e-01,  5.06968737e-01, -1.99683458e-02,\n",
            "        1.14560001e-01, -4.70775440e-02, -5.81321791e-02, -2.69618839e-01,\n",
            "        1.25693783e-01,  3.70099008e-01, -1.09378375e-01,  1.64893463e-01,\n",
            "       -3.48838158e-02, -1.98086590e-01,  4.34878729e-02, -3.39601934e-01,\n",
            "        5.29612005e-01,  2.19508797e-01,  3.75539139e-02, -5.24593033e-02,\n",
            "       -1.18102953e-01,  4.75230575e-01,  1.86038017e-02,  1.97369918e-01,\n",
            "       -2.33621791e-01, -3.49177331e-01, -4.98595059e-01,  2.88687855e-01,\n",
            "        2.46008396e-01,  1.23749882e-01, -1.50927097e-01,  3.63298625e-01,\n",
            "        5.85648954e-01,  2.72715688e-01,  1.54278576e-01,  5.77605367e-02,\n",
            "        3.57806951e-01,  6.04358554e-01, -2.96082109e-01,  7.79068191e-03,\n",
            "        2.75118738e-01, -2.74977386e-01,  5.25608286e-03, -6.67557195e-02,\n",
            "       -2.69430608e-01,  6.07509166e-04,  3.00490618e-01,  3.47459793e-01,\n",
            "       -2.55929708e-01,  2.79511988e-01,  4.84148979e-01, -1.66096330e-01,\n",
            "       -2.11414754e-01,  1.77191719e-01, -3.79847586e-01,  1.67095274e-01,\n",
            "       -2.09102735e-01, -2.79184371e-01, -5.90854585e-01, -6.05582356e-01,\n",
            "       -2.04820916e-01,  1.00609325e-02,  3.56318176e-01, -3.82639825e-01,\n",
            "       -2.15983957e-01, -4.04368550e-01,  1.20338805e-01, -2.46961445e-01,\n",
            "       -2.59302914e-01,  2.86250561e-03, -1.17148280e-01,  1.08068556e-01,\n",
            "       -4.48394895e-01, -3.88504773e-01, -1.05085745e-01,  1.03364602e-01,\n",
            "       -3.85302633e-01,  4.07022908e-02, -8.68644863e-02,  1.24260955e-01,\n",
            "        7.32327178e-02,  2.12967973e-02, -9.35449675e-02, -1.94130063e-01,\n",
            "       -1.34665057e-01,  1.27371341e-01,  2.10936628e-02, -1.19752735e-01,\n",
            "        1.66157752e-01,  1.29909515e-01,  1.85035482e-01,  3.72405827e-01,\n",
            "       -2.60314405e-01, -4.80191886e-01, -2.51900047e-01,  6.05071902e-01,\n",
            "        6.30801991e-02,  6.71991259e-02, -7.88425431e-02, -7.91617036e-01,\n",
            "       -4.84307930e-02, -8.68869573e-02, -3.00218314e-01, -1.51034057e-01,\n",
            "        3.57934922e-01,  2.19951764e-01, -2.26230964e-01, -1.38020635e-01,\n",
            "       -8.60039145e-02,  1.78267419e-01,  3.57560813e-01, -1.95785999e-01,\n",
            "        1.04428805e-01, -3.20942670e-01,  5.99240735e-02,  3.73155139e-02,\n",
            "       -9.76041928e-02, -2.36338079e-01,  3.68124783e-01,  2.93257058e-01,\n",
            "        2.01403946e-01,  1.11909993e-01,  1.44875750e-01, -1.98575690e-01,\n",
            "        6.05457306e-01, -5.15341684e-02, -6.91190660e-02,  4.79286075e-01,\n",
            "        1.23798307e-02,  1.63554221e-01,  4.87633646e-01, -1.87542871e-01,\n",
            "       -5.56372255e-02,  2.02528298e-01,  1.39067069e-01, -2.91511416e-03,\n",
            "       -2.91005045e-01, -4.54497412e-02,  5.21909952e-01,  5.64604044e-01,\n",
            "        2.47139215e-01,  1.01143055e-01,  3.48533690e-01,  3.63887310e-01,\n",
            "       -4.44324106e-01,  1.78214192e-01, -1.47006497e-01, -8.02582279e-02,\n",
            "        3.03420007e-01,  1.62816763e-01,  2.97711253e-01, -3.94347191e-01,\n",
            "       -1.66535914e-01,  1.78408682e-01,  2.06555143e-01, -1.01819634e-03,\n",
            "       -5.85938245e-03,  1.67660192e-01,  4.50536877e-01, -4.11542833e-01,\n",
            "       -4.39081937e-01, -4.57525402e-02, -3.09820205e-01, -2.64625192e-01,\n",
            "       -1.73535012e-02, -6.03687949e-02,  3.61380950e-02, -3.96620810e-01,\n",
            "       -1.15891203e-01, -2.44982421e-01,  1.73511729e-02,  1.87966809e-01,\n",
            "        8.19262415e-02,  2.81987861e-02,  1.35686308e-01, -3.82886156e-02,\n",
            "       -1.40746266e-01,  3.82047445e-01,  2.24043891e-01, -2.75499195e-01,\n",
            "        1.60152435e-01, -2.05801830e-01,  1.53618798e-01,  9.00667012e-02,\n",
            "       -1.69424146e-01, -3.26376230e-01, -1.88891664e-01, -1.15530506e-01,\n",
            "        3.29814821e-01, -2.18129367e-01, -1.26854423e-02,  3.86976451e-02,\n",
            "       -1.30239934e-01, -2.04344019e-01, -1.84131116e-01,  3.48755270e-01,\n",
            "       -2.35855281e-01, -1.18927851e-01, -1.26908675e-01,  2.10536599e-01,\n",
            "       -2.61207104e-01,  1.24800086e-01,  5.98668940e-02, -1.31512821e-01,\n",
            "        4.08328980e-01,  1.72003582e-01, -9.83459353e-02,  3.88577700e-01,\n",
            "       -1.17429314e+01,  2.94883430e-01,  6.13969445e-01, -8.31695274e-02,\n",
            "        4.54630628e-02, -5.40033132e-02,  2.15357587e-01, -1.19762510e-01,\n",
            "        2.57931054e-01, -4.98760372e-01, -9.46867019e-02,  7.81283900e-02,\n",
            "       -1.13336638e-01, -9.02556181e-02, -7.14830279e-01, -5.54479286e-02,\n",
            "        1.73141778e-01, -6.73277915e-01,  1.59747198e-01,  4.90052998e-03,\n",
            "        2.65507996e-01, -3.85320485e-01, -1.60572112e-01,  2.87233740e-01,\n",
            "       -3.73569399e-01, -1.89437866e-01, -3.86817843e-01, -4.98442650e-02,\n",
            "        3.52882221e-02,  2.72139043e-01, -1.45993903e-01,  7.63582960e-02,\n",
            "       -4.07941416e-02,  2.74676561e-01, -2.25898236e-01, -3.56608331e-01,\n",
            "       -2.18926370e-03, -1.73441827e-01, -2.21761703e-01, -1.79940000e-01,\n",
            "       -1.32086530e-01,  2.26228148e-01, -2.65512943e-01,  2.47158274e-01,\n",
            "        1.45148158e-01,  5.51359534e-01,  6.09398186e-01,  8.05412889e-01,\n",
            "       -1.20436259e-01, -6.85662106e-02, -2.86001414e-02, -6.73984170e-01,\n",
            "       -6.44723698e-02, -4.59293187e-01, -8.89232606e-02, -1.53242320e-01,\n",
            "        4.27117586e-01, -5.18885911e-01,  3.26148659e-01, -1.92106739e-02,\n",
            "        5.11320680e-02,  2.42003739e-01,  6.46554679e-02,  1.32657200e-01,\n",
            "       -3.97246242e-01,  1.05777785e-01,  3.66813779e-01,  1.30991340e-01,\n",
            "       -1.09781921e-02, -5.20703435e-01,  1.20432734e-01, -2.59815976e-02,\n",
            "        3.51273000e-01,  1.53994769e-01, -2.36967653e-01,  3.16623412e-02,\n",
            "       -2.90439308e-01, -2.47962132e-01,  1.10268438e+00,  1.99704707e-01,\n",
            "        1.27208725e-01,  1.47404909e-01,  8.14192742e-03, -3.51728976e-01,\n",
            "       -4.02094126e-01, -1.73223704e-01, -3.58104333e-02, -1.49788275e-01,\n",
            "        7.95095488e-02,  1.48867175e-01, -3.95595342e-01, -1.60156786e-01,\n",
            "       -1.65961295e-01,  1.91091001e-01,  2.29632080e-01,  5.06217897e-01,\n",
            "       -4.12250340e-01,  1.81368217e-01,  4.23113927e-02,  2.67515004e-01,\n",
            "        3.11130106e-01, -1.17267162e-01, -2.07957700e-01, -4.32281077e-01,\n",
            "        8.99012238e-02, -3.27213824e-01,  7.32344389e-02,  3.01169604e-01,\n",
            "        2.28087530e-01,  1.74024269e-01, -1.23599254e-01,  7.77390227e-02,\n",
            "        4.10142094e-01,  1.93978712e-01, -2.72105962e-01, -5.14596581e-01,\n",
            "        2.65482128e-01,  2.87032545e-01, -3.86544585e-01, -3.64298999e-01,\n",
            "       -2.56015837e-01, -3.63096520e-02,  5.48122048e-01,  8.96988250e-03,\n",
            "       -3.80248278e-01,  2.66444951e-01, -5.95460534e-01,  6.03562593e-02,\n",
            "        3.49256322e-02,  2.57723570e-01,  5.68138719e-01,  3.85952592e-01,\n",
            "       -1.62588745e-01, -9.42189246e-03, -2.41367817e-01, -3.25813204e-01,\n",
            "       -9.00309801e-01,  6.14687651e-02,  4.43821281e-01, -3.56508017e-01,\n",
            "        1.91390291e-01,  7.99737312e-03,  1.24520406e-01, -8.47590044e-02,\n",
            "        3.39517862e-01,  1.81086913e-01,  1.45049691e-01,  2.08418757e-01,\n",
            "       -1.31760448e-01,  4.04489711e-02, -9.85371321e-02, -1.96808055e-01,\n",
            "       -8.66386145e-02, -1.79832056e-01,  4.50272299e-02, -4.59971130e-01,\n",
            "       -4.40414548e-01,  1.09577686e-01, -5.29343784e-02, -1.10479206e-01,\n",
            "        3.01305261e-02,  1.79177076e-01, -5.31240553e-02, -1.70918301e-01,\n",
            "        5.46046674e-01,  8.26269761e-02,  4.77638021e-02, -3.50872397e-01,\n",
            "        1.51897930e-02,  3.86416093e-02,  9.08963680e-02,  2.13238746e-01,\n",
            "        1.91832811e-01, -4.82147932e-01, -5.07071137e-01,  1.22665472e-01,\n",
            "       -1.12530425e-01,  9.64233994e-01, -4.42283690e-01, -2.53345311e-01,\n",
            "        8.25033560e-02,  2.87552416e-01, -1.26689717e-01, -2.36583516e-01,\n",
            "        1.48429275e-02,  1.88941896e-01, -2.05655068e-01, -2.29966015e-01,\n",
            "        5.23531586e-02, -1.04535379e-01,  3.15263182e-01,  2.94449687e-01,\n",
            "        2.17143014e-01, -1.72996908e-01,  4.03838128e-01, -2.28352666e-01,\n",
            "       -4.01512653e-01,  6.06707595e-02, -2.37926617e-01, -1.54142138e-02,\n",
            "        5.26601553e-01, -3.36522460e-01,  8.25971812e-02, -1.02842361e-01,\n",
            "        1.43458113e-01,  1.51141450e-01,  3.04247499e-01,  1.35851860e+00,\n",
            "       -9.47680175e-02, -1.32146776e-02,  6.86998129e-01,  9.52135921e-02,\n",
            "       -2.35814348e-01,  1.21137142e-01,  7.05884844e-02,  1.35266185e-01,\n",
            "        4.37002897e-01, -3.14917028e-01, -9.32934582e-02, -1.65435672e-01,\n",
            "        2.56309211e-01,  2.25104392e-01, -3.98956746e-01, -4.24930066e-01,\n",
            "       -9.61482972e-02,  4.18139696e-01, -1.34194866e-01,  3.81267726e-01,\n",
            "        1.02850907e-01,  2.71844923e-01,  3.06010246e-01, -5.62563360e-01,\n",
            "        1.48064092e-01, -1.26503766e-01,  1.77884966e-01, -2.27515042e-01,\n",
            "       -3.64820898e-01, -2.68027365e-01, -2.11739331e-01, -2.69342899e-01,\n",
            "        1.28390014e-01,  4.49981153e-01, -5.35806537e-01,  1.84146073e-02,\n",
            "        4.69357371e-01, -2.85202324e-01,  1.49704009e-01,  3.21722254e-02,\n",
            "        1.47787273e-01,  1.07798144e-01, -1.27620578e-01, -3.44754457e-01,\n",
            "        1.89061463e-01, -2.53981799e-01, -5.85400939e-01,  1.79393291e-01,\n",
            "        4.28246111e-01, -1.06921181e-01,  8.80989358e-02, -4.24233526e-02,\n",
            "       -2.81929851e-01, -1.27068788e-01, -2.42634565e-01, -2.15478137e-01,\n",
            "       -1.18565068e-01,  1.06594831e-01, -9.99171287e-02, -3.20822954e-01,\n",
            "        1.23883687e-01,  4.17958200e-02,  3.22868526e-02,  1.86424017e-01,\n",
            "        2.72299707e-01,  9.71236885e-01, -6.48612231e-02, -3.56335193e-02,\n",
            "        1.98192999e-01,  3.28334212e-01, -9.56673399e-02,  4.64336991e-01,\n",
            "       -2.58813262e-01, -3.79687369e-01, -3.96808498e-02,  4.26376313e-02,\n",
            "       -2.44839609e-01,  1.37584031e-01, -1.93182901e-01,  8.01440775e-02,\n",
            "       -2.20809817e-01, -2.54775584e-01,  2.73624003e-01,  1.14708796e-01,\n",
            "        2.17331901e-01, -4.55659598e-01, -1.27350613e-02,  2.19848379e-01,\n",
            "        1.13620743e-01,  1.96874633e-01,  1.74003035e-01,  3.86286587e-01,\n",
            "       -1.76266469e-02,  5.37178293e-02, -4.94191945e-01, -1.34659126e-01,\n",
            "       -1.54113039e-01, -3.78444046e-03, -2.27510214e-01, -2.19559401e-01,\n",
            "        3.00528586e-01,  1.95431113e-01,  3.99287969e-01,  5.83518028e-01,\n",
            "       -2.00482845e-01, -3.79621029e-01, -3.22586715e-01, -5.44739515e-02,\n",
            "       -2.39838973e-01, -1.04224101e-01,  1.49716288e-01,  3.78865227e-02,\n",
            "       -2.06163853e-01, -5.47852039e-01, -3.42740446e-01, -1.65868536e-01,\n",
            "        9.62124765e-02, -2.11969823e-01,  4.75783914e-01, -1.31154835e-01,\n",
            "        2.38561034e-01,  3.25666904e-01,  2.04278588e-01,  5.08384779e-03,\n",
            "        4.47166562e-01,  8.86487216e-03, -3.40243131e-01, -3.73548269e-03,\n",
            "        1.18475497e-01, -2.49347568e-01, -2.44519562e-02, -6.08928263e-01,\n",
            "        1.86957330e-01, -8.51275474e-02,  8.23064506e-01,  2.86968708e-01],\n",
            "      dtype=float32), 'Family_history': array([-2.41666287e-01, -5.43418288e-01, -4.94711727e-01, -2.74584085e-01,\n",
            "       -1.62794560e-01, -4.61036742e-01, -3.63527954e-01, -1.68170035e-01,\n",
            "       -3.89445275e-01,  4.82321441e-01,  8.94053876e-02,  5.23825407e-01,\n",
            "       -8.40402469e-02, -4.31411117e-01, -4.98562813e-01, -3.24736208e-01,\n",
            "       -6.43189177e-02,  4.99854088e-01,  1.44165024e-01,  6.21579826e-01,\n",
            "        3.25320691e-01,  5.88249862e-01, -2.86492795e-01, -1.89069599e-01,\n",
            "       -2.89730936e-01,  1.63268551e-01, -2.22781658e-01, -9.69650075e-02,\n",
            "        6.11778021e-01, -4.29924488e-01,  3.46790761e-01,  4.52146977e-01,\n",
            "       -5.01861095e-01,  1.16609502e+00, -1.01574570e-01, -3.69450271e-01,\n",
            "       -1.76031590e-01, -8.33849311e-02,  2.79191464e-01, -5.68445139e-02,\n",
            "        4.26844299e-01, -1.05321042e-01, -2.72995442e-01, -9.87089813e-01,\n",
            "       -3.59434783e-02,  5.20068586e-01, -1.43819064e-01, -5.22939682e-01,\n",
            "        6.20821975e-02, -4.07968611e-02, -2.02508435e-01, -2.11798593e-01,\n",
            "       -1.69234425e-02,  1.92302406e-01, -6.64154470e-01, -3.14393014e-01,\n",
            "       -3.31623629e-02,  1.55721933e-01, -5.32029271e-01,  6.62918746e-01,\n",
            "       -2.84103513e-01,  6.91476703e-01, -1.25162184e-01, -2.71953821e-01,\n",
            "        3.80881876e-01, -3.77551019e-01, -5.79452872e-01, -2.13609040e-01,\n",
            "       -8.03607032e-02,  1.25288755e-01,  5.41177988e-02,  6.75845981e-01,\n",
            "        2.17958093e-01,  3.62196892e-01, -2.62553501e-03, -5.24334550e-01,\n",
            "       -1.37327552e-01, -1.77899957e-01, -7.32171983e-02, -9.16285962e-02,\n",
            "        7.38534093e-01,  4.61461365e-01, -2.16046184e-01, -4.63804573e-01,\n",
            "       -1.59660786e-01, -3.27643484e-01,  4.68148470e-01,  1.60369530e-01,\n",
            "        1.31208912e-01,  2.67365575e-01,  3.61742854e-01, -9.05978307e-02,\n",
            "        1.52971208e-01, -4.20770645e-01,  3.80074024e-01, -2.43877530e-01,\n",
            "        6.76993793e-03, -1.50893390e-01, -2.12024413e-02, -2.53335476e-01,\n",
            "       -6.67406023e-01, -1.54857263e-01,  2.16324985e-01, -1.10239789e-01,\n",
            "       -2.94348270e-01, -3.13613474e-01,  1.59674600e-01,  4.44485918e-02,\n",
            "       -2.46254548e-01, -2.51405081e-03, -1.13540791e-01,  1.42830923e-01,\n",
            "        1.92239076e-01, -4.81768489e-01,  1.63125306e-01,  7.42015302e-01,\n",
            "        4.06832784e-01, -2.45702505e-01, -2.75977515e-02, -9.66555327e-02,\n",
            "        2.09122032e-01, -3.94912094e-01, -5.93804680e-02,  9.51812714e-02,\n",
            "        2.01855868e-01, -2.30785057e-01,  3.13241214e-01, -6.38144612e-01,\n",
            "        1.65692787e-03,  6.74240351e-01,  2.60029525e-01, -3.57680649e-01,\n",
            "        9.33031663e-02, -2.05410039e-03, -6.36123195e-02,  7.58189857e-02,\n",
            "       -9.68693495e-02, -5.03626019e-02, -3.05848479e-01,  5.00364959e-01,\n",
            "       -2.16821760e-01, -6.23767376e-01, -1.28781721e-01, -3.04120272e-01,\n",
            "        1.53453156e-01,  5.89608848e-02, -1.26408249e-01, -3.40431541e-01,\n",
            "       -1.46875739e-01, -2.02232271e-01, -5.76455951e-01,  3.29148099e-02,\n",
            "        4.86601204e-01, -2.51791924e-01,  2.90808290e-01, -3.42299402e-01,\n",
            "        1.54224068e-01, -1.11882553e-01, -1.57329112e-01, -7.17013096e-03,\n",
            "        7.96968162e-01, -5.38939655e-01,  4.78643596e-01, -2.50431538e-01,\n",
            "        6.28566980e-01,  1.83911264e-01,  1.22039691e-01, -1.88175768e-01,\n",
            "        1.90434501e-01,  1.68887407e-01,  6.14707582e-02, -5.44245481e-01,\n",
            "        1.87624738e-01, -1.44074887e-01,  4.58229845e-03, -8.20946842e-02,\n",
            "        4.13250864e-01, -1.67070433e-01, -3.72056961e-02, -9.65429097e-02,\n",
            "        3.31213653e-01, -2.28996229e+00, -1.10521823e-01,  1.90210342e-01,\n",
            "       -1.59743592e-01, -5.96798778e-01,  2.08151922e-01, -1.53843630e-02,\n",
            "        1.12645969e-01,  4.84268665e-01, -5.38807869e-01,  2.85360925e-02,\n",
            "       -2.97510445e-01, -4.10905212e-01, -3.95740360e-01,  4.60850596e-01,\n",
            "        2.15265542e-01,  3.34412903e-01,  1.37940675e-01, -3.57205234e-02,\n",
            "        1.83910161e-01,  1.45929813e-01, -2.76182890e-01, -3.30118537e-01,\n",
            "       -3.50604475e-01, -1.20255705e-02,  2.76544869e-01,  2.94364184e-01,\n",
            "       -1.33728564e-01,  1.03061736e-01, -1.96582638e-02, -1.18056312e-01,\n",
            "       -4.44844246e-01, -2.97989901e-02,  3.62648129e-01,  8.01476717e-01,\n",
            "       -4.64907944e-01, -3.10024004e-02,  3.79195303e-01, -8.05416107e-02,\n",
            "        5.98904639e-02, -9.93167385e-02, -1.93755791e-01, -1.37534946e-01,\n",
            "       -2.43298516e-01,  7.05929935e-01, -2.84898460e-01, -1.65854484e-01,\n",
            "        6.49743378e-02, -1.92711234e-01,  4.02612388e-01, -6.67594075e-01,\n",
            "        2.49257803e-01,  3.34455341e-01, -3.17676887e-02, -3.43367189e-01,\n",
            "       -1.41210584e-02,  3.53031337e-01,  3.85874927e-01, -6.56255856e-02,\n",
            "       -3.57137859e-01, -3.92595857e-01, -1.31721646e-01,  2.45808780e-01,\n",
            "        1.63515180e-01,  3.04421693e-01,  6.23738840e-02,  2.66980287e-02,\n",
            "        4.83373813e-02,  2.08598375e-01,  3.15002680e-01,  4.12183940e-01,\n",
            "        3.03453147e-01,  1.97310612e-01, -3.88936877e-01,  4.05255795e-01,\n",
            "        2.52031058e-01, -4.06146526e-01,  2.15607509e-01,  5.00309430e-02,\n",
            "        2.43074760e-01,  2.58055538e-01, -5.00575267e-02,  3.27588081e-01,\n",
            "       -4.31176782e-01, -3.70550394e-01,  6.37483835e-01, -1.04715608e-01,\n",
            "       -3.57613027e-01,  1.82843983e-01, -1.68929368e-01,  4.11362022e-01,\n",
            "       -1.22394145e-01, -8.89605880e-02, -4.50604200e-01, -5.13739347e-01,\n",
            "       -1.39305919e-01,  3.58095497e-01, -4.21011746e-02, -6.01390541e-01,\n",
            "        1.98008224e-01, -5.62743783e-01, -2.64829397e-01, -3.31106968e-02,\n",
            "        5.47541305e-02,  2.04660386e-01, -7.21877366e-02,  1.73336007e-02,\n",
            "       -5.45121372e-01, -7.06119388e-02, -4.51138556e-01,  4.39647317e-01,\n",
            "       -1.77656591e-01,  2.46342570e-01,  2.22002357e-01,  1.67290434e-01,\n",
            "        2.28110671e-01,  2.53414273e-01, -6.15655541e-01, -5.26021719e-01,\n",
            "        1.62498891e-01,  3.25137466e-01,  2.16337204e-01, -1.06534645e-01,\n",
            "       -1.67755317e-02, -7.11970776e-02, -8.01673383e-02,  3.28368962e-01,\n",
            "        3.42766717e-02, -8.93618017e-02, -3.77377719e-01,  5.17172754e-01,\n",
            "       -1.99608188e-02, -5.20389639e-02,  2.78191924e-01, -5.67201138e-01,\n",
            "       -1.18426085e-01,  5.11368364e-02,  1.59429491e-01, -2.52068847e-01,\n",
            "        1.91252341e-03,  2.72970200e-01, -5.41323572e-02, -2.93427464e-02,\n",
            "        1.84764236e-01,  4.95433807e-01,  5.78230858e-01, -5.39460897e-01,\n",
            "        9.60849077e-02,  5.23080602e-02,  3.29442233e-01, -1.15914814e-01,\n",
            "       -1.11577027e-01, -8.06384906e-02,  3.82458538e-01,  3.72573227e-01,\n",
            "        2.54651517e-01,  1.69838756e-01, -5.85157089e-02,  2.41139442e-01,\n",
            "        2.27055907e-01,  1.15736477e-01,  1.97659343e-01,  3.87515873e-01,\n",
            "       -1.84295341e-01,  1.23817146e-01,  2.13085875e-01, -7.40063667e-01,\n",
            "       -2.48971626e-01,  2.38572717e-01,  1.53664157e-01,  2.16101706e-01,\n",
            "       -2.43548959e-01, -7.50298873e-02,  4.93687719e-01,  3.12416852e-01,\n",
            "        7.12058097e-02, -3.91494520e-02,  2.82009661e-01,  4.63577121e-01,\n",
            "       -5.73242128e-01,  1.43420398e-01,  5.20425588e-02, -2.98942924e-01,\n",
            "       -1.47996187e-01,  1.68058723e-02,  1.40461251e-01, -1.23098180e-01,\n",
            "       -1.40927434e-02, -1.87368780e-01,  2.56659627e-01,  3.01252306e-01,\n",
            "       -1.04794456e-02,  1.51141435e-01,  5.31435728e-01, -2.15053171e-01,\n",
            "       -1.34176433e-01,  2.70573378e-01, -3.08848768e-01, -3.02197397e-01,\n",
            "        1.21133566e-01,  1.54556289e-01,  1.06590383e-01, -2.86247998e-01,\n",
            "        3.43742850e-03, -3.31213206e-01,  1.14360675e-01,  7.37043396e-02,\n",
            "        3.37307230e-02,  1.49813756e-01,  1.76925823e-01,  1.08295016e-01,\n",
            "       -1.50850296e-01,  4.89682764e-01, -1.25787228e-01,  1.09108603e-02,\n",
            "        2.03901350e-01,  3.73041213e-01,  4.14464206e-01, -2.04371363e-01,\n",
            "       -4.13147926e-01, -1.55505687e-01,  6.40573204e-02,  2.01762006e-01,\n",
            "        4.67928350e-01, -2.92257965e-01, -2.80420035e-01, -1.02366820e-01,\n",
            "        8.44074786e-02, -2.46815130e-01,  9.30914283e-03,  7.21381664e-01,\n",
            "        9.93466824e-02, -1.51890650e-01, -8.79100785e-02,  1.99137375e-01,\n",
            "       -2.85491109e-01, -1.45658910e-01,  1.31769180e-01, -2.30554402e-01,\n",
            "        9.93961930e-01, -4.89013761e-01, -4.41520941e-03,  3.33808184e-01,\n",
            "       -1.15862331e+01,  6.27067327e-01,  6.92195654e-01, -3.96494806e-01,\n",
            "       -1.55524760e-01, -2.77584672e-01,  1.40824944e-01,  6.30051643e-02,\n",
            "        3.22400272e-01, -6.31237149e-01,  2.64859498e-01,  1.19187705e-01,\n",
            "       -1.87773019e-01, -5.70387244e-02, -3.17759275e-01,  3.51091385e-01,\n",
            "       -8.25080499e-02, -5.88693619e-01,  1.85664043e-01, -4.37752664e-01,\n",
            "        2.94975162e-01, -3.88419479e-01, -3.14524949e-01,  1.93018720e-01,\n",
            "       -4.66019630e-01,  3.92425619e-02, -2.27697164e-01, -5.77223077e-02,\n",
            "        1.86803028e-01,  3.39595258e-01, -1.34775609e-01,  1.35530904e-02,\n",
            "        2.67631859e-01,  3.01105976e-01, -3.99771661e-01,  5.19367866e-02,\n",
            "       -1.93979561e-01,  3.31168234e-01,  7.49969780e-02,  6.77855015e-02,\n",
            "       -3.98677558e-01,  8.96267295e-02, -6.37425929e-02, -2.90560722e-01,\n",
            "        5.29231489e-01,  5.22804856e-01,  6.13129377e-01,  9.84214187e-01,\n",
            "        1.91501062e-02,  5.53350598e-02, -3.19021791e-01, -1.16616464e+00,\n",
            "       -1.82056829e-01, -2.55142778e-01, -4.34503099e-03,  1.08562961e-01,\n",
            "        5.60847878e-01, -6.78740144e-01,  7.56604850e-01, -2.26279721e-04,\n",
            "       -2.73889542e-01,  3.49136591e-01,  5.45473769e-02,  3.25957417e-01,\n",
            "       -2.34169811e-01,  1.98636055e-01,  2.76642740e-01,  2.71767974e-01,\n",
            "        2.19017565e-02, -7.24102199e-01,  3.35648865e-01, -6.34993196e-01,\n",
            "        5.86697638e-01,  1.58325225e-01,  5.48612952e-01, -2.04058196e-02,\n",
            "       -2.98064351e-01,  9.54014659e-02,  6.23647809e-01, -1.50524974e-01,\n",
            "       -1.05158590e-01,  1.95076391e-02, -1.55627847e-01, -5.23737788e-01,\n",
            "       -4.88567293e-01, -2.92480960e-02, -1.48493737e-01, -2.28548974e-01,\n",
            "        1.62468314e-01,  1.87906370e-01, -1.86603904e-01, -1.03929579e-01,\n",
            "        7.57003874e-02, -1.83488235e-01,  3.31494510e-01,  3.62008303e-01,\n",
            "       -4.08786982e-01, -3.60124677e-01,  2.35233158e-01,  6.22039378e-01,\n",
            "       -4.48782772e-01, -3.92183632e-01, -9.81487259e-02, -4.54179257e-01,\n",
            "       -2.17201523e-02, -8.63919705e-02,  2.35632397e-02, -5.16715869e-02,\n",
            "        3.28145683e-01,  3.80774498e-01, -1.97830677e-01, -3.20060045e-01,\n",
            "       -1.44663513e-01,  2.65059710e-01, -2.38168955e-01, -4.07844901e-01,\n",
            "        6.82541579e-02,  4.65799659e-01, -6.46328449e-01, -3.56019378e-01,\n",
            "       -2.66603440e-01, -1.56556755e-01,  3.48008871e-01, -3.12610537e-01,\n",
            "       -2.59965032e-01,  2.09201738e-01, -4.03216958e-01, -5.37508950e-02,\n",
            "        2.58988112e-01,  4.90068905e-02,  3.84673923e-01,  5.51094770e-01,\n",
            "       -6.67096898e-02, -1.56811252e-01, -2.92818546e-01, -8.46635520e-01,\n",
            "       -7.37516820e-01, -2.47719556e-01,  2.46730715e-01, -3.30779016e-01,\n",
            "       -4.31870967e-02, -6.95572570e-02,  1.10850438e-01, -6.39803648e-01,\n",
            "        3.08998436e-01, -5.68652563e-02,  1.21406717e-02,  1.99422061e-01,\n",
            "       -1.69394672e-01,  8.88709724e-02,  2.22181767e-01, -4.10160750e-01,\n",
            "        4.34417278e-02, -6.43242896e-02,  3.64626497e-01, -6.27047002e-01,\n",
            "       -7.62399077e-01, -2.09139347e-01,  4.24667895e-01, -2.34628379e-01,\n",
            "        3.62848938e-01,  2.90522337e-01,  1.00399494e-01, -5.94519302e-02,\n",
            "        2.48182639e-01,  6.14337206e-01, -4.55263965e-02, -5.80033422e-01,\n",
            "       -5.96113019e-02,  1.66195363e-01,  5.16803414e-02,  1.77246064e-01,\n",
            "        5.16551256e-01, -4.57567662e-01, -3.03259969e-01, -2.95126557e-01,\n",
            "       -2.42654681e-01,  9.48768616e-01, -5.74687660e-01, -6.06449723e-01,\n",
            "        4.44884300e-02, -1.72220275e-01,  1.74383447e-01, -3.67296115e-03,\n",
            "       -2.04028293e-01, -2.09711846e-02, -1.55579019e-02, -5.28713584e-01,\n",
            "       -4.72175062e-01, -3.36452164e-02,  2.86995262e-01,  2.66249359e-01,\n",
            "        1.08973615e-01, -2.78602820e-02,  1.58917353e-01, -6.10437095e-02,\n",
            "        1.38755932e-01,  1.87325478e-01, -3.09355110e-01, -2.35821575e-01,\n",
            "        4.08687890e-02,  1.67324454e-01,  3.47014606e-01,  9.21538770e-02,\n",
            "        1.14350729e-01,  5.31262439e-03,  1.00719020e-01,  1.26105237e+00,\n",
            "       -4.22409862e-01, -3.83172572e-01,  1.00451648e+00,  1.10669017e-01,\n",
            "       -1.04870602e-01,  3.61735046e-01,  3.81378621e-01,  2.57289946e-01,\n",
            "        2.85565674e-01, -2.12380499e-01,  2.43168455e-02, -1.74182743e-01,\n",
            "       -2.12956667e-01,  3.71870100e-01, -5.67532778e-01, -6.15019023e-01,\n",
            "       -4.43054765e-01,  8.19777608e-01, -3.97689998e-01,  3.14446568e-01,\n",
            "        1.35399014e-01,  3.68511602e-02, -2.79817320e-02, -2.67195612e-01,\n",
            "        1.38623208e-01, -4.08336103e-01,  4.16624874e-01, -5.05342066e-01,\n",
            "       -5.71159124e-01, -2.10586429e-01,  6.95821047e-02, -1.76875815e-01,\n",
            "        1.29245341e-01,  7.09777325e-02, -8.68781269e-01,  2.52151251e-01,\n",
            "        7.01645076e-01, -3.67768228e-01,  1.97472479e-02,  4.88699228e-02,\n",
            "       -6.12807944e-02,  2.16222294e-02, -2.29951128e-01,  4.69401553e-02,\n",
            "        3.59020710e-01, -5.30191481e-01, -3.24669540e-01,  2.99019456e-01,\n",
            "        5.45347333e-01, -1.33773804e-01,  2.46077120e-01, -2.67373502e-01,\n",
            "       -3.25464219e-01, -1.95056140e-01,  4.88246940e-02, -2.51512766e-01,\n",
            "        1.06680673e-02,  4.54178974e-02, -2.24505037e-01,  5.67949228e-02,\n",
            "        2.98036933e-02,  1.08783096e-01, -3.46710294e-01,  3.62881958e-01,\n",
            "        1.19454727e-01,  3.33309770e-01, -7.97005668e-02, -5.75046539e-01,\n",
            "        1.10518061e-01,  6.36671841e-01, -4.36376750e-01,  6.27085805e-01,\n",
            "       -2.70309418e-01, -6.66933715e-01, -5.81037514e-02, -1.46151185e-02,\n",
            "        7.62202814e-02,  6.21214569e-01, -2.68361658e-01, -1.30281121e-01,\n",
            "       -2.41215706e-01, -2.70983279e-01,  6.73588753e-01, -6.27068281e-02,\n",
            "        1.57615006e-01, -4.45451260e-01, -9.32109058e-02,  6.42048299e-01,\n",
            "       -3.91640365e-02,  7.14794874e-01, -5.45086488e-02,  4.79736537e-01,\n",
            "       -2.69659251e-01,  9.34062600e-02, -6.13043189e-01, -1.22704774e-01,\n",
            "        4.60647345e-02, -6.65461421e-02, -1.71397299e-01, -5.33107758e-01,\n",
            "        2.84736186e-01,  1.71948653e-02,  5.85604846e-01,  7.26147354e-01,\n",
            "       -1.45342946e-01, -3.82028639e-01, -5.60643733e-01, -3.29031914e-01,\n",
            "       -2.64098253e-02, -2.27873132e-01, -1.32940620e-01,  5.39366901e-01,\n",
            "       -2.08234176e-01, -3.72247070e-01, -5.74605882e-01,  1.85008273e-01,\n",
            "        2.61625320e-01, -1.95501909e-01,  7.82892644e-01,  2.60379881e-01,\n",
            "       -3.34283561e-01,  3.45121264e-01,  4.54949617e-01,  1.16417579e-01,\n",
            "        5.63180745e-01, -4.40695226e-01, -5.71482301e-01,  4.15788256e-02,\n",
            "        2.02609748e-01, -2.56504953e-01,  1.27472937e-01, -6.01405144e-01,\n",
            "        4.17354345e-01, -1.74505264e-01,  9.05938804e-01,  3.27409208e-01],\n",
            "      dtype=float32), 'Diagnostic_procedure': array([-1.94219947e-01, -5.48264682e-01, -6.42031372e-01, -5.32237053e-01,\n",
            "       -2.71531582e-01, -8.67531002e-02, -5.61139882e-01,  3.12201053e-01,\n",
            "       -1.90819487e-01,  1.67459562e-01,  2.36618996e-01,  2.27180168e-01,\n",
            "        3.88214327e-02, -3.32281858e-01, -6.05713844e-01,  1.12363644e-01,\n",
            "       -4.86561447e-01,  2.57682115e-01, -1.08849049e-01,  5.18697083e-01,\n",
            "        3.09948951e-01,  2.21105292e-01, -6.61924034e-02, -7.83013105e-02,\n",
            "        2.34802946e-01,  2.65911371e-01,  1.96608379e-01, -2.70475954e-04,\n",
            "        5.14005661e-01, -4.35673356e-01,  4.27156687e-01,  4.84807491e-01,\n",
            "       -2.44426489e-01,  6.22076213e-01, -1.85437992e-01, -3.08186918e-01,\n",
            "       -1.80134699e-01, -3.13814431e-01,  7.30822325e-01,  1.53835313e-02,\n",
            "        2.10059926e-01,  6.33993149e-01,  2.52483338e-01,  2.35564504e-02,\n",
            "        2.00209305e-01,  4.91586000e-01,  2.18529984e-01, -6.57529533e-01,\n",
            "        7.51971304e-02, -2.28389502e-01, -3.56086135e-01, -6.09482825e-03,\n",
            "       -1.04974963e-01, -4.66056913e-03, -2.65204698e-01,  1.51743963e-01,\n",
            "        1.47242710e-01,  4.18832541e-01, -4.19784933e-01,  3.14281166e-01,\n",
            "       -3.10252994e-01,  5.64120948e-01, -2.75987953e-01, -1.30113974e-01,\n",
            "       -7.04762116e-02,  1.37505129e-01, -3.71555835e-01, -2.42311046e-01,\n",
            "       -4.18858171e-01, -5.22366576e-02, -5.20250015e-02,  6.78750947e-02,\n",
            "        9.25319567e-02, -2.37452105e-01,  3.39449830e-02, -2.04298496e-01,\n",
            "        2.61616737e-01, -2.14931324e-01,  1.82112437e-02, -3.61739635e-01,\n",
            "        9.00268480e-02,  4.70184654e-01, -3.34728137e-02, -7.54810357e-03,\n",
            "       -2.88759202e-01, -5.26652150e-02,  2.94778109e-01,  2.70085365e-01,\n",
            "        1.45031288e-01,  1.43649444e-01,  1.79215387e-01,  8.18168670e-02,\n",
            "       -3.47986937e-01, -7.55435005e-02,  1.96422324e-01, -6.98722899e-02,\n",
            "        8.66375640e-02, -1.06544770e-01,  4.70563956e-02, -2.85290748e-01,\n",
            "       -2.22728446e-01,  5.66667430e-02,  1.83831692e-01, -3.98305416e-01,\n",
            "       -1.61783010e-01, -3.43147367e-01,  3.65653515e-01,  7.71335289e-02,\n",
            "       -6.72080457e-01, -1.48357287e-01, -1.19474798e-01, -2.92482942e-01,\n",
            "        1.59179762e-01, -6.45920217e-01, -5.04283719e-02,  5.85120432e-02,\n",
            "        5.51799357e-01, -1.15144610e-01, -3.61984462e-01,  7.90426657e-02,\n",
            "        6.88087270e-02, -1.29064679e-01,  2.01779291e-01, -1.17988728e-01,\n",
            "        3.37465763e-01, -2.54691094e-01,  5.77451646e-01,  2.49475241e-03,\n",
            "        4.15682197e-02,  5.17205775e-01,  1.92592889e-02, -3.61801058e-01,\n",
            "       -4.40570921e-01, -2.88589358e-01,  1.93093613e-01, -8.43810569e-03,\n",
            "       -1.22593105e-01, -1.81004465e-01, -1.06886728e-02,  9.85019132e-02,\n",
            "       -3.25303644e-01, -2.14756489e-01, -3.15520674e-01, -2.99649715e-01,\n",
            "       -4.66126241e-02, -2.69657403e-01,  2.44133070e-01, -3.27577919e-01,\n",
            "       -3.61113787e-01, -4.32920307e-01, -5.31189859e-01, -4.99183983e-01,\n",
            "        2.41184950e-01,  3.46289307e-01,  2.02204823e-01, -4.17620152e-01,\n",
            "        2.30815366e-01,  8.68496299e-02,  1.27933025e-01,  2.58864790e-01,\n",
            "        2.76342809e-01, -2.00598761e-01,  2.49146521e-01, -2.66026497e-01,\n",
            "       -1.74500346e-01,  5.54306395e-02,  1.49618015e-01, -2.01591775e-01,\n",
            "       -5.43659814e-02,  3.27577949e-01,  1.69291794e-01, -2.37195253e-01,\n",
            "        5.00338189e-02, -5.34738600e-01, -6.11900985e-02, -2.82378644e-02,\n",
            "        1.68280527e-01,  2.11336408e-02,  6.64194077e-02,  2.95530949e-02,\n",
            "        2.38719508e-01, -2.05170441e+00,  1.24298232e-02, -1.93715632e-01,\n",
            "       -2.05428496e-01, -1.85972571e-01, -6.56646490e-03, -3.12112253e-02,\n",
            "        3.73378366e-01,  3.59621316e-01, -3.96450251e-01,  1.35240734e-01,\n",
            "       -1.43364621e-02, -1.67611659e-01, -1.09946214e-01,  5.62533081e-01,\n",
            "        5.78980863e-01,  4.91985947e-01,  2.00495973e-01,  1.33526683e-01,\n",
            "       -5.32689154e-01,  2.76776046e-01, -1.38817549e-01, -1.00316219e-01,\n",
            "       -1.73870921e-01,  2.10291103e-01, -3.77470136e-01,  4.45411503e-02,\n",
            "       -8.64256639e-03,  1.59117103e-01, -1.45198941e-01, -2.41588950e-01,\n",
            "       -4.30800110e-01, -1.25964209e-01,  4.79618311e-02,  3.22163880e-01,\n",
            "       -1.34243712e-01, -3.65948528e-01,  3.51464510e-01, -7.61586875e-02,\n",
            "        7.16958568e-02, -2.14686587e-01,  1.85701679e-02, -2.30025604e-01,\n",
            "        1.92590877e-01,  5.08226693e-01, -1.44256189e-01,  2.04723790e-01,\n",
            "       -2.93855518e-01, -1.47761866e-01, -6.07547641e-04, -3.77643347e-01,\n",
            "        4.06059831e-01,  2.51626939e-01, -1.15975536e-01,  1.20325334e-01,\n",
            "       -2.13641807e-01,  7.43363380e-01, -2.54019171e-01, -2.23713577e-01,\n",
            "       -2.20944881e-01, -2.95659482e-01, -3.03039581e-01,  2.56308764e-01,\n",
            "        5.79573102e-02,  3.41428327e-03,  8.07312652e-02,  5.78944862e-01,\n",
            "        5.84199846e-01,  4.99339372e-01,  3.91272493e-02, -6.32588863e-02,\n",
            "        4.30719763e-01,  3.06180269e-01, -6.85267523e-02, -1.95420921e-01,\n",
            "        2.03583166e-01, -5.75517230e-02,  3.51547480e-01, -1.59150019e-01,\n",
            "       -1.47609517e-01, -3.08205485e-02,  1.62202075e-01,  5.11408031e-01,\n",
            "       -8.80388841e-02,  1.72708020e-01,  4.17158216e-01, -1.72454238e-01,\n",
            "       -3.12054753e-01, -5.53007722e-02, -2.78139323e-01,  1.92239344e-01,\n",
            "       -2.10835710e-01, -2.43016526e-01, -5.64285755e-01, -1.39840946e-01,\n",
            "       -2.02027753e-01, -1.59593865e-01,  1.36005595e-01, -4.26112443e-01,\n",
            "       -1.60287723e-01, -4.57741879e-02,  1.77106664e-01, -7.00257942e-02,\n",
            "       -1.91367134e-01,  7.99116045e-02, -1.20575905e-01, -9.84211564e-02,\n",
            "       -5.01987815e-01, -2.48273656e-01, -8.52088705e-02,  1.06777698e-01,\n",
            "       -2.97263265e-01,  2.76855286e-02,  2.67039016e-02,  5.07255852e-01,\n",
            "        3.85486692e-01,  7.03649893e-02,  2.13604078e-01, -1.48646578e-01,\n",
            "        5.35283647e-02,  6.92302138e-02,  8.75706747e-02, -2.04943582e-01,\n",
            "        1.18070520e-01, -2.50204355e-02,  1.11226372e-01,  3.81388992e-01,\n",
            "       -4.97128576e-01, -1.45047531e-01,  6.72588823e-03,  4.82211262e-01,\n",
            "       -1.12902828e-01,  1.17121935e-01, -1.25479683e-01, -5.42095840e-01,\n",
            "        3.18912566e-01, -8.62762704e-02, -1.38381839e-01, -3.86951566e-01,\n",
            "        2.34754339e-01,  2.83438396e-02, -1.53079197e-01, -8.16742852e-02,\n",
            "       -9.73072276e-02,  2.69207507e-01,  2.35046923e-01,  1.76589131e-01,\n",
            "       -1.53850421e-01, -4.35215086e-01, -3.43725868e-02,  1.38035581e-01,\n",
            "       -1.16360717e-01, -1.69602379e-01,  3.42410326e-01,  4.50259775e-01,\n",
            "        2.60570914e-01, -2.49082729e-01, -5.27005456e-02,  4.79328334e-02,\n",
            "        5.28388739e-01,  3.44811529e-02, -1.33237004e-01,  1.45665661e-01,\n",
            "        1.07468605e-01, -8.04881305e-02,  4.81784195e-01, -1.75674722e-01,\n",
            "       -1.22215383e-01,  2.63952851e-01,  2.39715099e-01,  6.60709068e-02,\n",
            "        3.12433336e-02, -3.02245617e-01,  5.92977524e-01,  5.42991459e-01,\n",
            "        1.57410204e-01, -4.05031256e-02,  4.56487447e-01,  4.79290396e-01,\n",
            "       -4.96872693e-01,  2.23263696e-01, -1.58797458e-01, -2.10322797e-01,\n",
            "       -3.28922234e-02,  7.39912763e-02,  2.67418116e-01, -1.42544493e-01,\n",
            "       -2.87752692e-02, -1.35828359e-02,  3.45455319e-01, -1.84778795e-01,\n",
            "        4.52376492e-02,  6.44494593e-02,  4.94319163e-02, -1.94112852e-01,\n",
            "       -3.00461501e-01,  8.22452232e-02, -1.39897749e-01, -1.34772286e-01,\n",
            "        8.55071470e-02, -1.32097945e-01,  1.60626560e-01, -4.28636640e-01,\n",
            "       -4.58027005e-01, -2.43240595e-01,  3.40739302e-02,  2.37821177e-01,\n",
            "        1.05123997e-01, -7.32042268e-02,  7.21641481e-02, -8.67283344e-02,\n",
            "       -1.33459747e-01,  5.49080789e-01, -1.28413334e-01, -5.03946781e-01,\n",
            "        3.35779905e-01, -1.78379864e-02,  1.96261451e-01,  1.92117333e-01,\n",
            "       -2.80646324e-01,  2.57614344e-01, -2.51298752e-02,  9.96708944e-02,\n",
            "       -3.72947641e-02, -2.75387675e-01, -1.34904623e-01, -1.81864575e-01,\n",
            "       -4.11809325e-01, -2.70655066e-01, -2.35297307e-01,  4.23060656e-01,\n",
            "       -2.53256951e-02, -3.41264717e-03, -1.08336262e-01,  3.77283305e-01,\n",
            "       -1.28100231e-01,  9.00880471e-02, -6.92778304e-02, -6.00792877e-02,\n",
            "        4.67406183e-01,  1.30065396e-01,  3.08212396e-02,  3.80182773e-01,\n",
            "       -1.17484503e+01,  1.03178762e-01,  6.17627859e-01, -1.09586798e-01,\n",
            "       -4.69365530e-02, -2.07088571e-02,  3.36686283e-01, -3.24362725e-01,\n",
            "        2.18967810e-01, -2.22118810e-01, -4.69944514e-02, -3.59897763e-02,\n",
            "       -1.92236736e-01, -2.30329916e-01, -5.68173587e-01, -5.57342879e-02,\n",
            "        2.21736267e-01, -6.63242102e-01, -2.21958552e-02,  7.31015503e-02,\n",
            "        1.65490106e-01, -2.12071121e-01, -2.30059382e-02,  3.99011970e-01,\n",
            "       -4.16843385e-01, -8.32180306e-02, -3.57938141e-01,  2.37821043e-03,\n",
            "       -1.66139662e-01,  3.57967734e-01, -3.71381432e-01, -1.41355217e-01,\n",
            "        7.52922595e-02,  3.15101355e-01, -2.56366134e-01, -3.10279697e-01,\n",
            "        4.12904732e-02, -6.37885407e-02, -3.02450329e-01, -2.31005684e-01,\n",
            "       -1.71045586e-01, -1.43624052e-01, -3.88245314e-01,  3.41730326e-01,\n",
            "        4.49435085e-01,  5.78755379e-01,  6.91675901e-01,  5.39345205e-01,\n",
            "        2.23422926e-02, -2.06477925e-01, -8.29638466e-02, -6.47556961e-01,\n",
            "        8.68543610e-02, -6.72724903e-01, -3.15518707e-01, -1.12268895e-01,\n",
            "        1.96908653e-01, -3.36846918e-01,  3.00568372e-01, -2.15863347e-01,\n",
            "        2.07031462e-02,  2.82536447e-01,  2.17332989e-02,  1.41520485e-01,\n",
            "       -2.01353550e-01,  9.89993587e-02,  3.40786368e-01, -3.82840723e-01,\n",
            "        3.47418077e-02, -2.69304961e-01, -1.26863912e-01,  6.18450195e-02,\n",
            "        6.09649360e-01,  3.09823424e-01,  5.95582509e-03,  1.49152517e-01,\n",
            "       -3.79890166e-02,  1.08010799e-01,  1.09275734e+00, -3.81601714e-02,\n",
            "        1.47565335e-01, -5.01299314e-02,  6.61861897e-02, -4.50512886e-01,\n",
            "       -5.11828482e-01, -2.18105569e-01, -1.16753064e-01, -2.39170909e-01,\n",
            "        5.94306290e-01,  4.50141430e-01, -1.41875878e-01, -1.61963418e-01,\n",
            "       -2.22644076e-01,  1.37222275e-01,  4.34948355e-01,  5.31382561e-01,\n",
            "       -4.55629557e-01,  4.96231318e-01, -9.48692486e-02,  1.72081307e-01,\n",
            "        9.86840948e-02, -3.79326604e-02, -8.25789049e-02, -1.45789936e-01,\n",
            "       -2.10176215e-01,  4.99701291e-01,  3.21763158e-02,  1.59286410e-01,\n",
            "        4.19413596e-02,  3.87482420e-02, -3.20087433e-01,  9.55844745e-02,\n",
            "        2.52596110e-01,  2.71047562e-01, -3.69888395e-01, -2.48094082e-01,\n",
            "        4.81623411e-01,  2.61928320e-01, -4.96496409e-01, -4.66590881e-01,\n",
            "       -3.35223466e-01, -1.01740181e-01,  4.53393549e-01, -3.81878316e-02,\n",
            "       -7.18195215e-02,  2.18837932e-01, -5.88695824e-01, -2.94569075e-01,\n",
            "        8.36396683e-03, -1.05597280e-01,  4.41699237e-01,  1.89799607e-01,\n",
            "        4.47151847e-02,  1.67586610e-01, -3.90782714e-01, -5.06419599e-01,\n",
            "       -1.03593242e+00, -7.72825852e-02,  4.00077492e-01, -3.16021442e-01,\n",
            "        6.33005053e-02,  1.58336554e-02, -1.38068944e-02, -5.41008972e-02,\n",
            "        2.81669080e-01, -1.65809035e-01,  2.21433356e-01,  9.94509980e-02,\n",
            "       -2.11233184e-01,  1.74191073e-01, -1.45707473e-01, -2.78507084e-01,\n",
            "       -1.23622321e-01, -1.03922719e-02, -2.44015679e-01, -3.14096957e-01,\n",
            "       -5.16983606e-02, -1.61382005e-01,  9.34836268e-03,  3.71447876e-02,\n",
            "        4.71307784e-02,  2.38221347e-01,  8.89443606e-03, -2.31063768e-01,\n",
            "        2.85489976e-01,  1.78628042e-01, -6.41683713e-02, -2.87423968e-01,\n",
            "       -4.25180912e-01, -5.47123142e-02,  1.88777566e-01,  1.66642234e-01,\n",
            "        1.69739529e-01, -1.96559668e-01, -3.63144279e-01, -3.41524333e-01,\n",
            "       -1.22167461e-01,  8.81161690e-01, -4.47419077e-01, -1.90344855e-01,\n",
            "       -2.61822436e-02,  3.38640839e-01, -1.38214916e-01, -1.59072742e-01,\n",
            "       -3.43871824e-02,  3.54007572e-01, -1.13086365e-01, -2.00693175e-01,\n",
            "        4.21872854e-01, -9.37418342e-02,  2.97500283e-01,  2.61137635e-01,\n",
            "        1.04173772e-01, -3.16668183e-01,  4.70584959e-01, -1.73360661e-01,\n",
            "       -2.50845909e-01,  6.24446124e-02, -1.61025211e-01, -8.80454015e-03,\n",
            "        3.92019182e-01, -2.60544151e-01,  1.91560581e-01, -3.80514175e-01,\n",
            "       -9.02034864e-02, -1.95696101e-01,  4.16094214e-01,  1.15371358e+00,\n",
            "       -1.07485414e-01,  1.34093508e-01,  7.46825516e-01, -1.15154423e-01,\n",
            "        5.72821200e-02,  2.67221540e-01, -6.35484532e-02,  3.14066596e-02,\n",
            "        4.80300933e-01, -3.41029078e-01, -3.66521358e-01,  3.89681756e-02,\n",
            "        4.40622002e-01,  2.05732897e-01, -5.22590101e-01, -4.69961256e-01,\n",
            "       -4.30383272e-02,  2.25573376e-01,  4.99074347e-02,  3.33880872e-01,\n",
            "       -1.68918982e-01,  3.32261175e-01,  3.77177000e-01, -6.52519047e-01,\n",
            "        8.57944712e-02, -1.41783521e-01,  6.40529692e-01, -6.51468337e-03,\n",
            "       -4.06139761e-01, -9.66727510e-02, -2.26147696e-01, -1.39648512e-01,\n",
            "       -4.59579267e-02,  5.84964693e-01, -4.21547085e-01, -4.08767946e-02,\n",
            "        6.01848543e-01, -4.40698832e-01,  1.98270380e-01,  3.90809812e-02,\n",
            "       -1.56996772e-02, -7.55377263e-02, -3.57877642e-01, -2.78678626e-01,\n",
            "        7.27622882e-02, -2.88597018e-01, -4.87117440e-01,  3.75366241e-01,\n",
            "        3.48555833e-01, -9.15666763e-03,  8.24353397e-02, -3.76395695e-02,\n",
            "       -1.99712172e-01, -1.52353704e-01, -1.25688255e-01, -1.84794828e-01,\n",
            "       -1.30032599e-02,  3.99960065e-03, -2.45816827e-01, -1.11949682e-01,\n",
            "        1.24986522e-01,  2.32209042e-01, -1.37959793e-01,  2.01972798e-01,\n",
            "        3.59268993e-01,  7.34974861e-01, -1.15500718e-01, -3.54881853e-01,\n",
            "        2.01199099e-01,  4.52881098e-01, -2.59733081e-01,  5.31332791e-01,\n",
            "        1.77466705e-01, -4.46328193e-01,  1.55195773e-01,  1.51745066e-01,\n",
            "       -5.98338425e-01,  2.86364079e-01, -2.18036518e-01, -2.00645626e-02,\n",
            "       -3.58707398e-01, -8.52512121e-02,  1.64142206e-01,  3.14166546e-01,\n",
            "       -4.65949858e-03, -4.12231684e-01, -9.03081894e-02,  3.96926135e-01,\n",
            "        9.44614485e-02,  3.55243117e-01, -8.10720176e-02,  7.12507486e-01,\n",
            "       -1.60901293e-01,  1.09184124e-01, -3.86128098e-01, -7.23969117e-02,\n",
            "        4.81685214e-02,  6.20102622e-02, -4.13363189e-01, -3.25511456e-01,\n",
            "        6.69362545e-01,  1.57435928e-02,  4.56391722e-01,  6.28521502e-01,\n",
            "       -1.53280869e-01, -6.18098974e-01, -2.94754297e-01, -7.40287155e-02,\n",
            "       -3.41156244e-01, -1.20215230e-01,  3.00148100e-01, -1.16279088e-01,\n",
            "       -1.42028943e-01, -6.59985423e-01, -2.76020989e-02, -9.24217328e-02,\n",
            "        1.52900323e-01, -1.25966266e-01,  4.19401497e-01, -1.85011268e-01,\n",
            "        4.37846452e-01,  2.83455640e-01,  3.30349028e-01, -2.59264499e-01,\n",
            "        4.64607328e-01, -1.45164773e-01, -3.32571119e-01,  9.83097553e-02,\n",
            "        2.78000772e-01, -1.89405635e-01, -2.79740870e-01, -5.73337018e-01,\n",
            "        9.25477445e-02, -1.90846756e-01,  6.11946225e-01,  6.11304879e-01],\n",
            "      dtype=float32)}\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "Now, all vectors get averaged, resulting in one vector for all entities."
      ],
      "metadata": {
        "id": "sAyBf-xnEDW0"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "vector_list = [torch.tensor(v) if isinstance(v, np.ndarray) else v for v in vector_dict_patient.values()]\n",
        "average_vector_patient = torch.stack(vector_list).mean(dim=0)\n",
        "print(average_vector_patient)"
      ],
      "metadata": {
        "id": "BEn0Cv1ZUVbz",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "24cba784-547c-43a9-baf5-2b9408b2c3c2"
      },
      "execution_count": 47,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "tensor([-2.6287e-01, -5.5217e-01, -6.4275e-01, -4.0841e-01, -2.1664e-01,\n",
            "        -2.1445e-01, -5.2317e-01,  3.8331e-01,  2.2948e-02,  1.9140e-01,\n",
            "         1.5245e-01,  2.5273e-01, -2.4588e-02, -3.8370e-01, -5.1313e-01,\n",
            "        -3.3356e-02,  6.3320e-03,  2.9500e-01,  5.5257e-02,  3.4000e-01,\n",
            "         4.5481e-01,  3.5739e-01, -6.3879e-02, -9.4655e-02, -1.4587e-01,\n",
            "         7.2917e-02, -1.0249e-02, -5.0220e-02,  4.6808e-01, -2.2201e-01,\n",
            "         1.6824e-01,  4.4901e-01, -3.6082e-01,  6.7337e-01, -1.1133e-01,\n",
            "        -3.3917e-01, -1.7223e-02, -2.1815e-01,  7.2779e-01,  6.5206e-02,\n",
            "         2.2852e-01,  3.0957e-01,  1.8308e-01,  2.3772e-02,  1.1512e-01,\n",
            "         3.8970e-01,  6.4321e-02, -5.2707e-01,  2.4572e-01, -2.1427e-01,\n",
            "        -2.5681e-01, -5.1376e-02,  1.1637e-01, -2.9529e-02, -5.6034e-01,\n",
            "        -1.6303e-02, -2.5902e-02,  2.0966e-01, -3.7153e-01,  2.5158e-01,\n",
            "        -1.5748e-01,  5.5396e-01, -1.3207e-01, -1.4678e-01,  7.7694e-02,\n",
            "        -1.0562e-01, -4.6436e-01, -2.5583e-01, -4.0529e-01, -2.5656e-02,\n",
            "        -3.6066e-02,  2.2220e-01,  1.0257e-01, -7.4625e-02,  2.0320e-01,\n",
            "        -2.0760e-01,  2.1406e-01, -1.4659e-01, -1.0848e-01, -3.1588e-01,\n",
            "         2.0989e-01,  3.9201e-01, -4.6940e-02, -2.0554e-01, -2.0441e-01,\n",
            "        -5.1210e-02,  2.4824e-01,  1.9123e-01,  3.5747e-02,  5.7915e-02,\n",
            "         2.5427e-01,  7.3743e-02, -3.2273e-01, -1.6654e-01,  1.9631e-01,\n",
            "        -2.2517e-01,  9.7135e-02, -1.3432e-01,  4.3402e-02, -3.3094e-01,\n",
            "        -1.1391e-01, -3.1989e-02,  1.4172e-01, -1.0461e-01, -2.4001e-01,\n",
            "        -3.4971e-01,  2.9661e-01,  3.2291e-02, -3.9670e-01, -1.5516e-01,\n",
            "        -1.1105e-01, -2.7570e-01,  1.6647e-01, -4.5391e-01, -6.5862e-03,\n",
            "         1.3797e-01,  3.9995e-01, -5.4798e-02, -2.2235e-01,  9.5597e-02,\n",
            "         2.1968e-01, -2.3952e-01,  5.2864e-02,  4.1782e-02,  2.5570e-01,\n",
            "        -2.2507e-01,  4.8079e-01, -1.6780e-01,  2.3956e-01,  5.1457e-01,\n",
            "         3.9928e-02, -1.2647e-01, -4.1894e-01, -1.4333e-01,  6.4817e-02,\n",
            "         1.0940e-01, -1.9939e-01, -2.4133e-01,  7.1268e-02,  1.8033e-01,\n",
            "        -9.1610e-02, -1.4713e-01, -3.2747e-01, -8.2038e-02, -7.9936e-02,\n",
            "        -2.1707e-01,  1.6394e-01, -3.2873e-01, -3.3163e-01, -3.1893e-01,\n",
            "        -6.7477e-01, -4.0883e-01,  4.5775e-01,  4.3252e-02,  1.1721e-01,\n",
            "        -3.8949e-01,  2.2478e-01,  1.3749e-02,  2.8390e-02,  2.2239e-01,\n",
            "         5.3168e-01, -2.2180e-01,  2.4405e-01, -2.8176e-01, -5.1453e-03,\n",
            "         1.2645e-01, -1.6739e-01, -2.0821e-01, -1.5377e-02,  1.1320e-01,\n",
            "         1.1219e-01, -2.4326e-01,  1.8184e-01, -3.2601e-01, -2.3010e-02,\n",
            "        -2.6121e-01,  1.0376e-01, -9.8725e-02, -4.6474e-02,  1.1243e-01,\n",
            "         2.0199e-01, -2.0895e+00,  2.8743e-02, -6.7717e-02, -6.6684e-02,\n",
            "        -3.7648e-01, -1.1942e-02, -2.2824e-02,  3.6964e-01,  2.1025e-02,\n",
            "        -1.9630e-01,  1.2399e-01, -1.8515e-01, -3.5266e-02, -1.7750e-01,\n",
            "         4.0350e-01,  4.3109e-01,  3.1796e-01,  8.7262e-02,  1.0887e-01,\n",
            "        -3.6120e-01,  2.8861e-01,  7.7558e-02, -1.0384e-02, -2.5016e-02,\n",
            "         2.0260e-01, -1.6062e-01,  1.9539e-01, -3.0861e-02,  1.0068e-01,\n",
            "        -1.3535e-01, -1.1105e-01, -3.3071e-01, -7.6679e-03,  3.0887e-02,\n",
            "         4.0645e-01, -2.8564e-01, -3.5821e-01,  3.8352e-01,  5.4559e-02,\n",
            "        -9.7333e-02, -1.2023e-01, -1.2094e-01, -2.9554e-01,  7.8517e-02,\n",
            "         4.5997e-01, -2.8188e-01,  2.2423e-01, -1.0900e-01, -2.3486e-01,\n",
            "         9.4573e-02, -2.2706e-01,  3.9402e-01,  2.5669e-01,  8.6285e-03,\n",
            "        -1.0325e-01, -2.1850e-01,  4.5845e-01, -1.7782e-01, -6.6759e-02,\n",
            "        -2.0071e-01, -3.9302e-01, -3.5899e-01,  1.3464e-01,  2.7106e-01,\n",
            "         1.3567e-01,  2.0435e-02,  5.4073e-01,  4.4865e-01,  4.1266e-01,\n",
            "         1.1899e-01,  8.7961e-02,  3.9448e-01,  3.3814e-01, -3.2170e-01,\n",
            "        -2.3007e-02,  2.1907e-01, -7.1454e-02,  2.2931e-01, -1.3600e-01,\n",
            "        -1.9512e-01, -6.9239e-02,  1.2865e-01,  4.6168e-01, -1.8070e-01,\n",
            "         1.1863e-02,  3.1449e-01, -1.8017e-01, -2.6149e-01,  1.1229e-01,\n",
            "        -2.6381e-01,  3.7751e-01, -1.7353e-01, -1.9903e-01, -5.9797e-01,\n",
            "        -6.6805e-01, -2.2064e-01, -1.3407e-01,  1.3389e-01, -5.9355e-01,\n",
            "        -3.0471e-01, -2.8745e-01,  5.9972e-02, -9.2360e-02, -1.1795e-01,\n",
            "         1.8523e-01, -7.7835e-02,  2.4983e-02, -3.0962e-01, -2.4346e-01,\n",
            "        -1.6633e-01,  9.2763e-02, -2.5612e-01, -5.4610e-03,  9.6301e-02,\n",
            "         1.1331e-02,  1.3852e-01,  1.4879e-01, -4.8687e-02, -1.9083e-01,\n",
            "        -1.3102e-02,  1.9658e-01,  1.0377e-01, -6.8384e-02,  5.2998e-02,\n",
            "         1.2872e-01,  1.3706e-01,  4.5050e-01, -3.4413e-01, -3.0333e-01,\n",
            "        -9.4469e-02,  3.7289e-01, -1.1317e-01, -1.5192e-02,  7.0304e-03,\n",
            "        -7.3221e-01,  3.7177e-02, -4.3052e-02, -2.7014e-01, -2.4756e-01,\n",
            "         2.4339e-01,  2.7901e-01, -1.2849e-01,  7.0387e-03, -1.2404e-01,\n",
            "         3.8474e-01,  4.9657e-01,  3.4214e-02, -4.2138e-02, -3.8370e-01,\n",
            "         7.7653e-04, -1.2854e-01, -1.6494e-01, -2.1058e-01,  3.0248e-01,\n",
            "         3.2785e-01,  1.8086e-01, -1.1881e-01, -1.9591e-02, -7.5693e-02,\n",
            "         4.2025e-01, -9.6914e-02, -1.8882e-01,  3.5260e-01, -4.0912e-02,\n",
            "         8.7906e-03,  3.4084e-01, -3.6287e-01, -1.1402e-01,  2.2443e-01,\n",
            "         2.2513e-01, -9.2524e-02, -2.6301e-01, -2.4577e-01,  4.6197e-01,\n",
            "         4.3636e-01,  2.0890e-01, -9.8289e-02,  4.0767e-01,  3.4605e-01,\n",
            "        -2.2766e-01,  1.8422e-01, -2.1664e-01, -2.1286e-01,  6.1400e-02,\n",
            "         3.0301e-01,  3.1754e-01, -3.2339e-01, -7.2780e-02, -1.6109e-01,\n",
            "         1.4676e-01, -1.7846e-01, -6.1206e-02,  1.5606e-01,  3.6420e-01,\n",
            "        -2.7927e-01, -4.3620e-01,  1.1745e-01, -3.4167e-01, -3.5339e-01,\n",
            "         6.5116e-02, -6.6304e-02,  3.7786e-02, -4.0960e-01, -1.7730e-01,\n",
            "        -2.5776e-01, -2.3382e-02,  2.9631e-01,  1.2960e-01,  5.3882e-02,\n",
            "        -1.4000e-02, -8.6875e-02, -1.5375e-01,  5.9190e-01,  4.0044e-02,\n",
            "        -2.9482e-01,  3.7466e-01, -6.0569e-02,  1.4720e-01,  3.0994e-01,\n",
            "        -3.0468e-01, -2.4338e-01, -1.8696e-01,  1.7489e-01,  1.2126e-01,\n",
            "        -2.7553e-01, -3.0266e-02,  5.4314e-02, -1.7151e-01, -1.0243e-01,\n",
            "        -1.6724e-01,  3.4517e-01, -8.1579e-02, -8.8129e-02, -8.7900e-02,\n",
            "         2.8912e-01, -1.8921e-01,  6.9490e-02,  8.2552e-02, -2.7967e-01,\n",
            "         5.4414e-01,  1.0198e-01,  4.3209e-02,  3.8887e-01, -1.1465e+01,\n",
            "         3.4369e-01,  5.3442e-01, -1.8238e-01,  3.0668e-02, -7.5481e-02,\n",
            "         1.7287e-01, -1.8467e-01,  2.6603e-01, -3.7919e-01, -4.6135e-02,\n",
            "        -1.3037e-02, -1.5734e-01, -1.7661e-01, -4.9833e-01,  1.2946e-02,\n",
            "         2.1403e-02, -5.7860e-01,  1.2958e-01, -1.6916e-01,  1.9609e-01,\n",
            "        -2.6756e-01, -1.4063e-01,  3.2269e-01, -2.1617e-01, -1.5603e-01,\n",
            "        -1.3314e-01, -6.1064e-02,  9.4368e-02,  2.6629e-01, -2.0314e-01,\n",
            "         5.6995e-02, -2.9472e-02,  2.4129e-01, -2.3502e-01, -2.9194e-01,\n",
            "         1.1704e-01, -2.2294e-01, -3.8057e-02, -1.4696e-01, -2.4723e-01,\n",
            "        -2.5242e-02, -2.6466e-01,  2.1483e-01,  4.2099e-01,  4.9641e-01,\n",
            "         8.1936e-01,  6.6986e-01,  4.6416e-02, -6.4600e-02, -1.1924e-01,\n",
            "        -6.9550e-01,  1.0819e-01, -5.1567e-01, -1.0954e-01, -1.0856e-01,\n",
            "         3.5861e-01, -4.7509e-01,  3.0956e-01, -2.4022e-01,  4.5223e-02,\n",
            "         2.5609e-01,  1.4531e-01,  7.4897e-02, -4.6838e-01,  6.9910e-02,\n",
            "         2.7745e-01, -1.2453e-01, -3.8349e-02, -4.0198e-01,  5.7497e-03,\n",
            "        -1.9234e-02,  4.1714e-01,  1.3652e-01,  4.9250e-02,  5.5536e-02,\n",
            "        -2.2928e-01,  1.1821e-02,  1.2509e+00,  5.7506e-03,  1.1315e-01,\n",
            "         3.3418e-02,  6.3507e-02, -1.2309e-01, -4.3435e-01, -1.6057e-01,\n",
            "        -1.4365e-02, -3.1511e-01,  2.5914e-01,  2.8581e-01, -2.0032e-01,\n",
            "        -6.4941e-02, -8.7640e-02,  7.7928e-02,  2.8920e-01,  5.8213e-01,\n",
            "        -6.6398e-01,  1.3425e-01, -5.7535e-02,  3.2026e-01,  1.9638e-02,\n",
            "        -1.5662e-02, -1.8258e-01, -4.0251e-01, -3.3188e-02, -1.8203e-02,\n",
            "         9.2520e-02,  1.2291e-01,  1.1946e-01,  2.4818e-01, -1.3097e-01,\n",
            "         8.7252e-02,  1.2949e-01,  2.2551e-01, -2.6733e-01, -4.1043e-01,\n",
            "         2.4829e-01,  2.5343e-01, -5.3004e-01, -3.1726e-01, -1.6138e-01,\n",
            "        -3.0262e-02,  3.2440e-01,  2.1071e-02, -2.8466e-01,  1.8761e-01,\n",
            "        -5.3233e-01, -7.3147e-03,  1.4854e-01,  7.9566e-02,  4.6900e-01,\n",
            "         3.9276e-01, -2.1029e-01,  3.6588e-02, -3.1417e-01, -4.4341e-01,\n",
            "        -9.3155e-01,  2.7036e-03,  3.9438e-01, -2.2462e-01,  1.5238e-01,\n",
            "        -1.4919e-01, -5.2574e-02, -7.0031e-02,  2.7023e-01,  1.0435e-01,\n",
            "         4.0280e-01,  1.9600e-01, -2.7996e-01,  1.0075e-01, -6.8370e-02,\n",
            "        -3.3487e-01, -1.2134e-01, -1.2618e-01, -1.0220e-01, -4.5012e-01,\n",
            "        -2.2647e-01, -7.9385e-02,  6.6408e-03, -6.6614e-02,  6.3210e-02,\n",
            "         1.0371e-02, -7.3242e-02, -1.7932e-01,  3.2708e-01,  2.7534e-01,\n",
            "         2.5007e-02, -3.6600e-01, -3.2937e-01, -6.3444e-02,  1.5618e-01,\n",
            "         2.3928e-01,  1.7661e-01, -2.0628e-01, -3.8005e-01, -7.7725e-02,\n",
            "        -1.7460e-01,  8.3137e-01, -4.5082e-01, -3.6449e-01,  5.8745e-02,\n",
            "         2.6850e-01, -1.1281e-01, -2.6833e-01,  2.8982e-02,  1.5785e-01,\n",
            "        -1.3803e-01, -2.0287e-01,  1.1779e-01, -2.3396e-02,  2.9092e-01,\n",
            "         2.8321e-01,  2.7114e-01, -2.1833e-01,  3.4202e-01,  5.6397e-02,\n",
            "        -2.5146e-01,  6.1288e-02, -1.8585e-01,  2.8929e-02,  3.2564e-01,\n",
            "        -1.5335e-01,  1.3660e-01, -1.6182e-01,  6.4964e-02, -8.5079e-02,\n",
            "         3.3707e-01,  1.2904e+00, -1.8526e-01, -5.6964e-02,  7.8164e-01,\n",
            "         1.0080e-01,  2.3765e-02,  2.0533e-01,  1.3877e-01,  5.1516e-02,\n",
            "         4.6396e-01, -1.6800e-01, -2.7701e-01,  1.7147e-02,  2.7413e-01,\n",
            "         3.0418e-01, -4.6829e-01, -3.5302e-01, -1.1806e-01,  3.0656e-01,\n",
            "        -9.3169e-02,  3.6552e-01, -7.4699e-02,  1.8674e-01,  1.8196e-01,\n",
            "        -6.0726e-01,  1.5790e-01,  3.6945e-02,  3.8616e-01, -1.1270e-01,\n",
            "        -3.0217e-01, -1.8507e-01, -7.9816e-02, -2.0480e-01,  1.3618e-01,\n",
            "         3.3508e-01, -7.0401e-01,  1.1342e-01,  5.1624e-01, -2.8807e-01,\n",
            "         1.7304e-01,  1.3394e-02,  1.2725e-01,  5.6979e-02, -2.6189e-02,\n",
            "        -6.6550e-02,  1.8514e-01, -3.4891e-01, -4.2338e-01,  2.9801e-01,\n",
            "         4.8908e-01, -1.7944e-02,  3.9998e-03, -3.8487e-02, -1.9081e-01,\n",
            "        -2.0702e-01, -1.0034e-01, -2.4193e-01,  2.3699e-02, -7.9897e-02,\n",
            "        -1.0692e-01, -2.1348e-01,  7.8400e-02,  2.7185e-01, -1.1951e-01,\n",
            "         1.8430e-01,  3.9183e-01,  6.9986e-01, -1.4614e-01, -2.3617e-01,\n",
            "         1.6599e-01,  4.2184e-01, -1.1378e-01,  5.6512e-01, -9.2312e-02,\n",
            "        -3.2201e-01,  6.7527e-02,  1.3171e-01, -2.8953e-01,  3.3865e-01,\n",
            "        -1.3611e-01, -2.4904e-02, -3.5928e-01, -2.3149e-01,  2.5027e-01,\n",
            "         1.8812e-01,  1.3930e-01, -4.6541e-01,  6.1997e-04,  3.6775e-01,\n",
            "         3.5735e-02,  3.9766e-01, -2.2289e-02,  5.7645e-01, -2.0697e-01,\n",
            "         3.9778e-02, -5.8186e-01, -6.1854e-02,  9.0533e-03, -2.0996e-02,\n",
            "        -4.0791e-01, -2.3533e-01,  3.0178e-01,  6.1441e-03,  5.4380e-01,\n",
            "         6.4665e-01, -1.2623e-01, -5.4681e-01, -4.0566e-01, -1.8247e-01,\n",
            "        -1.6767e-01, -3.8489e-02,  1.8041e-01,  1.3742e-01, -7.7425e-02,\n",
            "        -5.4124e-01, -1.6580e-01,  6.3397e-03,  1.9836e-01, -2.2218e-01,\n",
            "         5.9302e-01,  2.6311e-02,  1.6570e-01,  3.7648e-01,  2.9796e-01,\n",
            "        -6.6302e-03,  4.6312e-01, -1.5681e-02, -3.2446e-01,  8.8810e-02,\n",
            "         1.0636e-01, -2.4469e-01, -1.0515e-01, -7.2501e-01,  2.9036e-01,\n",
            "         4.8569e-02,  6.8871e-01,  3.9095e-01])\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "Now the experiments are processed into embeddings. We first read the randomly selected files and then proceed as with the patient description."
      ],
      "metadata": {
        "id": "k-jqqJehW_Cw"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "folder_path = '/content/trials'\n",
        "os.chdir(folder_path)\n",
        "xml_files = [f for f in os.listdir() if f.endswith('.xml')]"
      ],
      "metadata": {
        "id": "dyDOlqVkZcrz"
      },
      "execution_count": 48,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "trials_embedding_inclusion = {}\n",
        "trials_embedding_exclusion = {}"
      ],
      "metadata": {
        "id": "WeX6hDTbZmw4"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "The result of extracting the entities of each trial is output."
      ],
      "metadata": {
        "id": "YXD0ENuQEuFc"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "for xml_file in xml_files:\n",
        "    tree = ET.parse(xml_file)\n",
        "    root = tree.getroot()\n",
        "\n",
        "    text = extract_criteria_text(xml_file)\n",
        "    model.to(\"cpu\")\n",
        "    try:\n",
        "      print(xml_file)\n",
        "      list1_tags = get_sentence_tags(text)\n",
        "    except:\n",
        "      print(xml_file)\n",
        "      continue\n",
        "    liste2_tokens = text.split()\n",
        "\n",
        "    ergebnis_dict = {}\n",
        "    for key, value in zip(list1_tags, liste2_tokens):\n",
        "        if key in ergebnis_dict:\n",
        "            ergebnis_dict[key].append(value)\n",
        "        else:\n",
        "            ergebnis_dict[key] = [value]\n",
        "    if 'O' in ergebnis_dict:\n",
        "      del ergebnis_dict['O']\n",
        "\n",
        "    print(ergebnis_dict)\n",
        "\n",
        "    vector_dict_patient = {}\n",
        "    for key, values in ergebnis_dict.items():\n",
        "        vectors = create_word_vectors(values)\n",
        "        if vectors:\n",
        "            averaged_vector = np.mean(vectors, axis=0)\n",
        "            vector_dict_patient[key] = averaged_vector\n",
        "\n",
        "    vector_list = [torch.tensor(v) if isinstance(v, np.ndarray) else v for v in vector_dict_patient.values()]\n",
        "\n",
        "    average_vector_patient = torch.stack(vector_list).mean(dim=0)\n",
        "\n",
        "    trials_embedding_inclusion.update({xml_file: average_vector_patient})"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "gSFGT7GTZi0B",
        "outputId": "990c6bd5-cd28-4219-c94e-f84f54f8a7dd"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "NCT00623454.xml\n",
            "{'Biological_structure': ['chest', 'chest', 'chest'], 'Sign_symptom': ['pain', 'palpitation', 'pain', 'palpitations', 'pain', 'palpitation', 'pain'], 'Date': ['6', 'month', 'ago', '18', 'and', '65', 'years', '6', 'month'], 'Nonbiological_location': ['Cardiological', 'Out-patient', 'Hospital', 'Cardiological', 'Out-patient'], 'Therapeutic_procedure': ['coping', 'course', 'coping', 'cognitive', 'behaviour'], 'Detailed_description': ['three', 'sessions']}\n",
            "NCT01162902.xml\n",
            "{'Detailed_description': ['Stable'], 'Disease_disorder': ['angina'], 'Sign_symptom': ['lesions'], 'Diagnostic_procedure': ['angiography'], 'Medication': ['PCI', 'therapy', 'calcium-channel', 'beta']}\n",
            "NCT00356707.xml\n",
            "{'Diagnostic_procedure': ['BWHS'], 'Personal_background': ['American']}\n",
            "NCT00000408.xml\n",
            "{'Nonbiological_location': ['States']}\n",
            "NCT00848250.xml\n",
            "{'Family_history': ['Infants', 'children', '(newborn', 'to', '17', 'years', 'of', 'children', 'with', 'congenital', 'heart'], 'Therapeutic_procedure': ['cardiopulmonary', 'bypass'], 'Disease_disorder': ['heart'], 'Medication': ['ACE', 'inhibitor']}\n",
            "NCT00143195.xml\n",
            "{'Date': ['years', 'of', 'age'], 'Detailed_description': ['stable'], 'Disease_disorder': ['angina'], 'Dosage': ['daily'], 'Medication': ['amlodipine', 'isosorbide-5-mononitrate'], 'Biological_structure': ['myocardial']}\n",
            "NCT00162344.xml\n",
            "{'History': ['diabetes', 'at', 'least', '5', 'current'], 'Lab_value': ['elevated'], 'Diagnostic_procedure': ['cholesterol', 'heart', 'imaging'], 'Family_history': ['history', 'of'], 'Detailed_description': ['atypical'], 'Biological_structure': ['chest'], 'Disease_disorder': ['artery', 'disease']}\n",
            "NCT01724567.xml\n",
            "{'Sign_symptom': ['Weight', 'weight', 'loss'], 'Detailed_description': ['low', 'energy', 'very', 'low', 'energy'], 'Diagnostic_procedure': ['blood', 'oxygen'], 'Lab_value': ['normalized', 'high']}\n",
            "NCT01253486.xml\n",
            "{'Disease_disorder': ['Heart'], 'Lab_value': ['improve'], 'Diagnostic_procedure': ['physical', 'blood', 'pressure'], 'Clinical_event': ['visits'], 'Therapeutic_procedure': ['expressive', 'writing']}\n",
            "NCT00809029.xml\n",
            "{'Lab_value': ['normal', 'normal', 'impaired', 'normal'], 'Diagnostic_procedure': ['glucose', 'tolerance', 'glucose'], 'Medication': ['GIP'], 'Date': ['months']}\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "We will proceed the exact same way, just using the exclusion criteria. The only thing we have to adjust is, in the case of not having exclusion criteria, we will add a tensor with 0's, so that there will be no problems when continuing preprocessing the data."
      ],
      "metadata": {
        "id": "mwFd9ZCAFBCM"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "for xml_file in xml_files:\n",
        "    tree = ET.parse(xml_file)\n",
        "    root = tree.getroot()\n",
        "\n",
        "    text = extract_criteria_text_exclusion(xml_file)\n",
        "    model.to(\"cpu\")\n",
        "    list1_tags = get_sentence_tags(text)\n",
        "    liste2_tokens = text.split()\n",
        "\n",
        "    ergebnis_dict = {}\n",
        "    for key, value in zip(list1_tags, liste2_tokens):\n",
        "        if key in ergebnis_dict:\n",
        "            ergebnis_dict[key].append(value)\n",
        "        else:\n",
        "            ergebnis_dict[key] = [value]\n",
        "    if 'O' in ergebnis_dict:\n",
        "      del ergebnis_dict['O']\n",
        "\n",
        "    print(ergebnis_dict)\n",
        "\n",
        "    vector_dict_patient = {}\n",
        "    for key, values in ergebnis_dict.items():\n",
        "        vectors = create_word_vectors(values)\n",
        "        if vectors:\n",
        "            averaged_vector = np.mean(vectors, axis=0)\n",
        "            vector_dict_patient[key] = averaged_vector\n",
        "\n",
        "    try:\n",
        "      vector_list = [torch.tensor(v) if isinstance(v, np.ndarray) else v for v in vector_dict_patient.values()]\n",
        "      average_vector_patient = torch.stack(vector_list).mean(dim=0)\n",
        "      trials_embedding_exclusion.update({xml_file: average_vector_patient})\n",
        "    except:\n",
        "      trials_embedding_exclusion.update({xml_file: torch.zeros(torch.Size([768]))})"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "4WCKr7WAwsED",
        "outputId": "b788b8dc-bee4-4ed1-c8da-2870c4f7e662"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "{'Detailed_description': ['do', 'speak', 'norwegian'], 'History': ['mentally', 'retarded', 'psychosis', 'last', 'alcohol', 'drug', 'misuse'], 'Duration': ['6', 'month']}\n",
            "{'Date': ['within', 'one'], 'Disease_disorder': ['Angina', 'Liver', 'function', 'abnormality', 'renal', 'failure', 'heart', 'uncorrectable', 'hematologic', 'disease', 'pregnant', 'diabetes'], 'History': ['testing', 'Severe', 'Uncontrolled'], 'Diagnostic_procedure': ['life', 'span']}\n",
            "{}\n",
            "{'Therapeutic_procedure': ['Pregnancy', 'Back', 'surgery', 'legal', 'proceedings'], 'Duration': ['past', '6', 'months', 'next', '6', 'months', 'the', 'last', '6', 'months', 'past', 'year'], 'Biological_structure': ['Back', 'Back', 'back', 'back', 'back', 'crotch', 'area', 'back'], 'Sign_symptom': ['pain', 'pain', 'pain', 'Difficulty', 'with', 'pain', 'pain', 'Numbness', 'pain'], 'Disease_disorder': ['car', 'accident', 'injury', 'sciatica', 'systemic', 'disease', 'rheumatic', 'physical', 'or', 'mental', 'health', 'condition', 'sciatica', 'control', 'sciatica', 'sciatica', 'sciatica'], 'Clinical_event': ['involved', 'visits'], 'Nonbiological_location': ['doctor'], 'Date': ['18']}\n",
            "{'Lab_value': ['less', 'than', '3.5', 'kg'], 'Family_history': ['children', 'thought', 'to', 'be', 'noncompliant']}\n",
            "{'Disease_disorder': ['congestive', 'heart', 'cardiovascular'], 'Diagnostic_procedure': ['blood', 'pressure'], 'Lab_value': ['less', 'than'], 'Therapeutic_procedure': ['anti-anginal', 'therapies'], 'Detailed_description': ['sublingual']}\n",
            "{'Biological_structure': ['chest'], 'Sign_symptom': ['pain', 'unable', 'to'], 'Disease_disorder': ['heart', 'disease']}\n",
            "{'Disease_disorder': ['Diabetes', 'Mellitus', 'valve', 'disease', 'heart', 'Chronic', 'obstructive', 'Pulmonary', 'Disease', 'asthma', 'liver', 'disease'], 'Diagnostic_procedure': ['Fasting', 'plasma', 'glucose', 'Hba1c', 'Ejection', 'Fraction'], 'Lab_value': ['7', 'mmol/L'], 'Severity': ['moderate', 'Severe', 'severe'], 'Detailed_description': ['degree', 'more', 'than', '2'], 'Frequency': ['times', 'a'], 'Sign_symptom': ['weight', 'loss', 'weight', 'gain'], 'Date': ['month']}\n",
            "{'Detailed_description': ['manual'], 'Sign_symptom': ['limitations']}\n",
            "{'History': ['severe', 'hepatic', 'or', 'renal', 'disease', 'Current', 'malignant', 'disease', 'alcohol', 'misuse', 'Major', 'psychiatric', 'disease', 'use', 'of', 'major', 'eating', 'disorder', 'bulimia'], 'Disease_disorder': ['Thyroid', 'dysfunction', 'disturbance', 'growth', 'hormone', 'hypoadrenalism'], 'Sign_symptom': ['cortisol']}\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import torch\n",
        "import torch.nn.functional as F\n",
        "import numpy as np"
      ],
      "metadata": {
        "id": "HNjqtYgYFezF"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "`calculate_cosine_similarity` calculates the cosine similarity between the query embedding, which is the embedded single patient description, and a dictionary of embeddings, which is the dictionary of trial descriptions."
      ],
      "metadata": {
        "id": "oPAiza6KFkR5"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def calculate_cosine_similarity(query_embedding, embedding_dict):\n",
        "    similarities = []\n",
        "\n",
        "    for key, value_embedding in embedding_dict.items():\n",
        "        similarity = F.cosine_similarity(query_embedding, value_embedding, dim=0).item()\n",
        "        similarities.append((key, similarity))\n",
        "\n",
        "    similarities.sort(key=lambda x: x[1], reverse=True)\n",
        "    return similarities"
      ],
      "metadata": {
        "id": "Gs3Nz7qvb9ME"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Now, we will compute the similarities between the patient description and the inclusion criteria with detailed description and the summary. The results will be sorted ascending with its similarity value."
      ],
      "metadata": {
        "id": "VKEH8HfYFuOy"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "similarities_in = calculate_cosine_similarity(average_vector_patient, trials_embedding_inclusion)\n",
        "for id, similarity in similarities_in:\n",
        "    match_value = result.loc[result['trial'] == id.replace(\".xml\", \"\"), 'match'].values[0]\n",
        "    print(f'ID: {id}, Ähnlichkeit: {similarity:.4f}, Ground Truth: {match_value}')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "YllMLJ6IzVFq",
        "outputId": "feba54cb-d2e5-43c2-fb51-38393c18d042"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "ID: NCT00162344.xml, Ähnlichkeit: 0.9739, Ground Truth: 1\n",
            "ID: NCT00623454.xml, Ähnlichkeit: 0.9733, Ground Truth: 0\n",
            "ID: NCT00848250.xml, Ähnlichkeit: 0.9722, Ground Truth: 0\n",
            "ID: NCT01724567.xml, Ähnlichkeit: 0.9688, Ground Truth: 2\n",
            "ID: NCT00809029.xml, Ähnlichkeit: 0.9677, Ground Truth: 2\n",
            "ID: NCT01253486.xml, Ähnlichkeit: 0.9675, Ground Truth: 2\n",
            "ID: NCT00143195.xml, Ähnlichkeit: 0.9659, Ground Truth: 2\n",
            "ID: NCT01162902.xml, Ähnlichkeit: 0.9582, Ground Truth: 1\n",
            "ID: NCT00356707.xml, Ähnlichkeit: 0.9332, Ground Truth: 0\n",
            "ID: NCT00000408.xml, Ähnlichkeit: 0.8835, Ground Truth: 0\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "The same thing is done, now only the patient's description is compared with the exclusion criteria. This means that the higher the match, the worse the fit."
      ],
      "metadata": {
        "id": "nMmlnnN-F9F0"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "similarities_ex = calculate_cosine_similarity(average_vector_patient, trials_embedding_exclusion)\n",
        "for id, similarity in similarities_ex:\n",
        "    try:\n",
        "      match_value = result.loc[result['trial'] == id.replace(\".xml\", \"\"), 'match'].values[0]\n",
        "    except:\n",
        "      match_value = -100\n",
        "    print(f'ID: {id}, Ähnlichkeit: {similarity:.4f}, Ground Truth: {match_value}')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "H7mk4cbEcAlS",
        "outputId": "05c81070-885d-4923-d4ba-66fa30ec9ecb"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "ID: NCT00809029.xml, Ähnlichkeit: 1.0000, Ground Truth: 2\n",
            "ID: NCT01724567.xml, Ähnlichkeit: 0.9772, Ground Truth: 2\n",
            "ID: NCT01162902.xml, Ähnlichkeit: 0.9743, Ground Truth: 1\n",
            "ID: NCT00000408.xml, Ähnlichkeit: 0.9743, Ground Truth: 0\n",
            "ID: NCT00143195.xml, Ähnlichkeit: 0.9734, Ground Truth: 2\n",
            "ID: NCT00623454.xml, Ähnlichkeit: 0.9715, Ground Truth: 0\n",
            "ID: NCT00848250.xml, Ähnlichkeit: 0.9669, Ground Truth: 0\n",
            "ID: NCT00162344.xml, Ähnlichkeit: 0.9645, Ground Truth: 1\n",
            "ID: NCT01253486.xml, Ähnlichkeit: 0.9303, Ground Truth: 2\n",
            "ID: NCT00356707.xml, Ähnlichkeit: 0.0000, Ground Truth: 0\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "To take the exclusion criteria into account, a new function is defined.\n",
        "`rank_trials_with_penalty` takes the two lists of trial similarities, one based on inclusion criteria and the other on exclusion criteria, and computes a combined similarity score for each trial. It applies a penalty to trials with higher similarity to the exclusion criteria, controlled by the weight parameter. Finally, it returns a list of ranked trials based on their combined similarity scores, with higher scores indicating a better match to the patient's description."
      ],
      "metadata": {
        "id": "pkW48cwVGgQD"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def rank_trials_with_penalty(similarities_inclusion, similarities_exclusion, weight):\n",
        "    ranked_trials = []\n",
        "\n",
        "    for id, similarity_inclusion in similarities_inclusion:\n",
        "        similarity_exclusion = next((sim for sim_id, sim in similarities_exclusion if sim_id == id), None)\n",
        "\n",
        "        if similarity_exclusion is not None:\n",
        "            weighted_similarity = (1 - weight) * similarity_inclusion + weight * similarity_exclusion\n",
        "            ranked_trials.append((id, weighted_similarity))\n",
        "\n",
        "    ranked_trials.sort(key=lambda x: x[1], reverse=True)\n",
        "\n",
        "    return ranked_trials"
      ],
      "metadata": {
        "id": "ZR4pXsbm2Cob"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "ranked_trials = rank_trials_with_penalty(similarities_in, similarities_ex, weight=0.3)\n",
        "\n",
        "rankings = []\n",
        "relevant_items = []\n",
        "maybe_relevant_items = []\n",
        "\n",
        "for id, weighted_similarity in ranked_trials:\n",
        "    rankings.append(id)\n",
        "    if (match_value > 1):\n",
        "      relevant_items.append(id)\n",
        "    if (match_value > 0):\n",
        "      maybe_relevant_items.append(id)\n",
        "    match_value = result.loc[result['trial'] == id.replace(\".xml\", \"\"), 'match'].values[0]\n",
        "    print(f'ID: {id}, Gewichtete Ähnlichkeit: {weighted_similarity:.4f}, Ground Truth: {match_value}')"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "qR3HPd5-GV4D",
        "outputId": "da049463-0624-420b-f294-5912946874d5"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "ID: NCT00809029.xml, Gewichtete Ähnlichkeit: 0.9774, Ground Truth: 2\n",
            "ID: NCT00623454.xml, Gewichtete Ähnlichkeit: 0.9728, Ground Truth: 0\n",
            "ID: NCT01724567.xml, Gewichtete Ähnlichkeit: 0.9713, Ground Truth: 2\n",
            "ID: NCT00162344.xml, Gewichtete Ähnlichkeit: 0.9710, Ground Truth: 1\n",
            "ID: NCT00848250.xml, Gewichtete Ähnlichkeit: 0.9706, Ground Truth: 0\n",
            "ID: NCT00143195.xml, Gewichtete Ähnlichkeit: 0.9682, Ground Truth: 2\n",
            "ID: NCT01162902.xml, Gewichtete Ähnlichkeit: 0.9630, Ground Truth: 1\n",
            "ID: NCT01253486.xml, Gewichtete Ähnlichkeit: 0.9564, Ground Truth: 2\n",
            "ID: NCT00000408.xml, Gewichtete Ähnlichkeit: 0.9108, Ground Truth: 0\n",
            "ID: NCT00356707.xml, Gewichtete Ähnlichkeit: 0.6533, Ground Truth: 0\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "For evaluating this ranking, we choose normalized Discounted Cumulative Gain. This metric takes into account the relevance of documents and position in the ranking. It rates relevant documents higher and gives them more weight if they are higher in the ranking, which suits our ranking."
      ],
      "metadata": {
        "id": "Q2YVG6tV5cbd"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def discounted_cumulative_gain(rankings, relevant_items):\n",
        "    dcg = 0.0\n",
        "    for i, item in enumerate(rankings):\n",
        "        if item in relevant_items:\n",
        "            dcg += 1.0 / (i + 1)\n",
        "    return dcg\n",
        "\n",
        "def normalized_discounted_cumulative_gain(rankings, relevant_items):\n",
        "    dcg = discounted_cumulative_gain(rankings, relevant_items)\n",
        "    ideal_rankings = sorted(relevant_items, reverse=True)\n",
        "    idcg = discounted_cumulative_gain(ideal_rankings, relevant_items)\n",
        "    return dcg / idcg if idcg > 0 else 0.0"
      ],
      "metadata": {
        "id": "CxysRLp03n2A"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "source": [
        "Because nDCG can not divide between different relevance rankings, we will do two evaluations, one only with really relevant trials, the other one with conditional matches."
      ],
      "metadata": {
        "id": "TKQ64wusG0ZZ"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "ndcg = normalized_discounted_cumulative_gain(rankings, relevant_items)\n",
        "ndcg"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "jWDp2YK04CAQ",
        "outputId": "52fc780e-a670-4b3c-bbf0-ecdca50a11a2"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "0.48190476190476195"
            ]
          },
          "metadata": {},
          "execution_count": 127
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "ndcg = normalized_discounted_cumulative_gain(rankings, maybe_relevant_items)\n",
        "ndcg"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "HBDfxsJZ41Y9",
        "outputId": "16cf909c-78e1-42c1-91fb-d27a0487eba4"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "0.5424360220278588"
            ]
          },
          "metadata": {},
          "execution_count": 128
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "The nDCG values of 0.48 and 0.54 indicate the quality of the ranking, with higher values being better.\n",
        "\n",
        "Further discussion of results can be read in the paper."
      ],
      "metadata": {
        "id": "OHVPrng955Os"
      }
    }
  ]
}