def _setup_loss(self):
self.train_loss_function = SigmoidCrossEntropyLoss(
name='train_loss'
)
self.eval_loss_function = SigmoidCrossEntropyMetric(
name='eval_loss'
)
def __init__(
self,
feature_loss=None,
name='sigmoid_cross_entropy_metric'
):
super(SigmoidCrossEntropyMetric, self).__init__(name=name)
self.sigmoid_cross_entropy_function = SigmoidCrossEntropyLoss()
def __init__(self, **kwargs):
super().__init__()
self.sigmoid_cross_entropy_function = SigmoidCrossEntropyLoss(**kwargs)
def __init__(self, feature_loss=None, name="sigmoid_cross_entropy_metric"):
super().__init__(name=name)
self.sigmoid_cross_entropy_function = SigmoidCrossEntropyLoss(
feature_loss)