Struct FederatedClient 

pub struct FederatedClient {
    pub id: usize,
    pub parameters: HashMap<String, Tensor>,
    pub num_samples: usize,
    pub learning_rate: f32,
}

Client in federated learning

Fields

id: usize

Client ID

parameters: HashMap<String, Tensor>

Local model parameters

num_samples: usize

Number of local samples

learning_rate: f32

Local learning rate

Implementations

impl FederatedClient

pub fn new( id: usize, parameters: HashMap<String, Tensor>, num_samples: usize, ) -> Self

Create a new federated client

pub fn local_train(&mut self, num_epochs: usize, batch_size: usize)

Perform local training

pub fn get_update( &self, initial_params: &HashMap<String, Tensor>, ) -> HashMap<String, Tensor>

Get model update (difference from initial parameters)

pub fn set_parameters(&mut self, parameters: HashMap<String, Tensor>)

Set parameters

Trait Implementations

impl Clone for FederatedClient

fn clone(&self) -> FederatedClient

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FederatedClient

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

Formats the value using the given formatter.