Struct LearningRateScheduler 

pub struct LearningRateScheduler { /* private fields */ }

A learning-rate scheduler that adjusts the rate based on drift state.

• DriftLevel::None: use base_lr.
• DriftLevel::Warning: use base_lr * warning_multiplier.
• DriftLevel::Drift: use base_lr * drift_multiplier.

Space complexity: O(1).

Examples

use rill_ml::drift::{DriftLevel, LearningRateScheduler};

let mut sched = LearningRateScheduler::new(0.01, 2.0, 5.0).unwrap();
assert!((sched.current_lr() - 0.01).abs() < 1e-12);

sched.on_drift_level(DriftLevel::Warning);
assert!((sched.current_lr() - 0.02).abs() < 1e-12);

sched.on_drift_level(DriftLevel::Drift);
assert!((sched.current_lr() - 0.05).abs() < 1e-12);

Implementations

impl LearningRateScheduler

pub fn new( base_lr: f64, warning_multiplier: f64, drift_multiplier: f64, ) -> Result<Self, RillError>

Create a new scheduler.

Returns an error if:

• base_lr is not finite or not strictly positive.
• warning_multiplier is not finite or less than 1.
• drift_multiplier is not finite or less than warning_multiplier.

pub const fn base_lr(&self) -> f64

The configured base learning rate.

pub const fn warning_multiplier(&self) -> f64

The configured warning multiplier.

pub const fn drift_multiplier(&self) -> f64

The configured drift multiplier.

pub const fn current_state(&self) -> DriftLevel

The current drift state.

pub fn on_drift_level(&mut self, level: DriftLevel)

Update the scheduler with the latest drift level from a detector.

pub fn current_lr(&self) -> f64

The current learning rate, adjusted for the drift state.

pub fn reset(&mut self)

Reset to the None drift state (use base_lr).

Trait Implementations

impl Clone for LearningRateScheduler

fn clone(&self) -> LearningRateScheduler

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LearningRateScheduler

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

Formats the value using the given formatter.

impl Default for LearningRateScheduler

fn default() -> Self

Returns the “default value” for a type.