p2pfl.learning.frameworks.tensorflow.keras_learner module

Keras learner for P2PFL.

class p2pfl.learning.frameworks.tensorflow.keras_learner.KerasLearner(model=None, data=None, aggregator=None)[source]

Bases: Learner

Learner for TensorFlow/Keras models in P2PFL.

Parameters:

  • model (P2PFLModel | None) – The KerasModel instance.

  • data (P2PFLDataset | None) – The P2PFLDataset instance.

  • addr – The address of this node.

  • aggregator (Aggregator | None)

evaluate()[source]

Evaluate the Keras model.

Return type:

dict[str, float]

fit()[source]

Fit the model.

Return type:

P2PFLModel

get_framework()[source]

Retrieve the learner name.

Return type:

str

Returns:

The name of the learner class.

interrupt_fit()[source]

Interrupt the training process.

Return type:

None

set_addr(addr)[source]

Set the addr of the node.

Parameters:

addr (str)

Return type:

str

set_model(model)[source]

Set the model of the learner.

Parameters:

model (P2PFLModel | list[ndarray] | bytes) – The model of the learner.

Return type:

None