diff --git a/dev/.doctrees/environment.pickle b/dev/.doctrees/environment.pickle
index 37d2e9d21..9e509a56b 100644
Binary files a/dev/.doctrees/environment.pickle and b/dev/.doctrees/environment.pickle differ
diff --git a/dev/.doctrees/index.doctree b/dev/.doctrees/index.doctree
index c70da8ef0..f77422f10 100644
Binary files a/dev/.doctrees/index.doctree and b/dev/.doctrees/index.doctree differ
diff --git a/dev/.doctrees/notebooks/0.1-Theory.doctree b/dev/.doctrees/notebooks/0.1-Theory.doctree
index 7a62e66c9..8e01ba5d0 100644
Binary files a/dev/.doctrees/notebooks/0.1-Theory.doctree and b/dev/.doctrees/notebooks/0.1-Theory.doctree differ
diff --git a/dev/.doctrees/notebooks/0.2-Creating_networks.doctree b/dev/.doctrees/notebooks/0.2-Creating_networks.doctree
index 04f08807a..d08b375c9 100644
Binary files a/dev/.doctrees/notebooks/0.2-Creating_networks.doctree and b/dev/.doctrees/notebooks/0.2-Creating_networks.doctree differ
diff --git a/dev/.doctrees/notebooks/0.3-Generalised_filtering.doctree b/dev/.doctrees/notebooks/0.3-Generalised_filtering.doctree
index d32b804e7..3677e50f4 100644
Binary files a/dev/.doctrees/notebooks/0.3-Generalised_filtering.doctree and b/dev/.doctrees/notebooks/0.3-Generalised_filtering.doctree differ
diff --git a/dev/.doctrees/notebooks/1.1-Binary_HGF.doctree b/dev/.doctrees/notebooks/1.1-Binary_HGF.doctree
index 95f6cf050..a0de338ca 100644
Binary files a/dev/.doctrees/notebooks/1.1-Binary_HGF.doctree and b/dev/.doctrees/notebooks/1.1-Binary_HGF.doctree differ
diff --git a/dev/.doctrees/notebooks/1.2-Categorical_HGF.doctree b/dev/.doctrees/notebooks/1.2-Categorical_HGF.doctree
index de9c0273c..c2855a89f 100644
Binary files a/dev/.doctrees/notebooks/1.2-Categorical_HGF.doctree and b/dev/.doctrees/notebooks/1.2-Categorical_HGF.doctree differ
diff --git a/dev/.doctrees/notebooks/1.3-Continuous_HGF.doctree b/dev/.doctrees/notebooks/1.3-Continuous_HGF.doctree
index 941870450..068c0eef6 100644
Binary files a/dev/.doctrees/notebooks/1.3-Continuous_HGF.doctree and b/dev/.doctrees/notebooks/1.3-Continuous_HGF.doctree differ
diff --git a/dev/.doctrees/notebooks/2-Using_custom_response_functions.doctree b/dev/.doctrees/notebooks/2-Using_custom_response_functions.doctree
index 739885a94..d0d54cde1 100644
Binary files a/dev/.doctrees/notebooks/2-Using_custom_response_functions.doctree and b/dev/.doctrees/notebooks/2-Using_custom_response_functions.doctree differ
diff --git a/dev/.doctrees/notebooks/3-Multilevel_HGF.doctree b/dev/.doctrees/notebooks/3-Multilevel_HGF.doctree
index c48e461de..ae54eddd6 100644
Binary files a/dev/.doctrees/notebooks/3-Multilevel_HGF.doctree and b/dev/.doctrees/notebooks/3-Multilevel_HGF.doctree differ
diff --git a/dev/.doctrees/notebooks/4-Parameter_recovery.doctree b/dev/.doctrees/notebooks/4-Parameter_recovery.doctree
index 259a859f6..d3e9af98b 100644
Binary files a/dev/.doctrees/notebooks/4-Parameter_recovery.doctree and b/dev/.doctrees/notebooks/4-Parameter_recovery.doctree differ
diff --git a/dev/.doctrees/notebooks/5-Non_linear_value_coupling.doctree b/dev/.doctrees/notebooks/5-Non_linear_value_coupling.doctree
index 2156511a1..bbcde1cad 100644
Binary files a/dev/.doctrees/notebooks/5-Non_linear_value_coupling.doctree and b/dev/.doctrees/notebooks/5-Non_linear_value_coupling.doctree differ
diff --git a/dev/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree b/dev/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree
index 0bdd8d06d..7cdeedd10 100644
Binary files a/dev/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree and b/dev/.doctrees/notebooks/Example_1_Heart_rate_variability.doctree differ
diff --git a/dev/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree b/dev/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree
index 21849998f..c3695bd51 100644
Binary files a/dev/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree and b/dev/.doctrees/notebooks/Example_2_Input_node_volatility_coupling.doctree differ
diff --git a/dev/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree b/dev/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree
index d97d68492..8fefbe089 100644
Binary files a/dev/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree and b/dev/.doctrees/notebooks/Example_3_Multi_armed_bandit.doctree differ
diff --git a/dev/.doctrees/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.doctree b/dev/.doctrees/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.doctree
index 10993eca4..efba57103 100644
Binary files a/dev/.doctrees/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.doctree and b/dev/.doctrees/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.doctree differ
diff --git a/dev/.doctrees/notebooks/Exercise_2_Bayesian_reinforcement_learning.doctree b/dev/.doctrees/notebooks/Exercise_2_Bayesian_reinforcement_learning.doctree
index eecd11657..cefefbcc9 100644
Binary files a/dev/.doctrees/notebooks/Exercise_2_Bayesian_reinforcement_learning.doctree and b/dev/.doctrees/notebooks/Exercise_2_Bayesian_reinforcement_learning.doctree differ
diff --git a/dev/_images/0114064ed85fc215380fbdabb181c31d981c52cd56374cf008fa5b330f9abcd4.png b/dev/_images/0114064ed85fc215380fbdabb181c31d981c52cd56374cf008fa5b330f9abcd4.png
deleted file mode 100644
index 221e5806d..000000000
Binary files a/dev/_images/0114064ed85fc215380fbdabb181c31d981c52cd56374cf008fa5b330f9abcd4.png and /dev/null differ
diff --git a/dev/_images/0471bfc7f5b457136d95ba644a9bfe011b2f621280ffbb07488b7d5a793e6a04.png b/dev/_images/0471bfc7f5b457136d95ba644a9bfe011b2f621280ffbb07488b7d5a793e6a04.png
new file mode 100644
index 000000000..ea3bfd812
Binary files /dev/null and b/dev/_images/0471bfc7f5b457136d95ba644a9bfe011b2f621280ffbb07488b7d5a793e6a04.png differ
diff --git a/dev/_images/153b556a97f25d970d6291b1327cb3fcb130c3b279d03dc7b8cde6ca0282360c.png b/dev/_images/153b556a97f25d970d6291b1327cb3fcb130c3b279d03dc7b8cde6ca0282360c.png
deleted file mode 100644
index 8bb73c303..000000000
Binary files a/dev/_images/153b556a97f25d970d6291b1327cb3fcb130c3b279d03dc7b8cde6ca0282360c.png and /dev/null differ
diff --git a/dev/_images/15dc40f4e6e445ad3c9d895fa3a76748cacf076c24ee80be754b237051cdc03c.png b/dev/_images/15dc40f4e6e445ad3c9d895fa3a76748cacf076c24ee80be754b237051cdc03c.png
new file mode 100644
index 000000000..35e475074
Binary files /dev/null and b/dev/_images/15dc40f4e6e445ad3c9d895fa3a76748cacf076c24ee80be754b237051cdc03c.png differ
diff --git a/dev/_images/1b2cd3602e88f0369f1e7f6ddc2996c4ff2d4d8f6adad0df74624b86e44acae7.png b/dev/_images/1b2cd3602e88f0369f1e7f6ddc2996c4ff2d4d8f6adad0df74624b86e44acae7.png
new file mode 100644
index 000000000..70a6540ee
Binary files /dev/null and b/dev/_images/1b2cd3602e88f0369f1e7f6ddc2996c4ff2d4d8f6adad0df74624b86e44acae7.png differ
diff --git a/dev/_images/233a6798deb950bae4abbce540aa2dfeecb8eed2dff2588e1274a89d750fa37b.png b/dev/_images/233a6798deb950bae4abbce540aa2dfeecb8eed2dff2588e1274a89d750fa37b.png
deleted file mode 100644
index 1e31516e3..000000000
Binary files a/dev/_images/233a6798deb950bae4abbce540aa2dfeecb8eed2dff2588e1274a89d750fa37b.png and /dev/null differ
diff --git a/dev/_images/24a47a30e61963b01e9fe86fc49f9457db4b4cba70774597cb0b25bd4df90686.png b/dev/_images/24a47a30e61963b01e9fe86fc49f9457db4b4cba70774597cb0b25bd4df90686.png
deleted file mode 100644
index 052cb669e..000000000
Binary files a/dev/_images/24a47a30e61963b01e9fe86fc49f9457db4b4cba70774597cb0b25bd4df90686.png and /dev/null differ
diff --git a/dev/_images/27c79f989dc37a089cf770c31b5b78ec48ceb2afa29bf7edf7045201c97ee7e8.png b/dev/_images/27c79f989dc37a089cf770c31b5b78ec48ceb2afa29bf7edf7045201c97ee7e8.png
new file mode 100644
index 000000000..0f4520e72
Binary files /dev/null and b/dev/_images/27c79f989dc37a089cf770c31b5b78ec48ceb2afa29bf7edf7045201c97ee7e8.png differ
diff --git a/dev/_images/29295e06c1eb5848e8e93428b9578e19846e291904b3a10dac6b94bf6757c447.png b/dev/_images/29295e06c1eb5848e8e93428b9578e19846e291904b3a10dac6b94bf6757c447.png
deleted file mode 100644
index 2661d7162..000000000
Binary files a/dev/_images/29295e06c1eb5848e8e93428b9578e19846e291904b3a10dac6b94bf6757c447.png and /dev/null differ
diff --git a/dev/_images/346c7d30a296bd356a4761f846cd627fa166eed929d24c1963d7e4fa286b9638.png b/dev/_images/346c7d30a296bd356a4761f846cd627fa166eed929d24c1963d7e4fa286b9638.png
new file mode 100644
index 000000000..c49e58332
Binary files /dev/null and b/dev/_images/346c7d30a296bd356a4761f846cd627fa166eed929d24c1963d7e4fa286b9638.png differ
diff --git a/dev/_images/368f7131ff797a25bddc793b46fefba095de3636305043a4322297ba6f4b8b45.png b/dev/_images/368f7131ff797a25bddc793b46fefba095de3636305043a4322297ba6f4b8b45.png
new file mode 100644
index 000000000..2264d76b2
Binary files /dev/null and b/dev/_images/368f7131ff797a25bddc793b46fefba095de3636305043a4322297ba6f4b8b45.png differ
diff --git a/dev/_images/3f9c9d096d1b45be79de680296b186aa4ef181d7e4390ab20a6046f038f7a98c.png b/dev/_images/3f9c9d096d1b45be79de680296b186aa4ef181d7e4390ab20a6046f038f7a98c.png
new file mode 100644
index 000000000..0d9824499
Binary files /dev/null and b/dev/_images/3f9c9d096d1b45be79de680296b186aa4ef181d7e4390ab20a6046f038f7a98c.png differ
diff --git a/dev/_images/42ece0c760ca3462d1a64a909fdbb445dec3a2a13903bb304382033cc9c81657.png b/dev/_images/42ece0c760ca3462d1a64a909fdbb445dec3a2a13903bb304382033cc9c81657.png
new file mode 100644
index 000000000..c637e37f4
Binary files /dev/null and b/dev/_images/42ece0c760ca3462d1a64a909fdbb445dec3a2a13903bb304382033cc9c81657.png differ
diff --git a/dev/_images/444a5b2d70e469c857d0b6f9ff4444b2da08c87eb1a82cf600adef5ec0674682.png b/dev/_images/444a5b2d70e469c857d0b6f9ff4444b2da08c87eb1a82cf600adef5ec0674682.png
deleted file mode 100644
index 3b31c8bb9..000000000
Binary files a/dev/_images/444a5b2d70e469c857d0b6f9ff4444b2da08c87eb1a82cf600adef5ec0674682.png and /dev/null differ
diff --git a/dev/_images/4535da753e514502a56155b3635236909cb08ff6c3118a75b2b5c95c8c290167.png b/dev/_images/4535da753e514502a56155b3635236909cb08ff6c3118a75b2b5c95c8c290167.png
new file mode 100644
index 000000000..2d850aaf5
Binary files /dev/null and b/dev/_images/4535da753e514502a56155b3635236909cb08ff6c3118a75b2b5c95c8c290167.png differ
diff --git a/dev/_images/484b00abaaf4ac30b26ae8ca84aea19f5e0a7f9f5a6630a5ff919f28ff2e7177.png b/dev/_images/484b00abaaf4ac30b26ae8ca84aea19f5e0a7f9f5a6630a5ff919f28ff2e7177.png
deleted file mode 100644
index 44626bf58..000000000
Binary files a/dev/_images/484b00abaaf4ac30b26ae8ca84aea19f5e0a7f9f5a6630a5ff919f28ff2e7177.png and /dev/null differ
diff --git a/dev/_images/552486a3c2daff615e891fe08898274f80724ae60307d06194a91ea136381d07.png b/dev/_images/552486a3c2daff615e891fe08898274f80724ae60307d06194a91ea136381d07.png
new file mode 100644
index 000000000..f858418d6
Binary files /dev/null and b/dev/_images/552486a3c2daff615e891fe08898274f80724ae60307d06194a91ea136381d07.png differ
diff --git a/dev/_images/5cd5e876923055f6e9a080fad3af8d7656f54bdf27cfed81f3dd2225946b8e1f.png b/dev/_images/5cd5e876923055f6e9a080fad3af8d7656f54bdf27cfed81f3dd2225946b8e1f.png
new file mode 100644
index 000000000..71e7f9226
Binary files /dev/null and b/dev/_images/5cd5e876923055f6e9a080fad3af8d7656f54bdf27cfed81f3dd2225946b8e1f.png differ
diff --git a/dev/_images/5d435b3fb0dce7a7d45dea210135000420dfc59cdf7bc28380708a90f0e7acd2.png b/dev/_images/5d435b3fb0dce7a7d45dea210135000420dfc59cdf7bc28380708a90f0e7acd2.png
deleted file mode 100644
index 049a78753..000000000
Binary files a/dev/_images/5d435b3fb0dce7a7d45dea210135000420dfc59cdf7bc28380708a90f0e7acd2.png and /dev/null differ
diff --git a/dev/_images/621bd4bc41eb3bb1c3ba37af682abf523053170fe477709fab35f9e596212895.png b/dev/_images/621bd4bc41eb3bb1c3ba37af682abf523053170fe477709fab35f9e596212895.png
new file mode 100644
index 000000000..3e90ee4d8
Binary files /dev/null and b/dev/_images/621bd4bc41eb3bb1c3ba37af682abf523053170fe477709fab35f9e596212895.png differ
diff --git a/dev/_images/685fda99af028c72672e3206ea8ee16c1a52789d27948be8eaf87fa8d7d98181.png b/dev/_images/685fda99af028c72672e3206ea8ee16c1a52789d27948be8eaf87fa8d7d98181.png
new file mode 100644
index 000000000..fa5ef7e40
Binary files /dev/null and b/dev/_images/685fda99af028c72672e3206ea8ee16c1a52789d27948be8eaf87fa8d7d98181.png differ
diff --git a/dev/_images/75ba141784c329581c8af75cb93813932383738b8ca02503f189e93e19a34736.png b/dev/_images/75ba141784c329581c8af75cb93813932383738b8ca02503f189e93e19a34736.png
deleted file mode 100644
index 134cbbec1..000000000
Binary files a/dev/_images/75ba141784c329581c8af75cb93813932383738b8ca02503f189e93e19a34736.png and /dev/null differ
diff --git a/dev/_images/76b5d8f3b2720d8b557c96ba18307540d719cc34c421c5c0daea3276bc5e79cc.png b/dev/_images/76b5d8f3b2720d8b557c96ba18307540d719cc34c421c5c0daea3276bc5e79cc.png
deleted file mode 100644
index 203b164e4..000000000
Binary files a/dev/_images/76b5d8f3b2720d8b557c96ba18307540d719cc34c421c5c0daea3276bc5e79cc.png and /dev/null differ
diff --git a/dev/_images/777aa544ac4dcbf5f106862e13b673416c3e48aed23b0ed7616079e97a3e5e32.png b/dev/_images/777aa544ac4dcbf5f106862e13b673416c3e48aed23b0ed7616079e97a3e5e32.png
new file mode 100644
index 000000000..9a10c1199
Binary files /dev/null and b/dev/_images/777aa544ac4dcbf5f106862e13b673416c3e48aed23b0ed7616079e97a3e5e32.png differ
diff --git a/dev/_images/7e3bd8562916a392a1a27ed3e938adba5309476da2be4f67e8bcd1e4f37b375e.png b/dev/_images/7e3bd8562916a392a1a27ed3e938adba5309476da2be4f67e8bcd1e4f37b375e.png
deleted file mode 100644
index cca2517f1..000000000
Binary files a/dev/_images/7e3bd8562916a392a1a27ed3e938adba5309476da2be4f67e8bcd1e4f37b375e.png and /dev/null differ
diff --git a/dev/_images/845bbb99a5a68380c8f37c9a41b24da15cdbf5600b51de63e556a08bea0cebf0.png b/dev/_images/845bbb99a5a68380c8f37c9a41b24da15cdbf5600b51de63e556a08bea0cebf0.png
deleted file mode 100644
index 9776e833c..000000000
Binary files a/dev/_images/845bbb99a5a68380c8f37c9a41b24da15cdbf5600b51de63e556a08bea0cebf0.png and /dev/null differ
diff --git a/dev/_images/8b6be87835fa80f2997c25a27b88d7803528429bdf38de7af260c18a3c7aec03.png b/dev/_images/8b6be87835fa80f2997c25a27b88d7803528429bdf38de7af260c18a3c7aec03.png
new file mode 100644
index 000000000..817d52ae9
Binary files /dev/null and b/dev/_images/8b6be87835fa80f2997c25a27b88d7803528429bdf38de7af260c18a3c7aec03.png differ
diff --git a/dev/_images/8f02834ccc45a1fbb4bd4aabfac2d75a89a86b3ef90438ded772ac7bf43de688.svg b/dev/_images/8fc9c210ab62f5dd9a3be3ce533d3c8c7ee05351b66a0f94da4f8a26be10ae39.svg
similarity index 100%
rename from dev/_images/8f02834ccc45a1fbb4bd4aabfac2d75a89a86b3ef90438ded772ac7bf43de688.svg
rename to dev/_images/8fc9c210ab62f5dd9a3be3ce533d3c8c7ee05351b66a0f94da4f8a26be10ae39.svg
index 0255e9b57..a85f1c417 100644
--- a/dev/_images/8f02834ccc45a1fbb4bd4aabfac2d75a89a86b3ef90438ded772ac7bf43de688.svg
+++ b/dev/_images/8fc9c210ab62f5dd9a3be3ce533d3c8c7ee05351b66a0f94da4f8a26be10ae39.svg
@@ -9,22 +9,22 @@
PyHGF is a Python library to create and manipulate dynamic probabilistic networks for predictive coding. The networks can approximate Bayesian inference and optimize beliefs through the diffusion of predictions and precision-weighted prediction errors, and their structure is flexible during both observation and inference. These systems can serve as biologically plausible cognitive models for computational psychiatry and reinforcement learning or as a generalisation of Bayesian filtering to arbitrarily sized dynamic graphical structures for signal processing or decision-making agents. The default implementation supports the generalisation and nodalisation of the Hierarchical Gaussian Filters for predictive coding (gHGF, Weber et al., 2024), but the framework is flexible enough to support any possible algorithm. The library is written on top of JAX, the core functions are derivable and JIT-able whenever feasible and it is possible to sample free parameters from a network under given observations. It is conceived to facilitate manipulation and modularity, so the user can focus on modeling while interfacing smoothly with other libraries in the ecosystem for Bayesian inference or optimization. A binding with an implementation in Rust - that will provide full flexibility on structures during inference - is also under active development. PyHGF is a Python library for creating and manipulating dynamic probabilistic networks for predictive coding. These networks approximate Bayesian inference by optimizing beliefs through the diffusion of predictions and precision-weighted prediction errors. The network structure remains flexible during message-passing steps, allowing for dynamic adjustments. They can be used as a biologically plausible cognitive model in computational neuroscience or as a generalization of Bayesian filtering for designing efficient, modular decision-making agents. The default implementation supports the generalized Hierarchical Gaussian Filters (gHGF, Weber et al., 2024), but the framework is designed to be adaptable to other algorithms. Built on top of JAX, the core functions are differentiable and JIT-compiled where applicable. The library is optimized for modularity and ease of use, allowing seamless integration with other libraries in the ecosystem for Bayesian inference and optimization. Additionally, a binding with an implementation in Rust is under active development, which will further enhance flexibility during inference.PyHGF: A Neural Network Library for Predictive Coding#
-
-
You can find a deeper introduction to how to create and manipulate networks under the following link:
NUTS: [tonic_volatility_2]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 5 seconds.
Array(202.52985, dtype=float32)
+Array(202.52992, dtype=float32)
NUTS: [tonic_volatility_2, tonic_volatility_3]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 9 seconds.
+
+
+Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 8 seconds.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -990,7 +990,7 @@ Sampling#
-
+
Array(203.00745, dtype=float32)
+Array(203.00859, dtype=float32)
NUTS: [tonic_volatility_1]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 7 seconds.
/tmp/ipykernel_3226/2516081684.py:2: UserWarning: The figure layout has changed to tight
+/tmp/ipykernel_3070/2516081684.py:2: UserWarning: The figure layout has changed to tight
plt.tight_layout()
-
+
Array(-1106.1246, dtype=float32)
+Array(-1106.1111, dtype=float32)
NUTS: [tonic_volatility_1]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 9 seconds.
+
+
+Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 10 seconds.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -1040,7 +1040,7 @@ Sampling#
-
+
Array(-1118.0105, dtype=float32)
+Array(-1118.0354, dtype=float32)
NUTS: [tonic_volatility_2]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 5 seconds.
The results above indicate that given the responses provided by the participant, the most likely values for the parameter \(\omega_2\) are between -4.9 and -3.1, with a mean at -3.9 (you can find slightly different values if you sample different actions from the decisions function). We can consider this as an excellent estimate given the sparsity of the data, and the complexity of the model.
@@ -990,14 +990,14 @@NUTS: [mu_volatility, sigma_volatility, volatility, mu_temperature, sigma_temperature, inverse_temperature]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 27 seconds.
The reference values on both posterior distributions indicate the mean of the distribution used for simulation.
@@ -902,17 +902,17 @@Computed from 2000 posterior samples and 3200 observations log-likelihood matrix.
Estimate SE
-elpd_loo -2511.06 76.83
-p_loo 739.62 -
+elpd_loo -2554.73 78.30
+p_loo 794.16 -
There has been a warning during the calculation. Please check the results.
------
Pareto k diagnostic values:
Count Pct.
-(-Inf, 0.70] (good) 2167 67.7%
+(-Inf, 0.70] (good) 2281 71.3%
(0.70, 1] (bad) 1 0.0%
- (1, Inf) (very bad) 1032 32.2%
+ (1, Inf) (very bad) 918 28.7%
NUTS: [censored_volatility, inverse_temperature]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 49 seconds.
+
+
+Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 63 seconds.
-There were 1999 divergences after tuning. Increase `target_accept` or reparameterize.
+There were 1997 divergences after tuning. Increase `target_accept` or reparameterize.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -751,7 +751,7 @@ Visualizing parameters recovery
-
+
@@ -776,15 +776,15 @@ System configuration
diff --git a/dev/notebooks/5-Non_linear_value_coupling.html b/dev/notebooks/5-Non_linear_value_coupling.html
index 45b1e93ec..89d9de3d1 100644
--- a/dev/notebooks/5-Non_linear_value_coupling.html
+++ b/dev/notebooks/5-Non_linear_value_coupling.html
@@ -988,13 +988,13 @@ System configuration
diff --git a/dev/notebooks/Example_1_Heart_rate_variability.html b/dev/notebooks/Example_1_Heart_rate_variability.html
index c026e5bbc..3ccf70aa7 100644
--- a/dev/notebooks/Example_1_Heart_rate_variability.html
+++ b/dev/notebooks/Example_1_Heart_rate_variability.html
@@ -50,7 +50,7 @@
-
+
@@ -574,16 +574,16 @@ Loading and preprocessing physiological recordingDownloading ECG channel: 0%| | 0/2 [00:00<?, ?it/s]
Downloading ECG channel: 50%|█████ | 1/2 [00:00<00:00, 2.35it/s]
+Downloading ECG channel: 50%|█████ | 1/2 [00:01<00:01, 1.14s/it]
-Downloading Respiration channel: 50%|█████ | 1/2 [00:00<00:00, 2.35it/s]
+Downloading Respiration channel: 50%|█████ | 1/2 [00:01<00:01, 1.14s/it]
-Downloading Respiration channel: 100%|██████████| 2/2 [00:00<00:00, 2.19it/s]
+Downloading Respiration channel: 100%|██████████| 2/2 [00:02<00:00, 1.09s/it]
-Downloading Respiration channel: 100%|██████████| 2/2 [00:00<00:00, 2.22it/s]
+Downloading Respiration channel: 100%|██████████| 2/2 [00:02<00:00, 1.10s/it]
@@ -657,7 +657,7 @@ Model#<
-
+
@@ -680,9 +680,9 @@ Model#<
NUTS: [tonic_volatility_2]
-
-
-Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 10 seconds.
+
+
+Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 11 seconds.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -697,7 +697,7 @@ Model#<
-
+
@@ -729,7 +729,7 @@ Model#<
-
+
@@ -754,14 +754,14 @@ System configuration
diff --git a/dev/notebooks/Example_2_Input_node_volatility_coupling.html b/dev/notebooks/Example_2_Input_node_volatility_coupling.html
index 4097345c0..8fbb38e1f 100644
--- a/dev/notebooks/Example_2_Input_node_volatility_coupling.html
+++ b/dev/notebooks/Example_2_Input_node_volatility_coupling.html
@@ -704,12 +704,12 @@ System configuration
diff --git a/dev/notebooks/Example_3_Multi_armed_bandit.html b/dev/notebooks/Example_3_Multi_armed_bandit.html
index a6cc9ef5c..ddb261107 100644
--- a/dev/notebooks/Example_3_Multi_armed_bandit.html
+++ b/dev/notebooks/Example_3_Multi_armed_bandit.html
@@ -52,7 +52,7 @@
-
+
@@ -1085,8 +1085,8 @@ Bayesian inferenceNUTS: [omega]
-
-
+
+
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 7 seconds.
@@ -1108,7 +1108,7 @@ Bayesian inference
-
+
@@ -1133,16 +1133,16 @@ System configuration
diff --git a/dev/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.html b/dev/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.html
index 4073c574c..0b1f34205 100644
--- a/dev/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.html
+++ b/dev/notebooks/Exercise_1_Introduction_to_the_generalised_hierarchical_gaussian_filter.html
@@ -1044,13 +1044,13 @@ System configuration
NUTS: [tonic_volatility_2]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 5 seconds.
NUTS: [bias]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 2 seconds.
Assess model fitting, here using leave-one-out cross-validation from the Arviz library.
@@ -925,12 +925,12 @@/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/multiprocessing/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
+/opt/hostedtoolcache/Python/3.12.7/x64/lib/python3.12/multiprocessing/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
self.pid = os.fork()
-Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 23 seconds.
+
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 22 seconds.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -945,7 +945,7 @@ Rescorla-Wagner
-
+
@@ -1036,8 +1036,8 @@ Two-level HGFNUTS: [tonic_volatility_2]
-
-
+
+
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 5 seconds.
@@ -1053,7 +1053,7 @@ Two-level HGF
-
+
We have saved the pointwise log probabilities as a variable, here we simply move this variable to the log_likelihoo field of the idata
object, so Arviz knows that this can be used later for model comparison.
NUTS: [tonic_volatility_2]
Sampling 2 chains for 1_000 tune and 1_000 draw iterations (2_000 + 2_000 draws total) took 7 seconds.
There were 44 divergences after tuning. Increase `target_accept` or reparameterize.
+There were 52 divergences after tuning. Increase `target_accept` or reparameterize.
We recommend running at least 4 chains for robust computation of convergence diagnostics
@@ -1167,7 +1167,7 @@ Three-level HGF
-
+
Move pointwise estimate into the likelihood field.
@@ -1240,11 +1240,11 @@ Model comparison
-
+
Looking at the final result, we can see that the three-level HGF had the best predictive performance on the participant decision, suggesting that higher-level uncertainty is important here to understand the agent’s behaviour.
@@ -1413,7 +1413,7 @@The resulting samples show belief trajectories for 10 samples for each model (we are not depicting the biased random here for clarity). The trajectories are highly similar, but we can see that the two and three-level HGF are slightly adjusting their learning rates in a way that was more consistent with the observed behaviours.
@@ -1486,15 +1486,15 @@