{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"machine_shape": "hm",
"gpuType": "T4",
"toc_visible": true,
"authorship_tag": "ABX9TyMJITJkyU+HQa0z76874fcr",
"include_colab_link": true
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
},
"accelerator": "GPU",
"widgets": {
"application/vnd.jupyter.widget-state+json": {
"7c1400f6c1434c2ab29732bce3b0377f": {
"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_ea99620dd03943bdbb7ac7699ab0d572",
"IPY_MODEL_735d081f4a684c57854f7f2665885f31",
"IPY_MODEL_b1b25e005a4c4df0a96987d923f81388"
],
"layout": "IPY_MODEL_fdbc264d7a1f4fd19622cacd85bf69cc"
}
},
"ea99620dd03943bdbb7ac7699ab0d572": {
"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_7e5421fdbf574727ad84902e21f785b4",
"placeholder": "",
"style": "IPY_MODEL_ff84648ab9204723bea689bf376fde0a",
"value": "100%"
}
},
"735d081f4a684c57854f7f2665885f31": {
"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_de1b4d1af2a44426a81a6ee99b24d5b3",
"max": 10,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_87d9362d232141b988bf6f65eb688169",
"value": 10
}
},
"b1b25e005a4c4df0a96987d923f81388": {
"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_d186deefef4c4ecbb104a1b289304716",
"placeholder": "",
"style": "IPY_MODEL_3570737e05124ca0b6990f1198c62089",
"value": " 10/10 [00:02<00:00, 3.43it/s]"
}
},
"fdbc264d7a1f4fd19622cacd85bf69cc": {
"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
}
},
"7e5421fdbf574727ad84902e21f785b4": {
"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
}
},
"ff84648ab9204723bea689bf376fde0a": {
"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": ""
}
},
"de1b4d1af2a44426a81a6ee99b24d5b3": {
"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
}
},
"87d9362d232141b988bf6f65eb688169": {
"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": ""
}
},
"d186deefef4c4ecbb104a1b289304716": {
"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
}
},
"3570737e05124ca0b6990f1198c62089": {
"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": ""
}
},
"feccd063d54845389a4aeb794c13d0fe": {
"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_5cce98e2c802476c8496cb251825b167",
"IPY_MODEL_2150d389d12b4d909691829382026380",
"IPY_MODEL_f9727d2307c24e4985aa2c10b5f9382f"
],
"layout": "IPY_MODEL_750203722bef4c488fc9b2d065482e0b"
}
},
"5cce98e2c802476c8496cb251825b167": {
"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_c8b77adebd4c4372819533a8c550d196",
"placeholder": "",
"style": "IPY_MODEL_9d245436775d459f82de6dbb6a9e0ab5",
"value": "100%"
}
},
"2150d389d12b4d909691829382026380": {
"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_b94a655654cd4cee81da1f3b073eaae1",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_c75b1077f45b4e17bc33ff4f7118d075",
"value": 3000
}
},
"f9727d2307c24e4985aa2c10b5f9382f": {
"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_1909f638ed4c48f3aecab4736ee02896",
"placeholder": "",
"style": "IPY_MODEL_95d5c89082bc410bb706447463f9451b",
"value": " 3000/3000 [2:05:33<00:00, 1.87s/it]"
}
},
"750203722bef4c488fc9b2d065482e0b": {
"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
}
},
"c8b77adebd4c4372819533a8c550d196": {
"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
}
},
"9d245436775d459f82de6dbb6a9e0ab5": {
"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": ""
}
},
"b94a655654cd4cee81da1f3b073eaae1": {
"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
}
},
"c75b1077f45b4e17bc33ff4f7118d075": {
"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": ""
}
},
"1909f638ed4c48f3aecab4736ee02896": {
"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
}
},
"95d5c89082bc410bb706447463f9451b": {
"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": ""
}
},
"e632b12bcf5a42ffba8de6e209e51802": {
"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_cc51e6e500aa4e0e93cbfb0052c9f36b",
"IPY_MODEL_58b8cf4830df49cf8096d6b4e85bc400",
"IPY_MODEL_daec6c20905e492780d4c19cb7ecf7f6"
],
"layout": "IPY_MODEL_6d6c1e45253248afbd289fe23006c682"
}
},
"cc51e6e500aa4e0e93cbfb0052c9f36b": {
"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_e2061b2f5b22413e8bd6077e0454b9d7",
"placeholder": "",
"style": "IPY_MODEL_3cbb1444ed23412d95b59ba38d291587",
"value": "100%"
}
},
"58b8cf4830df49cf8096d6b4e85bc400": {
"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_bb1e44cc1e404c9fb6b23fd0d6ae79cd",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_ac61e485491d47b18215a578579e1547",
"value": 3000
}
},
"daec6c20905e492780d4c19cb7ecf7f6": {
"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_b35cddc9c45b4dfd98e155f3b3702374",
"placeholder": "",
"style": "IPY_MODEL_99a2a3afc2584a96819ed42f95414625",
"value": " 3000/3000 [2:05:26<00:00, 2.54s/it]"
}
},
"6d6c1e45253248afbd289fe23006c682": {
"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
}
},
"e2061b2f5b22413e8bd6077e0454b9d7": {
"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
}
},
"3cbb1444ed23412d95b59ba38d291587": {
"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": ""
}
},
"bb1e44cc1e404c9fb6b23fd0d6ae79cd": {
"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
}
},
"ac61e485491d47b18215a578579e1547": {
"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": ""
}
},
"b35cddc9c45b4dfd98e155f3b3702374": {
"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
}
},
"99a2a3afc2584a96819ed42f95414625": {
"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": ""
}
},
"d5abd2c22b29411fb1ce9c54a485f4c5": {
"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_90451898cca3479fa93b2cc1e9f14f16",
"IPY_MODEL_6382233329e84818a60968e1c9da6758",
"IPY_MODEL_6d83695c23a34119bbe09b8e83367bbb"
],
"layout": "IPY_MODEL_bf7cb5c3ed0a43bdb0a4be7b78f2ad40"
}
},
"90451898cca3479fa93b2cc1e9f14f16": {
"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_aa505e0a02fe4d249738657862b7a980",
"placeholder": "",
"style": "IPY_MODEL_1d7a151bdac54a92a34e7bdc4c05374b",
"value": "100%"
}
},
"6382233329e84818a60968e1c9da6758": {
"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_c4ee5190d8f94d86b5cc95bd68612ab3",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_175a78d1fe8f481e8fe14ea4d5b18666",
"value": 3000
}
},
"6d83695c23a34119bbe09b8e83367bbb": {
"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_aca036122f924c92ae58502c7d439789",
"placeholder": "",
"style": "IPY_MODEL_8e081c813a8848f3928437bfe7d301dd",
"value": " 3000/3000 [2:05:31<00:00, 2.16s/it]"
}
},
"bf7cb5c3ed0a43bdb0a4be7b78f2ad40": {
"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
}
},
"aa505e0a02fe4d249738657862b7a980": {
"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
}
},
"1d7a151bdac54a92a34e7bdc4c05374b": {
"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": ""
}
},
"c4ee5190d8f94d86b5cc95bd68612ab3": {
"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
}
},
"175a78d1fe8f481e8fe14ea4d5b18666": {
"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": ""
}
},
"aca036122f924c92ae58502c7d439789": {
"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
}
},
"8e081c813a8848f3928437bfe7d301dd": {
"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": ""
}
},
"870d4d61eaeb4aab92f798fbba752253": {
"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_76a61cf5da7f4d4684b0a8c35e4f2ea1",
"IPY_MODEL_cf1c632223bb48e6b4ceeee2e7c45ef6",
"IPY_MODEL_cf2eb502ff914fcebf8a5572118d1e51"
],
"layout": "IPY_MODEL_68ba771bfe164d5f87cf619817f3a8d4"
}
},
"76a61cf5da7f4d4684b0a8c35e4f2ea1": {
"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_00c68158dd2c48baa153116bdbb0a868",
"placeholder": "",
"style": "IPY_MODEL_0810fa11439443dda0e9357a6379d684",
"value": "100%"
}
},
"cf1c632223bb48e6b4ceeee2e7c45ef6": {
"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_e4906a61bb314d0bbb8db5765cd6ad8c",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_1491fdf4681c439896e058a953d5f401",
"value": 3000
}
},
"cf2eb502ff914fcebf8a5572118d1e51": {
"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_8898af43f9294a519fcc740ab1aa6008",
"placeholder": "",
"style": "IPY_MODEL_428b7d5e2d2741e9b86944179aeb75f4",
"value": " 3000/3000 [2:05:17<00:00, 2.45s/it]"
}
},
"68ba771bfe164d5f87cf619817f3a8d4": {
"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
}
},
"00c68158dd2c48baa153116bdbb0a868": {
"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
}
},
"0810fa11439443dda0e9357a6379d684": {
"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": ""
}
},
"e4906a61bb314d0bbb8db5765cd6ad8c": {
"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
}
},
"1491fdf4681c439896e058a953d5f401": {
"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": ""
}
},
"8898af43f9294a519fcc740ab1aa6008": {
"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
}
},
"428b7d5e2d2741e9b86944179aeb75f4": {
"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": ""
}
},
"6b64d5496f8149dc999c47e734dcece8": {
"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_acbb7b7ef8f24c799e30aff0fe4fb433",
"IPY_MODEL_98eb65a5721442d48f4a2b83de7bf761",
"IPY_MODEL_44c2f62dd1b541c1a59eb0e06b9da9fb"
],
"layout": "IPY_MODEL_18de04bd37bd476895e6817f5a6cc210"
}
},
"acbb7b7ef8f24c799e30aff0fe4fb433": {
"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_f6a388f00293415dbd44025879ce0a80",
"placeholder": "",
"style": "IPY_MODEL_c2f5bf52be3a42c2b0b3b93f187b31ab",
"value": "100%"
}
},
"98eb65a5721442d48f4a2b83de7bf761": {
"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_ae1db5c296dc4d3e98fde32a84b2a52a",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_d17a3f960aa14b809660a7f62721aaca",
"value": 3000
}
},
"44c2f62dd1b541c1a59eb0e06b9da9fb": {
"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_0cabbcd580fc45a6a7e23b0a78e155ca",
"placeholder": "",
"style": "IPY_MODEL_8d6ae2d087444473949758a5cfadb2a6",
"value": " 3000/3000 [2:05:30<00:00, 2.23s/it]"
}
},
"18de04bd37bd476895e6817f5a6cc210": {
"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
}
},
"f6a388f00293415dbd44025879ce0a80": {
"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
}
},
"c2f5bf52be3a42c2b0b3b93f187b31ab": {
"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": ""
}
},
"ae1db5c296dc4d3e98fde32a84b2a52a": {
"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
}
},
"d17a3f960aa14b809660a7f62721aaca": {
"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": ""
}
},
"0cabbcd580fc45a6a7e23b0a78e155ca": {
"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
}
},
"8d6ae2d087444473949758a5cfadb2a6": {
"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": ""
}
},
"6c9c1669fa7f4a2e8f5a6aefcee2c112": {
"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_08994674066c463ba2d0046da22823f7",
"IPY_MODEL_0da1db8b08e2438685ebea608982cd4a",
"IPY_MODEL_141cceefd9194286b7d426583099c9c1"
],
"layout": "IPY_MODEL_f4ab5ac84663416cbc7be99100660c3e"
}
},
"08994674066c463ba2d0046da22823f7": {
"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_42649b3598cf4a8aa601df8d9b80df2c",
"placeholder": "",
"style": "IPY_MODEL_ad9138188ac447d18a2bc773d4313614",
"value": "100%"
}
},
"0da1db8b08e2438685ebea608982cd4a": {
"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_240af12c9862407bb6501c51d69c4198",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_3ba185f6a2384745a68d553ce2f19a68",
"value": 3000
}
},
"141cceefd9194286b7d426583099c9c1": {
"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_b0bc8d7f149e4cec84181cbe6ee52196",
"placeholder": "",
"style": "IPY_MODEL_0cfd9fd6ec854e25b7b8908b2135f437",
"value": " 3000/3000 [2:05:28<00:00, 2.30s/it]"
}
},
"f4ab5ac84663416cbc7be99100660c3e": {
"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
}
},
"42649b3598cf4a8aa601df8d9b80df2c": {
"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
}
},
"ad9138188ac447d18a2bc773d4313614": {
"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": ""
}
},
"240af12c9862407bb6501c51d69c4198": {
"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
}
},
"3ba185f6a2384745a68d553ce2f19a68": {
"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": ""
}
},
"b0bc8d7f149e4cec84181cbe6ee52196": {
"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
}
},
"0cfd9fd6ec854e25b7b8908b2135f437": {
"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": ""
}
},
"9bf5bf95dd3146e98760efb4d6328895": {
"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_26ba0a71ace5492080b7eff302ac426d",
"IPY_MODEL_df76c1955c2f4a0c83791c36c94bda9f",
"IPY_MODEL_e30003b7f2a94311844ff86f428dbc74"
],
"layout": "IPY_MODEL_61001bc2877c4eeeae9593ffd83a69bb"
}
},
"26ba0a71ace5492080b7eff302ac426d": {
"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_d5b2d4ea333f4752b77cea478912643b",
"placeholder": "",
"style": "IPY_MODEL_53ef5c383c2d4565b1c3a36b626f91d0",
"value": "100%"
}
},
"df76c1955c2f4a0c83791c36c94bda9f": {
"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_186cb8d023d04bc9ababf86ac0aa01a7",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_6c82c613940e4d19bc7c7f72ea8ae2f6",
"value": 3000
}
},
"e30003b7f2a94311844ff86f428dbc74": {
"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_f9e227e64f7d4a5abfd1e63fa449b41d",
"placeholder": "",
"style": "IPY_MODEL_8598dfcc57fc4378bf4e6c29ac6ece25",
"value": " 3000/3000 [2:05:32<00:00, 1.84s/it]"
}
},
"61001bc2877c4eeeae9593ffd83a69bb": {
"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
}
},
"d5b2d4ea333f4752b77cea478912643b": {
"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
}
},
"53ef5c383c2d4565b1c3a36b626f91d0": {
"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": ""
}
},
"186cb8d023d04bc9ababf86ac0aa01a7": {
"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
}
},
"6c82c613940e4d19bc7c7f72ea8ae2f6": {
"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": ""
}
},
"f9e227e64f7d4a5abfd1e63fa449b41d": {
"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
}
},
"8598dfcc57fc4378bf4e6c29ac6ece25": {
"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": ""
}
},
"a10c22ae98b8446289dfad20a0332c47": {
"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_fbea43164ce3498cad565a95534fe770",
"IPY_MODEL_87a73918c0154c18b6e7f3213418ca87",
"IPY_MODEL_7482228547d44d02ba1a212d23ac620b"
],
"layout": "IPY_MODEL_aa462e5caa6e4b2ebfda965180ff26a1"
}
},
"fbea43164ce3498cad565a95534fe770": {
"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_05ae6ee07e054df490692cfd338b8aa4",
"placeholder": "",
"style": "IPY_MODEL_eea18326c15c453db11d83adbcda42f6",
"value": "100%"
}
},
"87a73918c0154c18b6e7f3213418ca87": {
"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_b5968328c2dd42a9941862a053808597",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_385929b4d4a5413e8bbe3d446997a34a",
"value": 3000
}
},
"7482228547d44d02ba1a212d23ac620b": {
"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_19716db98904411c8a906b111e37bda0",
"placeholder": "",
"style": "IPY_MODEL_6d4b9c3ca98547b4aa070052454bc4c4",
"value": " 3000/3000 [2:05:25<00:00, 2.57s/it]"
}
},
"aa462e5caa6e4b2ebfda965180ff26a1": {
"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
}
},
"05ae6ee07e054df490692cfd338b8aa4": {
"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
}
},
"eea18326c15c453db11d83adbcda42f6": {
"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": ""
}
},
"b5968328c2dd42a9941862a053808597": {
"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
}
},
"385929b4d4a5413e8bbe3d446997a34a": {
"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": ""
}
},
"19716db98904411c8a906b111e37bda0": {
"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
}
},
"6d4b9c3ca98547b4aa070052454bc4c4": {
"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": ""
}
},
"1c62e3407ce24ef287bccdd2ca7d94b0": {
"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_38a3cc3a1e5c42eea9a6fd7743e1d916",
"IPY_MODEL_0d02ae67ce4b439c9234227fc3cebae5",
"IPY_MODEL_013cd883f7cf4e1c8610e5aa5ca52dbe"
],
"layout": "IPY_MODEL_52a9f4e4d2084836913d738db7f26912"
}
},
"38a3cc3a1e5c42eea9a6fd7743e1d916": {
"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_c6f7de90fd6848ac8dd6f910f8ff16c2",
"placeholder": "",
"style": "IPY_MODEL_41b9fbecd1a14383b39368489d13d908",
"value": "100%"
}
},
"0d02ae67ce4b439c9234227fc3cebae5": {
"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_c5c18a1f60a842ac9bea877848266b07",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_214f59f40c474760bf7afada2974679c",
"value": 3000
}
},
"013cd883f7cf4e1c8610e5aa5ca52dbe": {
"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_01c2522f6ff2467180e92d241aac7605",
"placeholder": "",
"style": "IPY_MODEL_91e297e4bbb34d2c99e87aa03767a7ef",
"value": " 3000/3000 [31:55<00:00, 1.59it/s]"
}
},
"52a9f4e4d2084836913d738db7f26912": {
"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
}
},
"c6f7de90fd6848ac8dd6f910f8ff16c2": {
"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
}
},
"41b9fbecd1a14383b39368489d13d908": {
"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": ""
}
},
"c5c18a1f60a842ac9bea877848266b07": {
"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
}
},
"214f59f40c474760bf7afada2974679c": {
"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": ""
}
},
"01c2522f6ff2467180e92d241aac7605": {
"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
}
},
"91e297e4bbb34d2c99e87aa03767a7ef": {
"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": ""
}
},
"499f79ff11d04e3d9c89dd311dbb3a18": {
"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_7c20d1d8e8d340ae806e806d8cebbfa0",
"IPY_MODEL_04154e9c25224695ac5739444a50c158",
"IPY_MODEL_a969196910394454a757bdff842c9542"
],
"layout": "IPY_MODEL_29bcc292a9564f398baa3f7cb4383da0"
}
},
"7c20d1d8e8d340ae806e806d8cebbfa0": {
"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_c671dccb95b348a9a31c97b6918c5166",
"placeholder": "",
"style": "IPY_MODEL_1c428ea82d544f1da4573fc41d077d17",
"value": "100%"
}
},
"04154e9c25224695ac5739444a50c158": {
"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_36d6e7e248f04fd393db82bd23850060",
"max": 3000,
"min": 0,
"orientation": "horizontal",
"style": "IPY_MODEL_2e6b91ab64274d9e8f6bfeb19caa674a",
"value": 3000
}
},
"a969196910394454a757bdff842c9542": {
"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_d522dbab4c8c4dd1858aeb027869da8f",
"placeholder": "",
"style": "IPY_MODEL_50ddd740217a4e99b039b3d45e5ee354",
"value": " 3000/3000 [31:49<00:00, 2.30it/s]"
}
},
"29bcc292a9564f398baa3f7cb4383da0": {
"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
}
},
"c671dccb95b348a9a31c97b6918c5166": {
"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
}
},
"1c428ea82d544f1da4573fc41d077d17": {
"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": ""
}
},
"36d6e7e248f04fd393db82bd23850060": {
"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
}
},
"2e6b91ab64274d9e8f6bfeb19caa674a": {
"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": ""
}
},
"d522dbab4c8c4dd1858aeb027869da8f": {
"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
}
},
"50ddd740217a4e99b039b3d45e5ee354": {
"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",
"metadata": {
"id": "view-in-github",
"colab_type": "text"
},
"source": [
""
]
},
{
"cell_type": "code",
"source": [
"!pip install datasets &> /dev/null"
],
"metadata": {
"id": "NvKAnofUunGq"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "markdown",
"source": [
"#In-silico directed evolution using Metropolis–Hastings Markov chain Monte Carlo sampling"
],
"metadata": {
"id": "p_GWodsj3uKP"
}
},
{
"cell_type": "markdown",
"source": [
"In the notebooks before this one, we tackled the challenge of understanding the relationship between protein sequences and their functions. We took two different paths: [first notebook](https://github.com/arjan-hada/protein-variant-prediction/blob/master/00_protein_seq2fxn_LLM.ipynb), training a regressor using fixed embeddings of our sequences from a protein language model in transformers, and [second notebook](https://github.com/arjan-hada/protein-variant-prediction/blob/master/01_protein_seq2fxn_finetune_transformers.ipynb), fine-tuning a pretrained protein transformer model with our DMS data.\n",
"\n",
"Now, we're diving into the realm of in-silico directed evolution to navigate the vast protein sequence landscape. Think of it like using a surrogate fitness landscape created by our earlier sequence-to-function model. Our mission here is to implement the Metropolis–Hastings Markov chain Monte Carlo algorithm. This fancy term essentially means we're using a computational method for simulated evolution to propose potentially optimized protein designs.\n",
"\n",
"**Goals**\n",
"- Dive into the Metropolis–Hastings Markov chain Monte Carlo (MH MCMC) algorithm for in-silico directed evolution on our surrogate fitness landscape."
],
"metadata": {
"id": "XYT6DGubs59i"
}
},
{
"cell_type": "markdown",
"source": [
"## Metropolis– Hastings Markov chain Monte Carlo algorithm\n",
"\n",
"Algorithm description inspired from: https://github.com/SuperSecretBioTech/evolutionary_monte_carlo_search and https://www.nature.com/articles/s41592-021-01100-y"
],
"metadata": {
"id": "FPL6goGcKisu"
}
},
{
"cell_type": "markdown",
"source": [
"In the Metropolis-Hastings algorithm, especially when applied to scenarios like in silico directed evolution, the acceptance probability when proposing a move from a current state $s_i$ with fitness $f(s_i)$ to a new state $s_{i+1}$ with fitness $f(s_{i+1})$ is typically expressed as:\n",
"\n",
"$$\n",
"\\text{Accept with probability } \\min\\left(1, \\exp\\left(\\frac{f(s_{i+1}) - f(s_i)}{T}\\right)\\right)\n",
"$$\n",
"\n",
"For our in silico directed evolution algorithm we need:\n",
"- a. An initial sequence.\n",
"- b. A sequence-to-function model that predicts the quantitative function or fitness of an amino acid sequence.\n",
"- c. Temperature, T\n",
"- d. Trust radius (the number of mutations relative to wild-type allowed in proposed designs).\n",
"\n",
"The protocol for the algorithm is as follow:\n",
"\n",
"1. Start with a state sequence $s_{i}$ with fitness $f(s_i)$. Set state sequence s equal to a provided initial sequence.\n",
"2. Propose a new sequence $s_{i+1}$ with fitness $f(s_{i+1})$. The new sequence is proposed by randomly adding m ~ Poisson(μ − 1) + 1 mutations to $s_{i}$, where μ is the sequence proposal mutation rate. The sequence proposal mutation rate μ for each trajectory was set to be a random draw from a uniform(1, 2.5) distribution.\n",
"3. If $f(s_{i+1})$ > $f(s_i)$, we accept the new sequence and the algorithm starts over. In this scenario $f(s_{i+1})$ - $f(s_i)$ is positive, making $exp\\left(\\frac{f_{s+1} - f_{s}}{T}\\right)$ greater than 1. Here $, min(1,value) $ ensures the probability doesn't exceed 1.\n",
"4. If $f(s_{i+1}) < f(s_i)$, the expression $\\exp\\left(\\frac{f(s_{i+1}) - f(s_i)}{T}\\right)$ results in a value between 0 and 1, representing the probability of accepting a move to a less fit state.\n",
"\n",
" Note that, if the sequence proposal has more mutations than the input trust radius, its predicted fitness is set, post hoc, to negative infinity, thereby forcing rejection of the proposal.\n",
"\n",
"5. Iterate steps 3 and 4 for a predetermined number of iterations.\n",
"\n",
"\n"
],
"metadata": {
"id": "CEIGKyfNF6M8"
}
},
{
"cell_type": "markdown",
"source": [
"Metropolis–Hastings Markov chain Monte Carlo (MH_MCMC) has two main **hyperparameters: temperature and the number of iterations**. Typically, we determine the number of iterations by experimenting with different values and visually inspecting the fitness-versus-iterations plot to find a convergence point with acceptable fitness.\n",
"\n",
"The temperature (called the temperature because that's what the term represents in the boltzmann distribution of energy states at a finite temperature) is the hyperparameter which decides how liberally non-optimal moves are accepted.\n",
"\n",
"The temperature, named so because of its role in the Boltzmann distribution, dictates how generously we accept non-optimal moves. **High temperatures** increase acceptance probabilities, leading to overly explorative trajectories lingering in low predicted fitness areas. Excessively high temperatures, accepting more than 50-60% non-optimal moves, hinder convergence as the algorithm struggles to reject moves near an optimum, favoring broad searches.\n",
"\n",
"Conversely, **low temperatures** reduce acceptance probabilities, resulting in overly exploitative trajectories with consistently increasing fitness traces. Lower temperatures make MH_MCMC more conservative, strictly accepting moves that marginally increase or decrease fitness. This facilitates movement towards an optimum but limits exploration to directions where fitness only increases, akin to an unguided steepest descent algorithm.\n",
"\n",
"Selecting the right temperature is crucial, balancing search depth (exploitation) and breadth (exploration). A common guideline is choosing a temperature that leads to convergence while accepting about 15-20% non-optimal moves during the simulation, though this can vary depending on the problem.\n",
"\n",
"MH_MCMC's strength lies in its simplicity, low computational cost, and ease of debugging/analysis. However, its effectiveness heavily relies on the initial sequence and the local landscape around it, determining whether convergence to a satisfactory fitness occurs."
],
"metadata": {
"id": "fxBqcozXKyUz"
}
},
{
"cell_type": "markdown",
"source": [
"## Import and setup"
],
"metadata": {
"id": "UXtEl9rc1PD2"
}
},
{
"cell_type": "code",
"source": [
"import pickle\n",
"import numpy as np\n",
"\n",
"import pandas as pd\n",
"\n",
"import matplotlib.pyplot as plt\n",
"import seaborn as sns\n",
"sns.set_context(context=\"paper\", font_scale=1.5)"
],
"metadata": {
"id": "R8ygnjCe7sjQ"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "markdown",
"source": [
"Load the protein sequence-to-function model we built in my previous notebook [Learning sequence-to-function relationship using language models](https://github.com/arjan-hada/protein-variant-prediction/blob/master/00_protein_seq2fxn_LLM.ipynb)."
],
"metadata": {
"id": "1vgKIfmN4wAJ"
}
},
{
"cell_type": "code",
"source": [
"# load sklearn model we built previously\n",
"with open('models/bag_emb_esm1v_rep7868aav2.pkl', 'rb') as f: bag_emb_esm1v = pickle.load(f)"
],
"metadata": {
"id": "wfJuZ-Gl7Eqo"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"source": [
"# Check to ensure md5sum of the model is b2adfa4cf9be0b8614ab2b0c6aeee622\n",
"!md5sum models/bag_emb_esm1v_rep7868aav2.pkl"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "baD-hs3v-GFF",
"outputId": "b3c77754-a050-4228-df46-dca2bf71992e"
},
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"b2adfa4cf9be0b8614ab2b0c6aeee622 models/bag_emb_esm1v_rep7868aav2.pkl\n"
]
}
]
},
{
"cell_type": "markdown",
"source": [
"## Basic sampling"
],
"metadata": {
"id": "4xuaLuxQ1gWR"
}
},
{
"cell_type": "markdown",
"source": [
"Here we will run one evolutionary trajectory on rep78 sequence using the MH MCMC algorithm described in the beginning. Each evolutionary trajectory consists of thousands of iterations depending on the length of protein. For basic sampling we need:\n",
" - starting sequence\n",
" - sequence-to-function-model\n",
" - Monte Carlo sampling alongside the Metropolis-Hastings criteria to score and rank-order new sequences"
],
"metadata": {
"id": "bJI4T6jy3C9Z"
}
},
{
"cell_type": "markdown",
"source": [
"### Starting sequence"
],
"metadata": {
"id": "kZSnsFjv4GpR"
}
},
{
"cell_type": "code",
"source": [
"# Wild type sequence of Rep78 from AAV2\n",
"rep78_seq = 'MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLNLIEQAPLTVAEKLQRDFLTEWRRVSKAPEALFFVQFEKGESYFHMHVLVETTGVKSMVLGRFLSQIREKLIQRIYRGIEPTLPNWFAVTKTRNGAGGGNKVVDECYIPNYLLPKTQPELQWAWTNMEQYLSACLNLTERKRLVAQHLTHVSQTQEQNKENQNPNSDAPVIRSKTSARYMELVGWLVDKGITSEKQWIQEDQASYISFNAASNSRSQIKAALDNAGKIMSLTKTAPDYLVGQQPVEDISSNRIYKILELNGYDPQYAASVFLGWATKKFGKRNTIWLFGPATTGKTNIAEAIAHTVPFYGCVNWTNENFPFNDCVDKMVIWWEEGKMTAKVVESAKAILGGSKVRVDQKCKSSAQIDPTPVIVTSNTNMCAVIDGNSTTFEHQQPLQDRMFKFELTRRLDHDFGKVTKQEVKDFFRWAKDHVVEVEHEFYVKKGGAKKRPAPSDADISEPKRVRESVAQPSTSDAEASINYADRYQNKCSRHVGMNLMLFPCRQCERMNQNSNICFTHGQKDCLECFPVSESQPVSVVKKAYQKLCYIHHIMGKVPDACTACDLVNVDLDDCIFEQ'\n",
"len(rep78_seq)"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "9puJIhBfa8Il",
"outputId": "fca9c72d-65e1-45ed-f61c-5fb972caa4ab"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
"621"
]
},
"metadata": {},
"execution_count": 5
}
]
},
{
"cell_type": "code",
"source": [
"# change M225G. This will be considered wild type sequence from here on. Python string is zero-indexed\n",
"wt_seq = rep78_seq[:224] + 'G' + rep78_seq[225:]\n",
"len(wt_seq) # ensure length remains same after substitution"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "EQsbvwQFa-YY",
"outputId": "34749035-a53d-4913-8a2c-5e2c1600d138"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
"621"
]
},
"metadata": {},
"execution_count": 6
}
]
},
{
"cell_type": "markdown",
"source": [
"### Sequence to function model"
],
"metadata": {
"id": "FbF9KuWk4KA2"
}
},
{
"cell_type": "code",
"source": [
"esm1v_checkpoint = \"facebook/esm1v_t33_650M_UR90S_1\"\n",
"# The AutoTokenizer class automatically retrieve the model's configuration, pretrained weights,\n",
"# or vocabulary from the name of the checkpoint.\n",
"from transformers import AutoTokenizer\n",
"tokenizer = AutoTokenizer.from_pretrained(esm1v_checkpoint)\n",
"\n",
"# Similar to the AutoTokenizer class, AutoModel has a from_pretrained() method\n",
"# to load the weights of a pretrained model\n",
"from transformers import AutoModel\n",
"import torch\n",
"\n",
"device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
"model = AutoModel.from_pretrained(esm1v_checkpoint).to(device)\n",
"\n",
"def get_esm1v_rep(rep78_variant):\n",
" \"\"\"\n",
" Converts rep78 protein sequence variant to fixed dimensional vector\n",
" representation using ESM1v model.\n",
"\n",
" rep78 sequence variant -> token encodings -> token embeddings ->\n",
" last hidden state -> mean -> fixed-dimensional vector representation\n",
" \"\"\"\n",
" # encode string and convert the tokens to Pytorch tensors\n",
" tokz = tokenizer(rep78_variant, return_tensors=\"pt\")\n",
"\n",
" # Place model inputs on the GPU\n",
" inputs = {k:v.to(device) for k,v in tokz.items()\n",
" if k in tokenizer.model_input_names}\n",
"\n",
" # Extract last hidden states\n",
" with torch.no_grad(): # disable automatic calculation of the gradient for inference\n",
" last_hidden_state = model(**inputs).last_hidden_state\n",
"\n",
" # Return vector for [CLS] token and place the final hidden state back on the CPU as NumPy array\n",
" return torch.mean(last_hidden_state, dim=1).cpu().numpy()"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "lc_DWkj7OTSN",
"outputId": "1d6f8448-f75e-4cb6-e988-937ab3154aaf"
},
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stderr",
"text": [
"/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_token.py:88: UserWarning: \n",
"The secret `HF_TOKEN` does not exist in your Colab secrets.\n",
"To authenticate with the Hugging Face Hub, create a token in your settings tab (https://huggingface.co/settings/tokens), set it as secret in your Google Colab and restart your session.\n",
"You will be able to reuse this secret in all of your notebooks.\n",
"Please note that authentication is recommended but still optional to access public models or datasets.\n",
" warnings.warn(\n",
"Some weights of EsmModel were not initialized from the model checkpoint at facebook/esm1v_t33_650M_UR90S_1 and are newly initialized: ['esm.pooler.dense.weight', 'esm.contact_head.regression.bias', 'esm.pooler.dense.bias', 'esm.contact_head.regression.weight']\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": [
"def scoring_func(sequence: str):\n",
" \"\"\"\n",
" scoring function that takes in a string protein sequence, and\n",
" outputs a fitness value.\n",
" \"\"\"\n",
" # protein sequence to fixed dimensional vector representation using ESM1v model\n",
" seq_repr = get_esm1v_rep(sequence)\n",
" # fixed dimensional vector representation to fitness value\n",
" pred = bag_emb_esm1v.predict(seq_repr)\n",
" return pred[0]"
],
"metadata": {
"id": "EPczAXztVtn8"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"source": [
"scoring_func(wt_seq)"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "ilXXM07SVtk7",
"outputId": "aa8f4e99-2a54-4812-92b7-9626c41450b7"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
"0.79098165"
]
},
"metadata": {},
"execution_count": 9
}
]
},
{
"cell_type": "code",
"source": [
"scoring_func(rep78_seq)"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "0lpPgZgV2WO_",
"outputId": "2f4361d2-4e3f-47b7-eb68-16dd9c87a15d"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
"0.92150164"
]
},
"metadata": {},
"execution_count": 10
}
]
},
{
"cell_type": "markdown",
"source": [
"### Monte Carlo sampling alongside the Metropolis-Hastings criteria"
],
"metadata": {
"id": "11eb-LYS6ed2"
}
},
{
"cell_type": "code",
"source": [
"from scripts.mh_mcmc_sampler import sample_one_chain"
],
"metadata": {
"id": "afk1lytHSDmg"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"source": [
"starter_seq = rep78_seq\n",
"sampled_sequences = sample_one_chain(rep78_seq,\n",
" starter_seq,\n",
" n_steps=10, # n_steps is a hyperparameter\n",
" scoring_func=scoring_func,\n",
" is_accepted_kwargs={\"temperature\":0.01},\n",
" trust_radius=3)\n",
"sampled_seqs_df = pd.DataFrame(sampled_sequences)\n",
"sampled_seqs_df"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 426,
"referenced_widgets": [
"7c1400f6c1434c2ab29732bce3b0377f",
"ea99620dd03943bdbb7ac7699ab0d572",
"735d081f4a684c57854f7f2665885f31",
"b1b25e005a4c4df0a96987d923f81388",
"fdbc264d7a1f4fd19622cacd85bf69cc",
"7e5421fdbf574727ad84902e21f785b4",
"ff84648ab9204723bea689bf376fde0a",
"de1b4d1af2a44426a81a6ee99b24d5b3",
"87d9362d232141b988bf6f65eb688169",
"d186deefef4c4ecbb104a1b289304716",
"3570737e05124ca0b6990f1198c62089"
]
},
"id": "ytqvFtMagbee",
"outputId": "672e47a9-e4bf-41ca-c130-028568638acc"
},
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/10 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "7c1400f6c1434c2ab29732bce3b0377f"
}
},
"metadata": {}
},
{
"output_type": "execute_result",
"data": {
"text/plain": [
" sequences scores accept\n",
"0 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.921502 True\n",
"1 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.943941 True\n",
"2 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.833681 False\n",
"3 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.875992 False\n",
"4 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.576772 False\n",
"5 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.732222 False\n",
"6 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.749478 False\n",
"7 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.051466 True\n",
"8 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.179487 True\n",
"9 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.078296 False\n",
"10 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.179487 True"
],
"text/html": [
"\n",
"
\n",
"
\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" sequences | \n",
" scores | \n",
" accept | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.921502 | \n",
" True | \n",
"
\n",
" \n",
" 1 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.943941 | \n",
" True | \n",
"
\n",
" \n",
" 2 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.833681 | \n",
" False | \n",
"
\n",
" \n",
" 3 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.875992 | \n",
" False | \n",
"
\n",
" \n",
" 4 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.576772 | \n",
" False | \n",
"
\n",
" \n",
" 5 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.732222 | \n",
" False | \n",
"
\n",
" \n",
" 6 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.749478 | \n",
" False | \n",
"
\n",
" \n",
" 7 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.051466 | \n",
" True | \n",
"
\n",
" \n",
" 8 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.179487 | \n",
" True | \n",
"
\n",
" \n",
" 9 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.078296 | \n",
" False | \n",
"
\n",
" \n",
" 10 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.179487 | \n",
" True | \n",
"
\n",
" \n",
"
\n",
"
\n",
"
\n",
"
\n"
]
},
"metadata": {},
"execution_count": 12
}
]
},
{
"cell_type": "markdown",
"source": [
"For us a temperature of 0.01 yielded good trajectory behavior - fitness traces improved on average but were not monotonic. "
],
"metadata": {
"id": "EdKplDqA9YaO"
}
},
{
"cell_type": "code",
"source": [
"from functools import partial\n",
"def find_mutated_aa(best_seq, starter_seq):\n",
" \"\"\"Return aminoacid substitution between two protein sequences of the same length\"\"\"\n",
" mutated_aa = [starter_seq[i]+str(i+1)+best_seq[i] for i in range(len(starter_seq))\n",
" if starter_seq[i] != best_seq[i]]\n",
" return mutated_aa\n",
"\n",
"mut_in_rep78 = partial(find_mutated_aa, starter_seq=rep78_seq)"
],
"metadata": {
"id": "VPAJfK-KiSyy"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"source": [
"sampled_seqs_df['mutated_aa'] = sampled_seqs_df['sequences'].apply(mut_in_rep78)\n",
"sampled_seqs_df"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 394
},
"id": "6XA4i5vV2qU4",
"outputId": "8721b25c-2108-4623-a7d5-6ce54eddee11"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
" sequences scores accept \\\n",
"0 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.921502 True \n",
"1 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.943941 True \n",
"2 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.833681 False \n",
"3 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.875992 False \n",
"4 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.576772 False \n",
"5 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.732222 False \n",
"6 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 0.749478 False \n",
"7 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.051466 True \n",
"8 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.179487 True \n",
"9 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.078296 False \n",
"10 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.179487 True \n",
"\n",
" mutated_aa \n",
"0 [] \n",
"1 [N556I] \n",
"2 [L454F, N556I] \n",
"3 [Q241H, N556I] \n",
"4 [I48S, N139V, L157D, N556I] \n",
"5 [Q244V, W319L, N556I] \n",
"6 [Q410G, N556I] \n",
"7 [D528T, N556I] \n",
"8 [L47V, D528T, N556I] \n",
"9 [L47V, S432E, D528T, N556I] \n",
"10 [L47V, D528T, N556I] "
],
"text/html": [
"\n",
" \n",
"
\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" sequences | \n",
" scores | \n",
" accept | \n",
" mutated_aa | \n",
"
\n",
" \n",
" \n",
" \n",
" 0 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.921502 | \n",
" True | \n",
" [] | \n",
"
\n",
" \n",
" 1 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.943941 | \n",
" True | \n",
" [N556I] | \n",
"
\n",
" \n",
" 2 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.833681 | \n",
" False | \n",
" [L454F, N556I] | \n",
"
\n",
" \n",
" 3 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.875992 | \n",
" False | \n",
" [Q241H, N556I] | \n",
"
\n",
" \n",
" 4 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.576772 | \n",
" False | \n",
" [I48S, N139V, L157D, N556I] | \n",
"
\n",
" \n",
" 5 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.732222 | \n",
" False | \n",
" [Q244V, W319L, N556I] | \n",
"
\n",
" \n",
" 6 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 0.749478 | \n",
" False | \n",
" [Q410G, N556I] | \n",
"
\n",
" \n",
" 7 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.051466 | \n",
" True | \n",
" [D528T, N556I] | \n",
"
\n",
" \n",
" 8 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.179487 | \n",
" True | \n",
" [L47V, D528T, N556I] | \n",
"
\n",
" \n",
" 9 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.078296 | \n",
" False | \n",
" [L47V, S432E, D528T, N556I] | \n",
"
\n",
" \n",
" 10 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.179487 | \n",
" True | \n",
" [L47V, D528T, N556I] | \n",
"
\n",
" \n",
"
\n",
"
\n",
"
\n",
"
\n"
]
},
"metadata": {},
"execution_count": 14
}
]
},
{
"cell_type": "markdown",
"source": [
"## Parallel sampling"
],
"metadata": {
"id": "aXp2sZK46sM9"
}
},
{
"cell_type": "markdown",
"source": [
"Here we will run many evolutionary trajectories in parallel."
],
"metadata": {
"id": "PZCGslPC7k1O"
}
},
{
"cell_type": "code",
"source": [
"# for a given sequence-to-function model, run 3,500 evolutionary trajectories in parallel for 3,000 iterations\n",
"from scripts.mh_mcmc_sampler import init_traj_seq, sample_one_chain\n",
"import dask\n",
"\n",
"starter_seq = rep78_seq\n",
"\n",
"delayed_results = []\n",
"\n",
"for i in range(10): # sample 3500 independent chains\n",
" #starter_seq = init_traj_seq(rep78_seq, trust_radius=3)\n",
" starter_seq = rep78_seq\n",
" sampled_sequences = dask.delayed(sample_one_chain)(rep78_seq,\n",
" starter_seq,\n",
" n_steps=3000, # 3000 iterations\n",
" scoring_func=scoring_func,\n",
" is_accepted_kwargs={\"temperature\":0.01},\n",
" trust_radius=3)\n",
" delayed_results.append(sampled_sequences)\n",
"\n",
"results = dask.compute(*delayed_results)\n",
"\n",
"# Convert everything into a single DataFrame\n",
"sampled_sequences_df = pd.concat([pd.DataFrame(r) for r in results])"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 337,
"referenced_widgets": [
"feccd063d54845389a4aeb794c13d0fe",
"5cce98e2c802476c8496cb251825b167",
"2150d389d12b4d909691829382026380",
"f9727d2307c24e4985aa2c10b5f9382f",
"750203722bef4c488fc9b2d065482e0b",
"c8b77adebd4c4372819533a8c550d196",
"9d245436775d459f82de6dbb6a9e0ab5",
"b94a655654cd4cee81da1f3b073eaae1",
"c75b1077f45b4e17bc33ff4f7118d075",
"1909f638ed4c48f3aecab4736ee02896",
"95d5c89082bc410bb706447463f9451b",
"e632b12bcf5a42ffba8de6e209e51802",
"cc51e6e500aa4e0e93cbfb0052c9f36b",
"58b8cf4830df49cf8096d6b4e85bc400",
"daec6c20905e492780d4c19cb7ecf7f6",
"6d6c1e45253248afbd289fe23006c682",
"e2061b2f5b22413e8bd6077e0454b9d7",
"3cbb1444ed23412d95b59ba38d291587",
"bb1e44cc1e404c9fb6b23fd0d6ae79cd",
"ac61e485491d47b18215a578579e1547",
"b35cddc9c45b4dfd98e155f3b3702374",
"99a2a3afc2584a96819ed42f95414625",
"d5abd2c22b29411fb1ce9c54a485f4c5",
"90451898cca3479fa93b2cc1e9f14f16",
"6382233329e84818a60968e1c9da6758",
"6d83695c23a34119bbe09b8e83367bbb",
"bf7cb5c3ed0a43bdb0a4be7b78f2ad40",
"aa505e0a02fe4d249738657862b7a980",
"1d7a151bdac54a92a34e7bdc4c05374b",
"c4ee5190d8f94d86b5cc95bd68612ab3",
"175a78d1fe8f481e8fe14ea4d5b18666",
"aca036122f924c92ae58502c7d439789",
"8e081c813a8848f3928437bfe7d301dd",
"870d4d61eaeb4aab92f798fbba752253",
"76a61cf5da7f4d4684b0a8c35e4f2ea1",
"cf1c632223bb48e6b4ceeee2e7c45ef6",
"cf2eb502ff914fcebf8a5572118d1e51",
"68ba771bfe164d5f87cf619817f3a8d4",
"00c68158dd2c48baa153116bdbb0a868",
"0810fa11439443dda0e9357a6379d684",
"e4906a61bb314d0bbb8db5765cd6ad8c",
"1491fdf4681c439896e058a953d5f401",
"8898af43f9294a519fcc740ab1aa6008",
"428b7d5e2d2741e9b86944179aeb75f4",
"6b64d5496f8149dc999c47e734dcece8",
"acbb7b7ef8f24c799e30aff0fe4fb433",
"98eb65a5721442d48f4a2b83de7bf761",
"44c2f62dd1b541c1a59eb0e06b9da9fb",
"18de04bd37bd476895e6817f5a6cc210",
"f6a388f00293415dbd44025879ce0a80",
"c2f5bf52be3a42c2b0b3b93f187b31ab",
"ae1db5c296dc4d3e98fde32a84b2a52a",
"d17a3f960aa14b809660a7f62721aaca",
"0cabbcd580fc45a6a7e23b0a78e155ca",
"8d6ae2d087444473949758a5cfadb2a6",
"6c9c1669fa7f4a2e8f5a6aefcee2c112",
"08994674066c463ba2d0046da22823f7",
"0da1db8b08e2438685ebea608982cd4a",
"141cceefd9194286b7d426583099c9c1",
"f4ab5ac84663416cbc7be99100660c3e",
"42649b3598cf4a8aa601df8d9b80df2c",
"ad9138188ac447d18a2bc773d4313614",
"240af12c9862407bb6501c51d69c4198",
"3ba185f6a2384745a68d553ce2f19a68",
"b0bc8d7f149e4cec84181cbe6ee52196",
"0cfd9fd6ec854e25b7b8908b2135f437",
"9bf5bf95dd3146e98760efb4d6328895",
"26ba0a71ace5492080b7eff302ac426d",
"df76c1955c2f4a0c83791c36c94bda9f",
"e30003b7f2a94311844ff86f428dbc74",
"61001bc2877c4eeeae9593ffd83a69bb",
"d5b2d4ea333f4752b77cea478912643b",
"53ef5c383c2d4565b1c3a36b626f91d0",
"186cb8d023d04bc9ababf86ac0aa01a7",
"6c82c613940e4d19bc7c7f72ea8ae2f6",
"f9e227e64f7d4a5abfd1e63fa449b41d",
"8598dfcc57fc4378bf4e6c29ac6ece25",
"a10c22ae98b8446289dfad20a0332c47",
"fbea43164ce3498cad565a95534fe770",
"87a73918c0154c18b6e7f3213418ca87",
"7482228547d44d02ba1a212d23ac620b",
"aa462e5caa6e4b2ebfda965180ff26a1",
"05ae6ee07e054df490692cfd338b8aa4",
"eea18326c15c453db11d83adbcda42f6",
"b5968328c2dd42a9941862a053808597",
"385929b4d4a5413e8bbe3d446997a34a",
"19716db98904411c8a906b111e37bda0",
"6d4b9c3ca98547b4aa070052454bc4c4",
"1c62e3407ce24ef287bccdd2ca7d94b0",
"38a3cc3a1e5c42eea9a6fd7743e1d916",
"0d02ae67ce4b439c9234227fc3cebae5",
"013cd883f7cf4e1c8610e5aa5ca52dbe",
"52a9f4e4d2084836913d738db7f26912",
"c6f7de90fd6848ac8dd6f910f8ff16c2",
"41b9fbecd1a14383b39368489d13d908",
"c5c18a1f60a842ac9bea877848266b07",
"214f59f40c474760bf7afada2974679c",
"01c2522f6ff2467180e92d241aac7605",
"91e297e4bbb34d2c99e87aa03767a7ef",
"499f79ff11d04e3d9c89dd311dbb3a18",
"7c20d1d8e8d340ae806e806d8cebbfa0",
"04154e9c25224695ac5739444a50c158",
"a969196910394454a757bdff842c9542",
"29bcc292a9564f398baa3f7cb4383da0",
"c671dccb95b348a9a31c97b6918c5166",
"1c428ea82d544f1da4573fc41d077d17",
"36d6e7e248f04fd393db82bd23850060",
"2e6b91ab64274d9e8f6bfeb19caa674a",
"d522dbab4c8c4dd1858aeb027869da8f",
"50ddd740217a4e99b039b3d45e5ee354"
]
},
"id": "p7ZeX6Gm6xp7",
"outputId": "3fdfd352-718f-43ae-ab27-4cdaba04ce26"
},
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "feccd063d54845389a4aeb794c13d0fe"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "e632b12bcf5a42ffba8de6e209e51802"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "d5abd2c22b29411fb1ce9c54a485f4c5"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "870d4d61eaeb4aab92f798fbba752253"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "6b64d5496f8149dc999c47e734dcece8"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "6c9c1669fa7f4a2e8f5a6aefcee2c112"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "9bf5bf95dd3146e98760efb4d6328895"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "a10c22ae98b8446289dfad20a0332c47"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "1c62e3407ce24ef287bccdd2ca7d94b0"
}
},
"metadata": {}
},
{
"output_type": "display_data",
"data": {
"text/plain": [
" 0%| | 0/3000 [00:00, ?it/s]"
],
"application/vnd.jupyter.widget-view+json": {
"version_major": 2,
"version_minor": 0,
"model_id": "499f79ff11d04e3d9c89dd311dbb3a18"
}
},
"metadata": {}
}
]
},
{
"cell_type": "code",
"source": [
"def find_best_seq_traj(results):\n",
" \"\"\"Find the best protein sequence in each evolutionary trajectory.\"\"\"\n",
" best_seq_traj = []\n",
" for traj, r in enumerate(results):\n",
" df = pd.DataFrame(r)\n",
" df['trajectory'] = traj\n",
" df_accept = df.query('accept == True')\n",
" best_seq_traj.append(df_accept.iloc[[np.argmax(df_accept['scores'].values)]])\n",
"\n",
" best_seq_traj = pd.concat(best_seq_traj)\n",
"\n",
" best_seq_traj['mutated_aa'] = best_seq_traj['sequences'].apply(mut_in_rep78)\n",
"\n",
" return best_seq_traj"
],
"metadata": {
"id": "ak7-cnf-8_i2"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "markdown",
"source": [
"Find best protein sequence in each evolutionary trajectory"
],
"metadata": {
"id": "feR8LA9BWnC9"
}
},
{
"cell_type": "code",
"source": [
"# find best protein sequence in each trajectory\n",
"traj_01_10 = find_best_seq_traj(results)\n",
"traj_01_10"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 363
},
"id": "12SOR07W9B_R",
"outputId": "027d0631-6b37-4dbb-b36a-136eaddbfc6d"
},
"execution_count": null,
"outputs": [
{
"output_type": "execute_result",
"data": {
"text/plain": [
" sequences scores accept \\\n",
"1657 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.369083 True \n",
"1864 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.213715 True \n",
"1187 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.245944 True \n",
"2652 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.501488 True \n",
"1186 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.465930 True \n",
"2362 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.822981 True \n",
"1320 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWILPPDSDMDLN... 1.562707 True \n",
"2681 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.264632 True \n",
"1477 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... 1.395800 True \n",
"1088 MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPLDSDMDLN... 1.755195 True \n",
"\n",
" trajectory mutated_aa \n",
"1657 0 [E75W, E226K, A496K] \n",
"1864 1 [L53T, L182F, I524H] \n",
"1187 2 [V196I, L276M, A513R] \n",
"2652 3 [V134T, G229K, V583G] \n",
"1186 4 [K487L, A522Y, Q621I] \n",
"2362 5 [N203Y, E510R, G539C] \n",
"1320 6 [E36I, M597D, V610N] \n",
"2681 7 [A56N, R494C, E551W] \n",
"1477 8 [D468H, D519W, T562P] \n",
"1088 9 [P39L, V215F, D519C] "
],
"text/html": [
"\n",
" \n",
"
\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" sequences | \n",
" scores | \n",
" accept | \n",
" trajectory | \n",
" mutated_aa | \n",
"
\n",
" \n",
" \n",
" \n",
" 1657 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.369083 | \n",
" True | \n",
" 0 | \n",
" [E75W, E226K, A496K] | \n",
"
\n",
" \n",
" 1864 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.213715 | \n",
" True | \n",
" 1 | \n",
" [L53T, L182F, I524H] | \n",
"
\n",
" \n",
" 1187 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.245944 | \n",
" True | \n",
" 2 | \n",
" [V196I, L276M, A513R] | \n",
"
\n",
" \n",
" 2652 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.501488 | \n",
" True | \n",
" 3 | \n",
" [V134T, G229K, V583G] | \n",
"
\n",
" \n",
" 1186 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.465930 | \n",
" True | \n",
" 4 | \n",
" [K487L, A522Y, Q621I] | \n",
"
\n",
" \n",
" 2362 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.822981 | \n",
" True | \n",
" 5 | \n",
" [N203Y, E510R, G539C] | \n",
"
\n",
" \n",
" 1320 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWILPPDSDMDLN... | \n",
" 1.562707 | \n",
" True | \n",
" 6 | \n",
" [E36I, M597D, V610N] | \n",
"
\n",
" \n",
" 2681 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.264632 | \n",
" True | \n",
" 7 | \n",
" [A56N, R494C, E551W] | \n",
"
\n",
" \n",
" 1477 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPPDSDMDLN... | \n",
" 1.395800 | \n",
" True | \n",
" 8 | \n",
" [D468H, D519W, T562P] | \n",
"
\n",
" \n",
" 1088 | \n",
" MPGFYEIVIKVPSDLDGHLPGISDSFVNWVAEKEWELPLDSDMDLN... | \n",
" 1.755195 | \n",
" True | \n",
" 9 | \n",
" [P39L, V215F, D519C] | \n",
"
\n",
" \n",
"
\n",
"
\n",
"
\n",
"
\n"
]
},
"metadata": {},
"execution_count": 18
}
]
},
{
"cell_type": "code",
"source": [
"traj_01_10.to_csv('data/traj_01_10.csv')"
],
"metadata": {
"id": "z_VXBWBUYdA2"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "code",
"source": [
"sampled_sequences_df.to_csv('data/sampled_sequences.csv')"
],
"metadata": {
"id": "RJwbamh3u334"
},
"execution_count": null,
"outputs": []
},
{
"cell_type": "markdown",
"source": [
"### Check MCMC trajectories for sampled seqs"
],
"metadata": {
"id": "DUN0gXPqDPrE"
}
},
{
"cell_type": "code",
"source": [
"traj1_scores = results[0]['scores']\n",
"# Plot fitness as a function of iterations for one evolutionary trajectory\n",
"plt.figure(figsize=(12, 6), dpi=800)\n",
"sns.lineplot(traj1_scores)\n",
"plt.xlabel('iterations');\n",
"plt.ylabel('fitness');\n",
"plt.title('Evolutionary trajectory 1');\n",
"plt.show()"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 682
},
"id": "Cp8YBR5GDKwp",
"outputId": "3440994e-086c-4ca1-d20f-6335bd9170c8"
},
"execution_count": null,
"outputs": [
{
"output_type": "display_data",
"data": {
"text/plain": [
"