Struct LrRangeTestAnalyzer 

pub struct LrRangeTestAnalyzer {
    pub learning_rates: Vec<f64>,
    pub losses: Vec<f64>,
}

Learning rate range test analyzer for finding optimal learning rates.

Fields

learning_rates: Vec<f64>

Learning rates tested

losses: Vec<f64>

Losses observed at each learning rate

Implementations

impl LrRangeTestAnalyzer

pub fn new(learning_rates: Vec<f64>, losses: Vec<f64>) -> TrainResult<Self>

Create a new analyzer.

pub fn suggest_lr(&self) -> Option<f64>

Find the learning rate with the steepest loss decrease.

pub fn lr_at_min_loss(&self) -> Option<f64>

Find the learning rate at minimum loss.

pub fn plot_ascii(&self, width: usize, height: usize) -> String

Plot the LR range test results (returns a simple ASCII plot).

Trait Implementations

impl Clone for LrRangeTestAnalyzer

fn clone(&self) -> LrRangeTestAnalyzer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LrRangeTestAnalyzer

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

Formats the value using the given formatter.