Deploying to gh-pages from @ 3218489 🚀

ComputationalPsychiatry · Dec 12, 2023 · 2ba4171 · 2ba4171
1 parent f5f892a
commit 2ba4171
Show file tree

Hide file tree

Showing 78 changed files with 223 additions and 218 deletions.
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/.doctrees/notebooks/0-Creating_networks.doctree b/.doctrees/notebooks/0-Creating_networks.doctree
diff --git a/.doctrees/notebooks/1.1-Binary_HGF.doctree b/.doctrees/notebooks/1.1-Binary_HGF.doctree
diff --git a/.doctrees/notebooks/1.2-Categorical_HGF.doctree b/.doctrees/notebooks/1.2-Categorical_HGF.doctree
diff --git a/.doctrees/notebooks/1.3-Continuous_HGF.doctree b/.doctrees/notebooks/1.3-Continuous_HGF.doctree
diff --git a/.doctrees/notebooks/2-Using_custom_response_functions.doctree b/.doctrees/notebooks/2-Using_custom_response_functions.doctree
diff --git a/.doctrees/notebooks/3-Multilevel_HGF.doctree b/.doctrees/notebooks/3-Multilevel_HGF.doctree
diff --git a/.doctrees/notebooks/4-Parameter_recovery.doctree b/.doctrees/notebooks/4-Parameter_recovery.doctree
diff --git a/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree b/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree
diff --git a/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree b/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree
diff --git a/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree b/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree
diff --git a/.doctrees/notebooks/Exercise_1_Using_the_HGF.doctree b/.doctrees/notebooks/Exercise_1_Using_the_HGF.doctree
diff --git a/_images/02c634157ece2f625f880471d44c53dd17c1db356c8b872123fec986d4d3176f.png b/_images/02c634157ece2f625f880471d44c53dd17c1db356c8b872123fec986d4d3176f.png
diff --git a/_images/0603a3b723cc7da25e39085b0e1d05d363dd21080afe169d05adc1128b7a6db4.png b/_images/0603a3b723cc7da25e39085b0e1d05d363dd21080afe169d05adc1128b7a6db4.png
diff --git a/...69ac0c4a8b7994776641b7ca80207aba0e073.svg → ...d7adb40e86a06d3c587c73103c1d2a5698bb6.svg b/...69ac0c4a8b7994776641b7ca80207aba0e073.svg → ...d7adb40e86a06d3c587c73103c1d2a5698bb6.svg
diff --git a/_images/1547fc8985144ee3fdea0b5ad284530b0e4e776ff3d09cd70cb738f973afe4eb.png b/_images/1547fc8985144ee3fdea0b5ad284530b0e4e776ff3d09cd70cb738f973afe4eb.png
diff --git a/_images/1591c0e0518f800ef8329844de19a26d494daf518fc0c2fea6da151a6731620a.png b/_images/1591c0e0518f800ef8329844de19a26d494daf518fc0c2fea6da151a6731620a.png
diff --git a/_images/1a832b6308977fbde18f5e43acf5911640a98c1a096f5e2f47dcd6a0896bf03f.png b/_images/1a832b6308977fbde18f5e43acf5911640a98c1a096f5e2f47dcd6a0896bf03f.png
diff --git a/_images/1e8bed49c7e2152e429978f89facf943ba05b41f5a6ad77d332e0a1cd85a65a1.png b/_images/1e8bed49c7e2152e429978f89facf943ba05b41f5a6ad77d332e0a1cd85a65a1.png
diff --git a/_images/1f0eb6eaeccd0b92a3c7deb4fb700d2fe8df85e83eeeccd301a0c3ba8628da93.png b/_images/1f0eb6eaeccd0b92a3c7deb4fb700d2fe8df85e83eeeccd301a0c3ba8628da93.png
diff --git a/_images/203680778166728b3f6e1753de7dc9e3244f813fcb8276bb07943bca5691d166.png b/_images/203680778166728b3f6e1753de7dc9e3244f813fcb8276bb07943bca5691d166.png
diff --git a/_images/2abb9d6acaec03ffbfddf693173bd54401a7adfd59d58ad2e05799e71566709e.png b/_images/2abb9d6acaec03ffbfddf693173bd54401a7adfd59d58ad2e05799e71566709e.png
diff --git a/...b557373c04de09f42d4af609a23e9e9c994de.svg → ...b9a65da5ccc9376913b3f70b8f1adfa705f8d.svg b/...b557373c04de09f42d4af609a23e9e9c994de.svg → ...b9a65da5ccc9376913b3f70b8f1adfa705f8d.svg
diff --git a/_images/31aef77256c20833dc830b8b2d4ad79279e6b6c1e4bcacebb47af98654a3ec6b.png b/_images/31aef77256c20833dc830b8b2d4ad79279e6b6c1e4bcacebb47af98654a3ec6b.png
diff --git a/_images/3802c9fcb157508be31068c3c491008339ef7239229926efcea583875ca5374f.png b/_images/3802c9fcb157508be31068c3c491008339ef7239229926efcea583875ca5374f.png
diff --git a/_images/3f6a8fabe0ba460a15cf8b7d24f2e662e48f90b70c9bc69c66be39d839ab89ac.png b/_images/3f6a8fabe0ba460a15cf8b7d24f2e662e48f90b70c9bc69c66be39d839ab89ac.png
diff --git a/_images/42b0814008f7e4f99d7c811575d5fac9c8f0d62900e6b23055edb55b08159686.png b/_images/42b0814008f7e4f99d7c811575d5fac9c8f0d62900e6b23055edb55b08159686.png
diff --git a/_images/46f9c3d4129eac43d7ff26e3f44c33997d030a5130fa95a41a0d6619b7093624.png b/_images/46f9c3d4129eac43d7ff26e3f44c33997d030a5130fa95a41a0d6619b7093624.png
diff --git a/_images/4801015994125f66960e84b25f07bb74bbef11dc7c4b160be02cf420a734c2a5.png b/_images/4801015994125f66960e84b25f07bb74bbef11dc7c4b160be02cf420a734c2a5.png
diff --git a/...4cbf6bfb92e6873d38804d27795de7b6a7546.svg → ...9fb24140bb3752126295bc90045190c4a8cac.svg b/...4cbf6bfb92e6873d38804d27795de7b6a7546.svg → ...9fb24140bb3752126295bc90045190c4a8cac.svg
diff --git a/_images/52f6ef4693c3a32f4d565295142f6733333ec8c0b115a2539adbf9f6b6051128.png b/_images/52f6ef4693c3a32f4d565295142f6733333ec8c0b115a2539adbf9f6b6051128.png
diff --git a/_images/54d605adba816c0c445027d28e28d7268a03d5c66d9110d6e1934c9def3ca8f4.png b/_images/54d605adba816c0c445027d28e28d7268a03d5c66d9110d6e1934c9def3ca8f4.png
diff --git a/_images/56e2554f73cd3e3bd91ba99ceb96b90d0b3b096011faccc04525b6b73cc94e1d.png b/_images/56e2554f73cd3e3bd91ba99ceb96b90d0b3b096011faccc04525b6b73cc94e1d.png
diff --git a/_images/5855cdd14837c67afb8ca28fa689a016c044589c1a7aae3a5ef3e2623ba2492d.png b/_images/5855cdd14837c67afb8ca28fa689a016c044589c1a7aae3a5ef3e2623ba2492d.png
diff --git a/_images/61e8a21afd0fcdbb1467fc6e86fcff86fc41e2b1d766279716c7236fbc23d0ec.png b/_images/61e8a21afd0fcdbb1467fc6e86fcff86fc41e2b1d766279716c7236fbc23d0ec.png
diff --git a/_images/62561bc0ac9d19b3344fa9b97293538f9820c7add17bbb4313fa5132d821b44f.png b/_images/62561bc0ac9d19b3344fa9b97293538f9820c7add17bbb4313fa5132d821b44f.png
diff --git a/_images/67b7dfc3bfb0eaaafdd5b1b82cc1e07b2376db74f50df660698f31bc0ae0c68b.png b/_images/67b7dfc3bfb0eaaafdd5b1b82cc1e07b2376db74f50df660698f31bc0ae0c68b.png
diff --git a/_images/7247c09f2e536514a9f4cca8e408d45cadbdb7a5550538ee6d94c624b6ac576f.png b/_images/7247c09f2e536514a9f4cca8e408d45cadbdb7a5550538ee6d94c624b6ac576f.png
diff --git a/_images/736602f2fe8922750c73be3bf6b2eb9bed942ba8f8d4af7ab181c4d47d8bd905.png b/_images/736602f2fe8922750c73be3bf6b2eb9bed942ba8f8d4af7ab181c4d47d8bd905.png
diff --git a/_images/749af733098704bc945fb1d97f3c5f8a42dfe280b04272e947d66d1afcb6e420.png b/_images/749af733098704bc945fb1d97f3c5f8a42dfe280b04272e947d66d1afcb6e420.png
diff --git a/_images/7be2d0afe59ca2c998555a253a597cba3132e47c01feeee87bb4a7dac66cd363.png b/_images/7be2d0afe59ca2c998555a253a597cba3132e47c01feeee87bb4a7dac66cd363.png
diff --git a/_images/87a001471503dc1d20c941a910362b3d5d9fbf43aaa339a59831214087c476be.png b/_images/87a001471503dc1d20c941a910362b3d5d9fbf43aaa339a59831214087c476be.png
diff --git a/_images/9892bd75ab3b858d1a9d1a83e4af772fa3a99e20a4e589d845720a0b40166c21.png b/_images/9892bd75ab3b858d1a9d1a83e4af772fa3a99e20a4e589d845720a0b40166c21.png
diff --git a/_images/addf2d33d91d92db11cc7d4de776ca962804fb42b09820afed606eb0efc9e6df.png b/_images/addf2d33d91d92db11cc7d4de776ca962804fb42b09820afed606eb0efc9e6df.png
diff --git a/_images/b1706272d5b5205ea3d7f679f3194a98715502327bed0cd7c86375265a0592ad.png b/_images/b1706272d5b5205ea3d7f679f3194a98715502327bed0cd7c86375265a0592ad.png
diff --git a/_images/b756552bdb30d0ca3e4da077c5ab6de7279332923ea9eb003baf06cde020dfc1.png b/_images/b756552bdb30d0ca3e4da077c5ab6de7279332923ea9eb003baf06cde020dfc1.png
diff --git a/_images/bb1ee501e9d2ba34e10b0640298db1a12c3ca8a7acef64f6e784a9bfe1cbfa07.png b/_images/bb1ee501e9d2ba34e10b0640298db1a12c3ca8a7acef64f6e784a9bfe1cbfa07.png
diff --git a/_images/bb1f410acecc59340d65b92f58131495b023ae3baef4692b5f738979d4b3ca3c.png b/_images/bb1f410acecc59340d65b92f58131495b023ae3baef4692b5f738979d4b3ca3c.png
diff --git a/_images/c0cc4e564b4fb757646628dd70c3f8873f5e8fff3281d6896747254b26bebf37.png b/_images/c0cc4e564b4fb757646628dd70c3f8873f5e8fff3281d6896747254b26bebf37.png
diff --git a/_images/c510c0ad1aa87da765fcd6507d202a68688c514f7dfacce929401bb373586b62.png b/_images/c510c0ad1aa87da765fcd6507d202a68688c514f7dfacce929401bb373586b62.png
diff --git a/_images/d004fa4e7cf4cab40f2a8943f7f0cefac5d68f8b50d0b898e758c89a81520368.png b/_images/d004fa4e7cf4cab40f2a8943f7f0cefac5d68f8b50d0b898e758c89a81520368.png
diff --git a/_images/d0f3f935f9ced429f11a94dbeabafde159c6974978248530871bea9b187e9e59.png b/_images/d0f3f935f9ced429f11a94dbeabafde159c6974978248530871bea9b187e9e59.png
diff --git a/_images/d17b99ca5f6c11895764d340b26aff9dd94e60da26bef32370033b59bea9b29a.png b/_images/d17b99ca5f6c11895764d340b26aff9dd94e60da26bef32370033b59bea9b29a.png
diff --git a/_images/d938a6b1bdbfd7937a73248f3b082e7c49497fb167f3d921c4c4563624201262.png b/_images/d938a6b1bdbfd7937a73248f3b082e7c49497fb167f3d921c4c4563624201262.png
diff --git a/_images/d9bf3e17e44459fca1af93f1b13c5b488f693f5ebff2d22c004c5fac35073132.png b/_images/d9bf3e17e44459fca1af93f1b13c5b488f693f5ebff2d22c004c5fac35073132.png
diff --git a/_images/dcb2781140f95bee68b01402d83dfeabc3645ec70e99c8a6ca24d80a328bd735.png b/_images/dcb2781140f95bee68b01402d83dfeabc3645ec70e99c8a6ca24d80a328bd735.png
diff --git a/_images/e0f67eded59e1e26f8f6d1258b13925b350e12a62e52c7b04ac6b0e035591928.png b/_images/e0f67eded59e1e26f8f6d1258b13925b350e12a62e52c7b04ac6b0e035591928.png
diff --git a/_images/e1213e5b662596f66cec25cd5c6b2ef252dc9c3005320470408c5028bb353e12.png b/_images/e1213e5b662596f66cec25cd5c6b2ef252dc9c3005320470408c5028bb353e12.png
diff --git a/...1304a4e4b0eb711fc8054497123f2e226d095.svg → ...90c832e9b204e89fed0d0688e3e2fdb1677dd.svg b/...1304a4e4b0eb711fc8054497123f2e226d095.svg → ...90c832e9b204e89fed0d0688e3e2fdb1677dd.svg
diff --git a/_images/e8f9a65605fc996c7427f152c6460019613460a502b7fb88db6bfa9089b5270a.png b/_images/e8f9a65605fc996c7427f152c6460019613460a502b7fb88db6bfa9089b5270a.png
diff --git a/_images/eb3496bc7a19bc2a60a0b55a880773cf8853b63c443b6d194073c98acf2f7a19.png b/_images/eb3496bc7a19bc2a60a0b55a880773cf8853b63c443b6d194073c98acf2f7a19.png
diff --git a/_images/edc1fe9410d0ab03622b76867d6ba242a6acd4fe797eb923bac4665ced396404.png b/_images/edc1fe9410d0ab03622b76867d6ba242a6acd4fe797eb923bac4665ced396404.png
diff --git a/_images/fa1861554af0cba6bc04193260708e75e5a359e2c56f55a283cce2192f7c9967.png b/_images/fa1861554af0cba6bc04193260708e75e5a359e2c56f55a283cce2192f7c9967.png
diff --git a/_images/fae290094c80891ce2bf427b85d679fe490cc607ee7d72c47196fb029166e00d.png b/_images/fae290094c80891ce2bf427b85d679fe490cc607ee7d72c47196fb029166e00d.png
diff --git a/_images/fcc702402a5ad873176d010c7ec444a7e81163d0c4aaa191fca73794dcdaf5c0.png b/_images/fcc702402a5ad873176d010c7ec444a7e81163d0c4aaa191fca73794dcdaf5c0.png
diff --git a/_images/fe961935a8860eeeefb37c92eb0e373d7bef36c7f7197d8e7887a1927a45530b.png b/_images/fe961935a8860eeeefb37c92eb0e373d7bef36c7f7197d8e7887a1927a45530b.png
diff --git a/notebooks/0-Creating_networks.html b/notebooks/0-Creating_networks.html
@@ -537,7 +537,7 @@ <h3>Visualizing probabilistic networks<a class="headerlink" href="#visualizing-p
 </div>
 <div class="cell_output docutils container">
 <div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
-I0000 00:00:1702371054.466878    2381 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
+I0000 00:00:1702372054.680589    2436 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
 No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
 </pre></div>
 </div>
@@ -707,7 +707,7 @@ <h5>Continuous value coupling<a class="headerlink" href="#continuous-value-coupl
 <div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Adding 1000 new observations.
 </pre></div>
 </div>
-<img alt="../_images/fae290094c80891ce2bf427b85d679fe490cc607ee7d72c47196fb029166e00d.png" src="../_images/fae290094c80891ce2bf427b85d679fe490cc607ee7d72c47196fb029166e00d.png" />
+<img alt="../_images/3802c9fcb157508be31068c3c491008339ef7239229926efcea583875ca5374f.png" src="../_images/3802c9fcb157508be31068c3c491008339ef7239229926efcea583875ca5374f.png" />
 </div>
 </div>
 <div class="admonition note">
@@ -867,7 +867,7 @@ <h5>Coupling with binary nodes<a class="headerlink" href="#coupling-with-binary-
 <div class="output stream highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Adding 1000 new observations.
 </pre></div>
 </div>
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;pyhgf.model.HGF at 0x7ff8186a5390&gt;
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>&lt;pyhgf.model.HGF at 0x7f3b88d81ab0&gt;
 </pre></div>
 </div>
 </div>
@@ -994,10 +994,10 @@ <h1>System configuration<a class="headerlink" href="#system-configuration" title
 jax   : 0.4.16
 jaxlib: 0.4.16
 
-seaborn   : 0.13.0
 numpy     : 1.22.0
-sys       : 3.10.13 (main, Aug 28 2023, 08:28:42) [GCC 11.4.0]
 matplotlib: 3.8.2
+sys       : 3.10.13 (main, Aug 28 2023, 08:28:42) [GCC 11.4.0]
+seaborn   : 0.13.0
 
 Watermark: 2.4.3
 </pre></div>

diff --git a/notebooks/1.1-Binary_HGF.html b/notebooks/1.1-Binary_HGF.html
@@ -539,7 +539,7 @@ <h4>Create the model<a class="headerlink" href="#create-the-model" title="Link t
 </div>
 <div class="cell_output docutils container">
 <div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
-I0000 00:00:1702371077.119374    2600 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
+I0000 00:00:1702372077.922783    2667 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
 No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
 </pre></div>
 </div>
@@ -816,7 +816,7 @@ <h4>Sampling<a class="headerlink" href="#sampling" title="Link to this heading">
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/7be2d0afe59ca2c998555a253a597cba3132e47c01feeee87bb4a7dac66cd363.png" src="../_images/7be2d0afe59ca2c998555a253a597cba3132e47c01feeee87bb4a7dac66cd363.png" />
+<img alt="../_images/749af733098704bc945fb1d97f3c5f8a42dfe280b04272e947d66d1afcb6e420.png" src="../_images/749af733098704bc945fb1d97f3c5f8a42dfe280b04272e947d66d1afcb6e420.png" />
 </div>
 </div>
 </section>
@@ -865,7 +865,7 @@ <h3>Using the learned parameters<a class="headerlink" href="#using-the-learned-p
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/bb1ee501e9d2ba34e10b0640298db1a12c3ca8a7acef64f6e784a9bfe1cbfa07.png" src="../_images/bb1ee501e9d2ba34e10b0640298db1a12c3ca8a7acef64f6e784a9bfe1cbfa07.png" />
+<img alt="../_images/56e2554f73cd3e3bd91ba99ceb96b90d0b3b096011faccc04525b6b73cc94e1d.png" src="../_images/56e2554f73cd3e3bd91ba99ceb96b90d0b3b096011faccc04525b6b73cc94e1d.png" />
 </div>
 </div>
 <div class="cell docutils container">
@@ -875,7 +875,7 @@ <h3>Using the learned parameters<a class="headerlink" href="#using-the-learned-p
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Array(202.53082, dtype=float32)
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Array(202.5297, dtype=float32)
 </pre></div>
 </div>
 </div>
@@ -927,7 +927,7 @@ <h4>Visualizing the model<a class="headerlink" href="#id8" title="Link to this h
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/69539e2739458d7218e514133841304a4e4b0eb711fc8054497123f2e226d095.svg" src="../_images/69539e2739458d7218e514133841304a4e4b0eb711fc8054497123f2e226d095.svg" /></div>
+<img alt="../_images/e3093df40c606a365fce51328bb90c832e9b204e89fed0d0688e3e2fdb1677dd.svg" src="../_images/e3093df40c606a365fce51328bb90c832e9b204e89fed0d0688e3e2fdb1677dd.svg" /></div>
 </div>
 </section>
 <section id="id9">
@@ -971,9 +971,9 @@ <h4>Sampling<a class="headerlink" href="#id9" title="Link to this heading">#</a>
 </div><div class="output text_html">
     <div>
       <progress value='4000' class='' max='4000' style='width:300px; height:20px; vertical-align: middle;'></progress>
-      100.00% [4000/4000 00:08&lt;00:00 Sampling 2 chains, 1 divergences]
+      100.00% [4000/4000 00:08&lt;00:00 Sampling 2 chains, 0 divergences]
     </div>
-    </div><div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 8 seconds.
+    </div><div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 9 seconds.
 </pre></div>
 </div>
 <div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -989,7 +989,7 @@ <h4>Sampling<a class="headerlink" href="#id9" title="Link to this heading">#</a>
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/7247c09f2e536514a9f4cca8e408d45cadbdb7a5550538ee6d94c624b6ac576f.png" src="../_images/7247c09f2e536514a9f4cca8e408d45cadbdb7a5550538ee6d94c624b6ac576f.png" />
+<img alt="../_images/203680778166728b3f6e1753de7dc9e3244f813fcb8276bb07943bca5691d166.png" src="../_images/203680778166728b3f6e1753de7dc9e3244f813fcb8276bb07943bca5691d166.png" />
 </div>
 </div>
 </section>
@@ -1039,7 +1039,7 @@ <h3>Using the learned parameters<a class="headerlink" href="#id10" title="Link t
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/addf2d33d91d92db11cc7d4de776ca962804fb42b09820afed606eb0efc9e6df.png" src="../_images/addf2d33d91d92db11cc7d4de776ca962804fb42b09820afed606eb0efc9e6df.png" />
+<img alt="../_images/1e8bed49c7e2152e429978f89facf943ba05b41f5a6ad77d332e0a1cd85a65a1.png" src="../_images/1e8bed49c7e2152e429978f89facf943ba05b41f5a6ad77d332e0a1cd85a65a1.png" />
 </div>
 </div>
 <div class="cell docutils container">
@@ -1049,7 +1049,7 @@ <h3>Using the learned parameters<a class="headerlink" href="#id10" title="Link t
 </div>
 </div>
 <div class="cell_output docutils container">
-<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Array(203.0094, dtype=float32)
+<div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>Array(203.0253, dtype=float32)
 </pre></div>
 </div>
 </div>
@@ -1077,11 +1077,11 @@ <h1>System configuration<a class="headerlink" href="#system-configuration" title
 jax   : 0.4.16
 jaxlib: 0.4.16
 
-arviz     : 0.16.1
-matplotlib: 3.8.2
-pymc      : 5.10.1
 sys       : 3.10.13 (main, Aug 28 2023, 08:28:42) [GCC 11.4.0]
+pymc      : 5.10.1
 jax       : 0.4.16
+matplotlib: 3.8.2
+arviz     : 0.16.1
 seaborn   : 0.13.0
 
 Watermark: 2.4.3

diff --git a/notebooks/1.2-Categorical_HGF.html b/notebooks/1.2-Categorical_HGF.html
@@ -534,7 +534,7 @@ <h2>Simulating a dataset<a class="headerlink" href="#simulating-a-dataset" title
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/5855cdd14837c67afb8ca28fa689a016c044589c1a7aae3a5ef3e2623ba2492d.png" src="../_images/5855cdd14837c67afb8ca28fa689a016c044589c1a7aae3a5ef3e2623ba2492d.png" />
+<img alt="../_images/1547fc8985144ee3fdea0b5ad284530b0e4e776ff3d09cd70cb738f973afe4eb.png" src="../_images/1547fc8985144ee3fdea0b5ad284530b0e4e776ff3d09cd70cb738f973afe4eb.png" />
 </div>
 </div>
 <div class="admonition note">
@@ -575,7 +575,7 @@ <h3>Creating the probabilistic network<a class="headerlink" href="#creating-the-
 </div>
 <div class="cell_output docutils container">
 <div class="output stderr highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
-I0000 00:00:1702371126.647286    2989 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
+I0000 00:00:1702372129.817608    3049 tfrt_cpu_pjrt_client.cc:349] TfrtCpuClient created.
 No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
 </pre></div>
 </div>
@@ -658,7 +658,7 @@ <h3>Fitting the model forwards<a class="headerlink" href="#fitting-the-model-for
 </div>
 </div>
 <div class="cell_output docutils container">
-<img alt="../_images/9892bd75ab3b858d1a9d1a83e4af772fa3a99e20a4e589d845720a0b40166c21.png" src="../_images/9892bd75ab3b858d1a9d1a83e4af772fa3a99e20a4e589d845720a0b40166c21.png" />
+<img alt="../_images/e0f67eded59e1e26f8f6d1258b13925b350e12a62e52c7b04ac6b0e035591928.png" src="../_images/e0f67eded59e1e26f8f6d1258b13925b350e12a62e52c7b04ac6b0e035591928.png" />
 </div>
 </div>
 <div class="cell docutils container">
@@ -672,15 +672,15 @@ <h3>Fitting the model forwards<a class="headerlink" href="#fitting-the-model-for
         1.       ],
        [2.0445216, 2.0445216, 2.0445216, ..., 2.5556629, 2.0445216,
         2.0445216],
-       [2.1143956, 2.1143956, 2.1143956, ..., 2.7494168, 2.1143956,
+       [2.1143956, 2.1143956, 2.1143956, ..., 2.222889 , 2.6317995,
         2.1143956],
        ...,
-       [4.411863 , 6.6914744, 3.7166734, ..., 2.4109159, 1.7010833,
-        1.5658604],
-       [4.37676  , 7.2455597, 3.6916356, ..., 2.4218128, 1.6985779,
-        1.5642372],
-       [4.3405523, 7.602443 , 3.6663218, ..., 2.431178 , 1.6961371,
-        1.5626398]], dtype=float32)
+       [5.485437 , 6.3858624, 4.574685 , ..., 2.1575747, 2.3001714,
+        1.4340663],
+       [5.080553 , 6.2617826, 5.09962  , ..., 2.1646917, 2.2970295,
+        1.4331563],
+       [5.0526347, 6.803068 , 4.649716 , ..., 2.1709945, 2.2936542,
+        1.4323133]], dtype=float32)
 </pre></div>
 </div>
 </div>
@@ -836,14 +836,14 @@ <h1>System configuration<a class="headerlink" href="#system-configuration" title
 jax   : 0.4.16
 jaxlib: 0.4.16
 
-pytensor  : 2.18.3
-jax       : 0.4.16
+arviz     : 0.16.1
+seaborn   : 0.13.0
 matplotlib: 3.8.2
 pymc      : 5.10.1
-sys       : 3.10.13 (main, Aug 28 2023, 08:28:42) [GCC 11.4.0]
-arviz     : 0.16.1
 numpy     : 1.22.0
-seaborn   : 0.13.0
+sys       : 3.10.13 (main, Aug 28 2023, 08:28:42) [GCC 11.4.0]
+jax       : 0.4.16
+pytensor  : 2.18.3
 
 Watermark: 2.4.3
 </pre></div>