p2pfl.learning.compression.manager module

Transmission compression manager.

class p2pfl.learning.compression.manager.CompressionManager[source]

Bases: object

Manager for compression strategies.

static apply(params, additional_info, techniques)[source]

Apply compression techniques in sequence to the data.

Parameters:

  • params (Any) – The parameters to compress (list[np.ndarray] for weights, dict for trees).

  • additional_info (dict) – Additional information to compress.

  • techniques (dict[str, dict[str, Any]]) – The techniques to apply.

Return type:

bytes

static get_registry()[source]

Return the registry of compression strategies.

Return type:

dict[str, Any]

static reverse(data)[source]

Reverse compression techniques in sequence.

Parameters:

data (bytes) – The serialized data to reverse.

Return type:

tuple[Any, dict]

Returns:

Tuple of (params, additional_info). Params type matches original.