p2pfl.learning.frameworks.learner module

NodeLearning Interface - Template Pattern.

class p2pfl.learning.frameworks.learner.Learner(model=None, data=None, aggregator=None)[source]

Bases: ABC, NodeComponent

Template to implement learning processes, including metric monitoring during training.

Parameters:

  • model (Optional[P2PFLModel]) – The model of the learner.

  • data (Optional[P2PFLDataset]) – The data of the learner.

  • self_addr – The address of the learner.

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any

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

Any