feat: add SuccessiveHalvingPruner for SHA-based trial pruning

2026-02-11 16:47:57 +01:00
parent a522fb2f34
commit 41150057d6
3 changed files with 471 additions and 2 deletions
@@ -202,7 +202,8 @@ pub use parameter::{
    BoolParam, Categorical, CategoricalParam, EnumParam, FloatParam, IntParam, ParamId, Parameter,
 };
 pub use pruner::{
-    MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner, ThresholdPruner,
+    MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner, SuccessiveHalvingPruner,
+    ThresholdPruner,
 };
 pub use sampler::CompletedTrial;
 pub use sampler::grid::GridSearchSampler;
@@ -227,7 +228,8 @@ pub mod prelude {
        BoolParam, Categorical, CategoricalParam, EnumParam, FloatParam, IntParam, Parameter,
    };
    pub use crate::pruner::{
-        MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner, ThresholdPruner,
+        MedianPruner, NopPruner, PatientPruner, PercentilePruner, Pruner, SuccessiveHalvingPruner,
+        ThresholdPruner,
    };
    pub use crate::sampler::CompletedTrial;
    pub use crate::sampler::grid::GridSearchSampler;