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 (Optional[P2PFLModel]) – The KerasModel instance.

  • data (Optional[P2PFLDataset]) – The P2PFLDataset instance.

  • addr – The address of this node.

evaluate(*args: Any, **kwargs: Any) → Any
Return type:

Any

fit(*args: Any, **kwargs: Any) → Any
Return type:

Any

get_framework(*args: Any, **kwargs: Any) → Any
Return type:

Any

interrupt_fit(*args: Any, **kwargs: Any) → Any
Return type:

Any

set_addr(addr)[source]

Set the addr of the node.

Return type:

str

set_model(*args: Any, **kwargs: Any) → Any
Return type:

Any