diff --git a/Cargo.toml b/Cargo.toml index 29d00dd..596e96f 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "nuts-rs" -version = "0.15.0" +version = "0.15.1" authors = [ "Adrian Seyboldt ", "PyMC Developers ", @@ -23,7 +23,7 @@ rand_distr = "0.5.0" multiversion = "0.8.0" itertools = "0.14.0" thiserror = "2.0.3" -arrow = { version = "54.1.0", default-features = false, features = ["ffi"] } +arrow = { version = "54.2.0", default-features = false, features = ["ffi"] } rand_chacha = "0.9.0" anyhow = "1.0.72" faer = { version = "0.21.4", default-features = false, features = [ @@ -35,7 +35,7 @@ pulp = "0.21.4" rayon = "1.10.0" [dev-dependencies] -proptest = "1.5.0" +proptest = "1.6.0" pretty_assertions = "1.4.0" criterion = "0.5.1" nix = "0.29.0" diff --git a/src/sampler.rs b/src/sampler.rs index 9e3f00a..e803228 100644 --- a/src/sampler.rs +++ b/src/sampler.rs @@ -151,8 +151,8 @@ impl Default for TransformedNutsSettings { Self { num_tune: 1500, num_draws: 1000, - maxdepth: 8, - max_energy_error: 1000f64, + maxdepth: 10, + max_energy_error: 20f64, store_gradient: false, store_unconstrained: false, store_divergences: false, diff --git a/src/transform_adapt_strategy.rs b/src/transform_adapt_strategy.rs index f4e9d3b..b1fa2b0 100644 --- a/src/transform_adapt_strategy.rs +++ b/src/transform_adapt_strategy.rs @@ -23,10 +23,10 @@ pub struct TransformedSettings { impl Default for TransformedSettings { fn default() -> Self { Self { - step_size_window: 0.1f64, - transform_update_freq: 50, + step_size_window: 0.07f64, + transform_update_freq: 128, use_orbit_for_training: false, - transform_train_max_energy_error: 50f64, + transform_train_max_energy_error: 20f64, dual_average_options: Default::default(), } }