AggregatorPsiManager#
- class AggregatorPsiManager#
Aggregator side of the PSI for federated learning module.
- get_verbosity(self: pyhelayers.AggregatorPsiManager) pyhelayers.Verbosity#
Get the Verbosity level.
- rearrange_indicator_vector(self: pyhelayers.AggregatorPsiManager, indicator_vector: pyhelayers.CTileTensor, mapping: numpy.ndarray[numpy.uint64]) pyhelayers.CTileTensor#
Rearranges the indicators vector produced in the PSI process according to the mapping sent by the first client.
- Parameters:
indicator_vector – The indicators vector to rearrange
mapping – The mapping by which the aggregator needs to rearrange the indicators vector
- set_verbosity(self: pyhelayers.AggregatorPsiManager, verbosity: pyhelayers.Verbosity) None#
Sets the verbosity level.
- Parameters:
verbosity – Verbosity level
-
class AggregatorPsiManager#
A class for managing the Private Set Intersection for Federated Learning protocol between two parties.
The class is initialized by each party (a.k.a Remote Training System - RTS), and the protocol runs between them with a third player, the aggregator, acting as a mediator. The output of the protocol for each RTS is a CTileTensor encrypted under the public key of the aggregator, containing only the samples which are in the intersection of the samples of each RTS, in the same order.
Public Functions
-
inline const HeContext &getPsiHeContext() const#
Returns the heContext to be used in the Private Set Intersection algorithm.
-
inline const HeContext &getFlHeContext() const#
Returns the heContext to be used in the Federated Learning algorithm.
-
void rearrangeIndicatorVector(CTileTensor &res, const CTileTensor &IndicatorVector, const std::vector<size_t> &mapping) const#
Rearranges the indicators vector produced in the PSI process according to the mapping sent by the first client.
- Parameters:
res – An empty CTileTensor
IndicatorVector – The indicators vector to rearrange
mapping – The mapping by which the aggregator needs to rearrange the indicators vector
-
inline const HeContext &getPsiHeContext() const#