Enum HyperparamSpace 

pub enum HyperparamSpace {
    Discrete(Vec<HyperparamValue>),
    Continuous {
        min: f64,
        max: f64,
    },
    LogUniform {
        min: f64,
        max: f64,
    },
    IntRange {
        min: i64,
        max: i64,
    },
}

Hyperparameter space definition.

Variants

Discrete(Vec<HyperparamValue>)

Discrete choices.

Continuous

Continuous range [min, max].

Fields

min: f64

max: f64

LogUniform

Log-uniform distribution [min, max].

Fields

min: f64

max: f64

IntRange

Integer range [min, max].

Fields

min: i64

max: i64

Implementations

impl HyperparamSpace

pub fn discrete(values: Vec<HyperparamValue>) -> TrainResult<Self>

Create a discrete choice space.

pub fn continuous(min: f64, max: f64) -> TrainResult<Self>

Create a continuous range space.

pub fn log_uniform(min: f64, max: f64) -> TrainResult<Self>

Create a log-uniform distribution space.

pub fn int_range(min: i64, max: i64) -> TrainResult<Self>

Create an integer range space.

pub fn sample(&self, rng: &mut StdRng) -> HyperparamValue

Sample a value from this space.

pub fn grid_values(&self, num_samples: usize) -> Vec<HyperparamValue>

Get all possible values for grid search (for discrete/int spaces).

Trait Implementations

impl Clone for HyperparamSpace

fn clone(&self) -> HyperparamSpace

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HyperparamSpace

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

Formats the value using the given formatter.