Class ParameterDictionaryAggregationStrategyBase<T>

Namespace

AiDotNet.FederatedLearning.Aggregators

Assembly

AiDotNet.dll

Base class for aggregation strategies operating on parameter dictionaries.

public abstract class ParameterDictionaryAggregationStrategyBase<T> : AggregationStrategyBase<Dictionary<string, T[]>, T>, IAggregationStrategy<Dictionary<string, T[]>>

Type Parameters

T

Numeric type.

Inheritance

object

FederatedLearningComponentBase<T>

AggregationStrategyBase<Dictionary<string, T[]>, T>

ParameterDictionaryAggregationStrategyBase<T>

Implements

IAggregationStrategy<Dictionary<string, T[]>>

Derived

FedAvgAggregationStrategy<T>

FedBNAggregationStrategy<T>

FedProxAggregationStrategy<T>

Inherited Members

AggregationStrategyBase<Dictionary<string, T[]>, T>.Aggregate(Dictionary<int, Dictionary<string, T[]>>, Dictionary<int, double>)

AggregationStrategyBase<Dictionary<string, T[]>, T>.GetStrategyName()

AggregationStrategyBase<Dictionary<string, T[]>, T>.GetTotalWeightOrThrow(Dictionary<int, double>, IEnumerable<int>, string)

FederatedLearningComponentBase<T>.NumOps

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

object.ToString()

Methods

AggregateLayerWeightedAverageInto(string, Dictionary<int, Dictionary<string, T[]>>, Dictionary<int, double>, double, T[])

protected void AggregateLayerWeightedAverageInto(string layerName, Dictionary<int, Dictionary<string, T[]>> clientModels, Dictionary<int, double> clientWeights, double totalWeight, T[] destination)

Parameters

layerName string

clientModels Dictionary<int, Dictionary<string, T[]>>

clientWeights Dictionary<int, double>

totalWeight double

destination T[]

AggregateWeightedAverage(Dictionary<int, Dictionary<string, T[]>>, Dictionary<int, double>)

protected Dictionary<string, T[]> AggregateWeightedAverage(Dictionary<int, Dictionary<string, T[]>> clientModels, Dictionary<int, double> clientWeights)

Parameters

clientModels Dictionary<int, Dictionary<string, T[]>>

clientWeights Dictionary<int, double>

Returns

Dictionary<string, T[]>

CreateZeroInitializedLayer(int)

protected static T[] CreateZeroInitializedLayer(int length)

Parameters

length int

Returns

T[]