QMLOptimizer

quantrs2_device::quantum_ml_integration::functions

Trait QMLOptimizer 

Source 
pub trait QMLOptimizer: Send + Sync {
    // Required methods
    fn compute_gradients(
        &self,
        model: &QMLModel,
        data: &QMLDataBatch,
    ) -> DeviceResult<Array1<f64>>;
    fn update_parameters(
        &mut self,
        model: &mut QMLModel,
        gradients: &Array1<f64>,
    ) -> DeviceResult<()>;
    fn get_state(&self) -> OptimizerState;
    fn set_state(&mut self, state: OptimizerState) -> DeviceResult<()>;
}
Expand description

QML optimizer trait

Required Methods§

Source

fn compute_gradients( &self, model: &QMLModel, data: &QMLDataBatch, ) -> DeviceResult<Array1<f64>>

Compute gradients

Source

fn update_parameters( &mut self, model: &mut QMLModel, gradients: &Array1<f64>, ) -> DeviceResult<()>

Update parameters

Source

fn get_state(&self) -> OptimizerState

Get optimizer state

Source

fn set_state(&mut self, state: OptimizerState) -> DeviceResult<()>

Set optimizer state

Implementors§