"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"hgf_mcmc.plot_trajectories();"
]
},
{
"cell_type": "code",
"execution_count": 32,
"id": "c33e047c-3e65-40da-9392-0bf9124dfd80",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"Array(203.01764, 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": {
"scrolled": true
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Last updated: Sat Aug 23 2025\n",
"\n",
"Python implementation: CPython\n",
"Python version : 3.12.10\n",
"IPython version : 9.4.0\n",
"\n",
"pyhgf : 0.2.7\n",
"jax : 0.7.0\n",
"jaxlib: 0.7.0\n",
"\n",
"arviz : 0.21.0\n",
"pyhgf : 0.2.7\n",
"IPython : 9.4.0\n",
"matplotlib: 3.10.1\n",
"sys : 3.12.10 (main, Apr 9 2025, 04:03:51) [Clang 20.1.0 ]\n",
"pymc : 5.25.1\n",
"jax : 0.7.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.10"
},
"widgets": {
"application/vnd.jupyter.widget-state+json": {
"state": {
"1e733bc982fb4ce68df29c032d65a5b9": {
"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
}
},
"6c4a097fe4f34e02890710bff8d98d16": {
"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
}
},
"6f0d0204a3fa4232b4917065908671ad": {
"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_6c4a097fe4f34e02890710bff8d98d16",
"msg_id": "",
"outputs": [
{
"data": {
"text/html": " \n Progress Draws Divergences Step size Grad evals Sampling Speed Elapsed Remaining \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n ━━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 1.10 1 885.94 draws/s 0:00:02 0:00:00 \n ━━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 1.00 3 440.27 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 1.10 1 885.94 draws/s 0:00:02 0:00:00 \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 1.00 3 440.27 draws/s 0:00:04 0:00:00 \n \n"
},
"metadata": {},
"output_type": "display_data"
}
],
"tabbable": null,
"tooltip": null
}
},
"85408e39501f414c858f0ccc1ce4de5e": {
"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_1e733bc982fb4ce68df29c032d65a5b9",
"msg_id": "",
"outputs": [
{
"data": {
"text/html": " \n Progress Draws Divergences Step size Grad evals Sampling Speed Elapsed Remaining \n ───────────────────────────────────────────────────────────────────────────────────────────────────────────────── \n ━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 1.20 3 1221.82 draws/s 0:00:01 0:00:00 \n ━━━━━━━━━━━━━━━━━━━━━━━━ 2000 0 1.01 3 651.85 draws/s 0:00:03 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 1.20 3 1221.82 draws/s 0:00:01 0:00:00 \n \u001b[38;2;31;119;180m━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m 2000 0 1.01 3 651.85 draws/s 0:00:03 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
}