Source code for otx.api.usecases.reporting.callback

"""Callback module."""

# Copyright (C) 2021-2022 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
#


[docs] class Callback: """Abstract base class used to build new callbacks. Properties params: dict. Training parameters (eg. verbosity, batch size, number of epochs...). model: instance of `keras.models.Model`. Reference of the model being trained. The `logs` dictionary that callback methods take as argument will contain keys for quantities relevant to the current batch or epoch. Currently, the `.fit()` method of the `Sequential` model class will include the following quantities in the `logs` that it passes to its callbacks: on_epoch_end: logs include `acc` and `loss`, and optionally include `val_loss` (if validation is enabled in `fit`), and `val_acc` (if validation and accuracy monitoring are enabled). on_batch_begin: logs include `size`, the number of samples in the current batch. on_batch_end: logs include `loss`, and optionally `acc` (if accuracy monitoring is enabled). """
[docs] def set_params(self, params): """Sets callback parameters.""" # pylint: disable=W0201 self.params = params
[docs] def set_model(self, model): """Sets callback model.""" # pylint: disable=W0201 self.model = model
[docs] def on_epoch_begin(self, epoch, logs=None): """It is called on epoch begin event."""
[docs] def on_epoch_end(self, epoch, logs=None): """It is called on epoch end event."""
[docs] def on_batch_begin(self, batch, logs=None): """It is called on batch begin event."""
[docs] def on_batch_end(self, batch, logs=None): """It is called on batch end event."""
[docs] def on_train_begin(self, logs=None): """It is called on train begin event."""
[docs] def on_train_end(self, logs=None): """It is called on train end event."""
[docs] def on_train_batch_begin(self, batch, logs): """It is called on train batch begin event."""
[docs] def on_train_batch_end(self, batch, logs): """It is called on train batch end event."""
[docs] def on_test_begin(self, logs): """It is called on test begin event."""
[docs] def on_test_end(self, logs): """It is called on test end event."""
[docs] def on_test_batch_begin(self, batch, logs): """It is called on test batch begin event."""
[docs] def on_test_batch_end(self, batch, logs): """It is called on test batch end event."""