Struct BanditSearchCV 

pub struct BanditSearchCV<E> { /* private fields */ }

Bandit-based cross-validation optimizer

Implementations

impl<E> BanditSearchCV<E>

where E: Clone,

pub fn new(estimator: E, param_space: Vec<ParameterValue>) -> Self

Create a new bandit search CV

pub fn with_strategy(self, strategy: BanditStrategy) -> Self

Set the bandit strategy

pub fn with_config(self, config: BanditConfig) -> Self

Set the configuration

pub fn with_scoring(self, scoring: Scoring) -> Self

Set the scoring method

pub fn with_param_config<F>(self, func: F) -> Self

where F: Fn(E, &ParameterValue) -> Result<E> + 'static,

Set the parameter configuration function

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

where F: Clone + Predict<ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>> + Score<ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>, Float = f64>, E: Fit<ArrayBase<OwnedRepr<Float>, Dim<[usize; 2]>, Float>, ArrayBase<OwnedRepr<Float>, Dim<[usize; 1]>, Float>, Fitted = F>, C: CrossValidator,

Fit the bandit search optimizer