"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"hgf_mcmc.plot_trajectories();"
]
},
{
"cell_type": "code",
"execution_count": 32,
"id": "c33e047c-3e65-40da-9392-0bf9124dfd80",
"metadata": {
"execution": {
"iopub.execute_input": "2025-05-11T20:53:38.919898Z",
"iopub.status.busy": "2025-05-11T20:53:38.919690Z",
"iopub.status.idle": "2025-05-11T20:53:38.927063Z",
"shell.execute_reply": "2025-05-11T20:53:38.926342Z"
}
},
"outputs": [
{
"data": {
"text/plain": [
"Array(203.0147, dtype=float32)"
]
},
"execution_count": 32,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"hgf_mcmc.surprise().sum()"
]
},
{
"cell_type": "markdown",
"id": "c5b16194-bacd-4f80-b20b-2ca9a23bbd2e",
"metadata": {},
"source": [
"# System configuration"
]
},
{
"cell_type": "code",
"execution_count": 33,
"id": "3874bd78-875a-45cd-85cf-da80e9c245bc",
"metadata": {
"execution": {
"iopub.execute_input": "2025-05-11T20:53:38.929101Z",
"iopub.status.busy": "2025-05-11T20:53:38.928910Z",
"iopub.status.idle": "2025-05-11T20:53:38.944805Z",
"shell.execute_reply": "2025-05-11T20:53:38.943965Z"
},
"scrolled": true
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Last updated: Sun May 11 2025\n",
"\n",
"Python implementation: CPython\n",
"Python version : 3.12.3\n",
"IPython version : 9.2.0\n",
"\n",
"pyhgf : 0.2.6\n",
"jax : 0.4.31\n",
"jaxlib: 0.4.31\n",
"\n",
"sys : 3.12.3 | packaged by conda-forge | (main, Apr 15 2024, 18:38:13) [GCC 12.3.0]\n",
"matplotlib: 3.10.1\n",
"arviz : 0.21.0\n",
"jax : 0.4.31\n",
"pyhgf : 0.2.6\n",
"IPython : 9.2.0\n",
"pymc : 5.22.0\n",
"\n",
"Watermark: 2.5.0\n",
"\n"
]
}
],
"source": [
"%load_ext watermark\n",
"%watermark -n -u -v -iv -w -p pyhgf,jax,jaxlib"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "e40c4155",
"metadata": {
"editable": true,
"slideshow": {
"slide_type": ""
}
},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.12.3"
},
"widgets": {
"application/vnd.jupyter.widget-state+json": {
"state": {
"2cd84e050ed14bf98ed948cd3d51e204": {
"model_module": "@jupyter-widgets/output",
"model_module_version": "1.0.0",
"model_name": "OutputModel",
"state": {
"_dom_classes": [],
"_model_module": "@jupyter-widgets/output",
"_model_module_version": "1.0.0",
"_model_name": "OutputModel",
"_view_count": null,
"_view_module": "@jupyter-widgets/output",
"_view_module_version": "1.0.0",
"_view_name": "OutputView",
"layout": "IPY_MODEL_90a31d94cd0d47af871aaf63fb9160c6",
"msg_id": "",
"outputs": [
{
"data": {
"text/html": " \n Progress Draws Divergences Step size Grad evals Sampling Speed Elapsed Remaining \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n ━━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 0.96 1 879.23 draws/s 0:00:02 0:00:00 \n ━━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 1.58 3 444.43 draws/s 0:00:04 0:00:00 \n \n
\n",
"text/plain": " \n \u001b[1m \u001b[0m\u001b[1mProgress \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mDraws\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mDivergences\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mStep size\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mGrad evals\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mSampling Speed\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mElapsed\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mRemaining\u001b[0m\u001b[1m \u001b[0m \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 0.96 1 879.23 draws/s 0:00:02 0:00:00 \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 1.58 3 444.43 draws/s 0:00:04 0:00:00 \n \n"
},
"metadata": {},
"output_type": "display_data"
}
],
"tabbable": null,
"tooltip": null
}
},
"68ff6a1285294cb9ae9702092a656c32": {
"model_module": "@jupyter-widgets/base",
"model_module_version": "2.0.0",
"model_name": "LayoutModel",
"state": {
"_model_module": "@jupyter-widgets/base",
"_model_module_version": "2.0.0",
"_model_name": "LayoutModel",
"_view_count": null,
"_view_module": "@jupyter-widgets/base",
"_view_module_version": "2.0.0",
"_view_name": "LayoutView",
"align_content": null,
"align_items": null,
"align_self": null,
"border_bottom": null,
"border_left": null,
"border_right": null,
"border_top": 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,
"padding": null,
"right": null,
"top": null,
"visibility": null,
"width": null
}
},
"90a31d94cd0d47af871aaf63fb9160c6": {
"model_module": "@jupyter-widgets/base",
"model_module_version": "2.0.0",
"model_name": "LayoutModel",
"state": {
"_model_module": "@jupyter-widgets/base",
"_model_module_version": "2.0.0",
"_model_name": "LayoutModel",
"_view_count": null,
"_view_module": "@jupyter-widgets/base",
"_view_module_version": "2.0.0",
"_view_name": "LayoutView",
"align_content": null,
"align_items": null,
"align_self": null,
"border_bottom": null,
"border_left": null,
"border_right": null,
"border_top": 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,
"padding": null,
"right": null,
"top": null,
"visibility": null,
"width": null
}
},
"fdc13b6721b347b5be0aa2f41511a6bb": {
"model_module": "@jupyter-widgets/output",
"model_module_version": "1.0.0",
"model_name": "OutputModel",
"state": {
"_dom_classes": [],
"_model_module": "@jupyter-widgets/output",
"_model_module_version": "1.0.0",
"_model_name": "OutputModel",
"_view_count": null,
"_view_module": "@jupyter-widgets/output",
"_view_module_version": "1.0.0",
"_view_name": "OutputView",
"layout": "IPY_MODEL_68ff6a1285294cb9ae9702092a656c32",
"msg_id": "",
"outputs": [
{
"data": {
"text/html": " \n Progress Draws Divergences Step size Grad evals Sampling Speed Elapsed Remaining \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n ━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 0.85 1 1441.29 draws/s 0:00:01 0:00:00 \n ━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 0.92 3 749.01 draws/s 0:00:02 0:00:00 \n \n
\n",
"text/plain": " \n \u001b[1m \u001b[0m\u001b[1mProgress \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mDraws\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mDivergences\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mStep size\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mGrad evals\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mSampling Speed \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mElapsed\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mRemaining\u001b[0m\u001b[1m \u001b[0m \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 0.85 1 1441.29 draws/s 0:00:01 0:00:00 \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 0.92 3 749.01 draws/s 0:00:02 0:00:00 \n \n"
},
"metadata": {},
"output_type": "display_data"
}
],
"tabbable": null,
"tooltip": null
}
}
},
"version_major": 2,
"version_minor": 0
}
}
},
"nbformat": 4,
"nbformat_minor": 5
}