Skip to content

core.md

Latest commit

 

History

History
78 lines (59 loc) · 1.19 KB

core.md

File metadata and controls

78 lines (59 loc) · 1.19 KB
Raw

Core API

The core module contains the base classes and fundamental components for collaborative learning.

Base Trainer

.. autoclass:: pydml.core.BaseCollaborativeTrainer
   :members:
   :undoc-members:
   :show-inheritance:

Losses

.. automodule:: pydml.core.losses
   :members:
   :undoc-members:
   :show-inheritance:

CrossEntropyLoss

.. autoclass:: pydml.core.losses.CrossEntropyLoss
   :members:
   :show-inheritance:

KLDivergenceLoss

.. autoclass:: pydml.core.losses.KLDivergenceLoss
   :members:
   :show-inheritance:

DMLLoss

.. autoclass:: pydml.core.losses.DMLLoss
   :members:
   :show-inheritance:

Callbacks

.. automodule:: pydml.core.callbacks
   :members:
   :undoc-members:
   :show-inheritance:

EarlyStopping

.. autoclass:: pydml.core.callbacks.EarlyStopping
   :members:
   :show-inheritance:

ModelCheckpoint

.. autoclass:: pydml.core.callbacks.ModelCheckpoint
   :members:
   :show-inheritance:

TensorBoard

.. autoclass:: pydml.core.callbacks.TensorBoard
   :members:
   :show-inheritance: