From 3f6717360b9e5dcbdbf1ca57c6142f9a81252c28 Mon Sep 17 00:00:00 2001 From: Adrian Seyboldt Date: Thu, 13 Mar 2025 20:20:56 +0100 Subject: [PATCH 1/3] feat: change defaults for transform adapt --- src/sampler.rs | 4 ++-- src/transform_adapt_strategy.rs | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) 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(), } } From 1dd4f0a46ed6b13dcd920bb148e89dddc78852dc Mon Sep 17 00:00:00 2001 From: Adrian Seyboldt Date: Thu, 13 Mar 2025 20:21:07 +0100 Subject: [PATCH 2/3] chore: update dependencies --- Cargo.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Cargo.toml b/Cargo.toml index 29d00dd..8b4173f 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -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" From fab434d0eeebdcb435f8f88e9859df2df801c26d Mon Sep 17 00:00:00 2001 From: Adrian Seyboldt Date: Tue, 18 Mar 2025 10:01:46 +0100 Subject: [PATCH 3/3] chore(release): bump version --- Cargo.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Cargo.toml b/Cargo.toml index 8b4173f..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 ",