Struct FederatedServer 

pub struct FederatedServer {
    pub global_parameters: HashMap<String, Tensor>,
    pub clients: Vec<FederatedClient>,
    pub aggregation: AggregationStrategy,
    pub round: usize,
}

Federated server

Fields

global_parameters: HashMap<String, Tensor>

Global model parameters

clients: Vec<FederatedClient>

Connected clients

aggregation: AggregationStrategy

Aggregation strategy

round: usize

Current round

Implementations

impl FederatedServer

pub fn new( global_parameters: HashMap<String, Tensor>, aggregation: AggregationStrategy, ) -> Self

Create a new federated server

pub fn register_client(&mut self, client: FederatedClient)

Register a client

pub fn select_clients(&self, fraction: f32) -> Vec<usize>

Select clients for training (random sampling)

pub fn distribute_model(&mut self, client_ids: &[usize])

Distribute global model to clients

pub fn aggregate(&mut self, client_ids: &[usize])

Aggregate client updates

pub fn train_round( &mut self, client_fraction: f32, local_epochs: usize, batch_size: usize, )

Run one round of federated learning

pub fn current_round(&self) -> usize

Get current round number