MCPcopy Create free account

hub / github.com/JianXu95/FedPAC / functions

Functions84 in github.com/JianXu95/FedPAC

⨍ Functions 84◇ Types & classes 8
↓ 6 callersFunctioncifar_iid
Sample I.I.D. client data from CIFAR10 dataset :param dataset: :param num_users: :return: dict of image index
data_loader.py:222
↓ 6 callersFunctioncifar_noniid_s
Sample non-I.I.D client data from MNIST dataset :param dataset: :param num_users: :return:
data_loader.py:305
↓ 5 callersMethodlocal_training
(self, local_epoch, round=0)
methods/fedavg.py:47
↓ 4 callersFunctionaverage_weights_weighted
Returns the average of the weights.
tools.py:61
↓ 4 callersMethodlocal_test
(self, test_loader, test_model=None)
methods/fedavg.py:29
↓ 4 callersMethodlocal_test
(self, test_loader)
methods/fedper.py:26
↓ 4 callersFunctionmnist_iid
Sample I.I.D. client data from MNIST dataset :param dataset: :param num_users: :return: dict of image index
data_loader.py:17
↓ 3 callersMethodlocal_test
(self, test_loader)
methods/fedpac.py:67
↓ 3 callersMethodupdate_local_model
(self, global_weight)
methods/fedavg.py:44
↓ 2 callersFunctioncinic10_imagenet
()
data_loader.py:752
↓ 2 callersFunctioncinic_separate_iid
Sample I.I.D. client data from CINIC-10 dataset :param dataset: :param num_users: :return: dict of image index
data_loader.py:377
↓ 2 callersFunctioncinic_separate_noniid_s
Sample non-I.I.D client data from MNIST dataset :param dataset: :param num_users: :return:
data_loader.py:510
↓ 2 callersFunctionemnist_noniid_s
Sample non-I.I.D client data from MNIST dataset :param dataset: :param num_users: :return:
data_loader.py:145
↓ 2 callersMethodget_local_protos
(self)
methods/fedpac.py:111
↓ 2 callersMethodlocal_test
(self, test_loader)
methods/local.py:19
↓ 2 callersMethodlocal_test
(self, test_loader)
methods/lg_fedavg.py:27
↓ 2 callersFunctionmnist_noniid_s
Sample non-I.I.D client data from MNIST dataset :param dataset: :param num_users: :return:
data_loader.py:77
↓ 1 callersMethod__init__
(self, num_classes=10)
models/cnn.py:7
↓ 1 callersFunctionagg_classifier_weighted_p
Returns the average of the weights.
tools.py:76
↓ 1 callersMethodaggregate_weight
(self)
methods/fedavg.py:24
↓ 1 callersMethodaggregate_weight
(self)
methods/lg_fedavg.py:22
↓ 1 callersMethodaggregate_weight
(self)
methods/fedpac.py:62
↓ 1 callersMethodaggregate_weight
(self)
methods/fedper.py:21
↓ 1 callersFunctionargs_parser
()
options.py:5
↓ 1 callersFunctioncifar10
()
data_loader.py:682
↓ 1 callersFunctioncinic10
()
data_loader.py:705
↓ 1 callersFunctioncinic10_cifar
()
data_loader.py:728
↓ 1 callersFunctionemnist
()
data_loader.py:667
↓ 1 callersFunctionflatten
(tensors)
tools.py:207
↓ 1 callersFunctionfmnist
()
data_loader.py:652
↓ 1 callersFunctionget_dataset
Returns train and test datasets and a user group which is a dict where the keys are the user index and the values are the corresponding data for
data_loader.py:792
↓ 1 callersFunctionget_gradient
(model)
tools.py:218
↓ 1 callersFunctionget_head_agg_weight
(num_users, Vars, Hs, *args, **kwargs)
tools.py:94
↓ 1 callersFunctiongrad_of
Get the gradient of a given tensor, make it zero if missing. Args: tensor Given instance of/deriving from Tensor Returns: Gradien
tools.py:156
↓ 1 callersFunctiongrads_of
Iterate of the gradients of the given tensors, make zero gradients if missing. Args: tensors Generator of/iterable on instances of/deriving
tools.py:173
↓ 1 callersFunctionlocal_update
(rule)
methods/__init__.py:8
↓ 1 callersFunctionmnist
()
data_loader.py:637
↓ 1 callersMethodnum_flat_features
(self, x)
models/cnn.py:37
↓ 1 callersMethodnum_flat_features
(self, x)
models/cnn.py:69
↓ 1 callersFunctionone_round_training
(rule)
running.py:11
↓ 1 callersFunctionpairwise
Simple generator of the pairs (x, y) in a tuple such that index x < index y. Args: data Indexable (including ability to query length) conta
tools.py:10
↓ 1 callersMethodprior_label
(self, dataset)
methods/fedpac.py:37
↓ 1 callersFunctionrelink
"Relink" the tensors of class (deriving from) Tensor by making them point to another contiguous segment of memory. Args: tensors Generator
tools.py:185
↓ 1 callersMethodsize_label
(self, dataset)
methods/fedpac.py:49
↓ 1 callersMethodstatistics_extraction
(self)
methods/fedpac.py:126
↓ 1 callersMethodupdate_base_model
(self, global_weight)
methods/fedpac.py:85
↓ 1 callersMethodupdate_global_protos
(self, global_protos)
methods/fedpac.py:101
↓ 1 callersMethodupdate_local_classifier
(self, new_weight)
methods/fedpac.py:93
Method__getitem__
(self, item)
data_loader.py:787
Method__init__
(self, dataset, index=None)
data_loader.py:780
Method__init__
(self, idx, args, train_set, test_set, model)
methods/local.py:10
Method__init__
(self, idx, args, train_set, test_set, model)
methods/fedavg.py:12
Method__init__
(self, idx, args, train_set, test_set, model)
methods/lg_fedavg.py:10
Method__init__
(self, idx, args, train_set, test_set, model)
methods/fedpac.py:16
Method__init__
(self, idx, args, train_set, test_set, model)
methods/fedper.py:10
Method__init__
(self, num_classes=10)
models/cnn.py:46
Method__len__
(self)
data_loader.py:784
Functioncifar_noniid
Sample non-I.I.D client data from CIFAR dataset :param dataset: :param num_users: :return:
data_loader.py:263
Functioncinic_separate_noniid
Sample I.I.D. client data from CINIC-10 dataset :param dataset: :param num_users: :return: dict of image index
data_loader.py:442
Methodfeature2logit
(self, x)
models/cnn.py:34
Methodfeature2logit
(self, x)
models/cnn.py:66
Methodforward
(self, x)
models/cnn.py:25
Methodforward
(self, x)
models/cnn.py:58
Functionget_gradient_values
(model)
tools.py:226
Functionget_parameter_values
(model)
tools.py:236
Functionget_protos
Returns the average of the feature embeddings of samples from per-class.
tools.py:23
Methodlocal_fine_tuning
(self, local_epoch, round=0)
methods/fedavg.py:96
Methodlocal_fine_tuning
(self, local_epoch, round=0)
methods/fedper.py:94
Methodlocal_training
(self, local_epoch)
methods/local.py:34
Methodlocal_training
(self, local_epoch)
methods/lg_fedavg.py:50
Methodlocal_training
(self, local_epoch, round=0)
methods/fedpac.py:162
Methodlocal_training
(self, local_epoch, round=0)
methods/fedper.py:49
Functionmnist_noniid
Sample non-I.I.D client data from MNIST dataset :param dataset: :param num_users: :return:
data_loader.py:35
Functionprotos_aggregation
(local_protos_list, local_sizes_list)
tools.py:37
Functionset_gradient
(model, gradient)
tools.py:222
Functionset_gradient_values
(model, gradient)
tools.py:230
Functionset_parameter_values
(model, parameter)
tools.py:240
Functiontrain_round_fedavg
(args, global_model, local_clients, rnd, **kwargs)
running.py:54
Functiontrain_round_fedpac
(args, global_model, local_clients, rnd, **kwargs)
running.py:176
Functiontrain_round_fedper
(args, global_model, local_clients, rnd, **kwargs)
running.py:136
Functiontrain_round_lgfedavg
(args, global_model, local_clients, rnd, **kwargs)
running.py:95
Functiontrain_round_standalone
(args, global_model, local_clients, rnd, **kwargs)
running.py:23
Methodupdate_local_model
(self, global_weight)
methods/lg_fedavg.py:42
Methodupdate_local_model
(self, global_weight)
methods/fedper.py:41