Struct BanditOptimization 

pub struct BanditOptimization<E, S> { /* private fields */ }

Bandit-based hyperparameter optimization with cross-validation

Implementations

impl<E, S> BanditOptimization<E, S>

where E: Clone + Fit<ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>> + Predict<ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>>, S: Fn(&E::Fitted, &ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, &ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>) -> Result<f64>,

pub fn new( estimator: E, parameter_space: Vec<ParameterValue>, scorer: Box<S>, cv_folds: usize, ) -> Self

Create a new bandit optimization

pub fn set_n_iter(&mut self, n_iter: usize)

Set number of iterations

pub fn set_strategy(&mut self, strategy: BanditStrategy)

Set bandit strategy

pub fn set_random_state(&mut self, seed: u64)

Set random state for reproducibility

pub fn fit( &mut self, x: &Array2<Float>, y: &Array1<Float>, ) -> Result<BanditOptimizationResult>

Run bandit optimization

Trait Implementations

impl<E: Clone, S: Clone> Clone for BanditOptimization<E, S>

fn clone(&self) -> BanditOptimization<E, S>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<E: Debug, S: Debug> Debug for BanditOptimization<E, S>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.