sup3r.utilities.loss_metrics.Sup3rLoss

sup3r.utilities.loss_metrics.Sup3rLoss#

class Sup3rLoss(gen_features='all', true_features=None)[source]#

Bases: Loss

Base class for custom sup3r loss metrics. This is meant to be used as a base class for loss metrics that require specific input features.

Initialize the loss with given input features

Parameters:

  • gen_features (list | str) – List of generator output features that the loss metric will be calculated on. If ‘all’, the loss will be calculated on all generator features. Otherwise, the loss will be calculated on the features specified in the list. The order of features in the list will be checked to determine the order of features in the generator output tensor.

  • true_features (list | str) – List of true features that the loss metric will be calculated on. If None, this will be the same as gen_features. The order of features in the list will be checked to determine the order of features in the ground truth tensor.

Methods

call(y_true, y_pred)

from_config(config)

get_config()

Attributes

dtype

__call__(y_true, y_pred, sample_weight=None)#

Call self as a function.

Contents