Enum OptimizationMessage 

pub enum OptimizationMessage {
    ParameterUpdate {
        worker_id: usize,
        parameters: HashMap<String, Array2<Float>>,
        gradient: HashMap<String, Array2<Float>>,
        loss: Float,
    },
    ComputeGradient {
        worker_id: usize,
        data_batch: Array2<Float>,
        labels_batch: Array1<usize>,
    },
    CheckConvergence {
        worker_id: usize,
        current_loss: Float,
    },
    TrainingComplete {
        final_parameters: HashMap<String, Array2<Float>>,
        final_loss: Float,
        iterations: usize,
    },
    OptimizationError {
        worker_id: usize,
        error: String,
    },
}

Message types for async optimization communication

Variants

ParameterUpdate

Parameter update request

Fields

worker_id: usize

parameters: HashMap<String, Array2<Float>>

gradient: HashMap<String, Array2<Float>>

loss: Float

ComputeGradient

Gradient computation request

Fields

worker_id: usize

data_batch: Array2<Float>

labels_batch: Array1<usize>

CheckConvergence

Convergence check request

Fields

worker_id: usize

current_loss: Float

TrainingComplete

Training completion signal

Fields

final_parameters: HashMap<String, Array2<Float>>

final_loss: Float

iterations: usize

OptimizationError

Error occurred during optimization

Fields

worker_id: usize

error: String

Trait Implementations

impl Clone for OptimizationMessage

fn clone(&self) -> OptimizationMessage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OptimizationMessage

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

Formats the value using the given formatter.