def _setup_loss(self):
if self.loss[TYPE] == 'softmax_cross_entropy':
self.train_loss_function = SequenceLoss()
elif self.loss[TYPE] == 'sampled_softmax_cross_entropy':
self.train_loss_function = SampledSoftmaxCrossEntropyLoss(
decoder_obj=self.decoder_obj,
num_classes=self.num_classes,
feature_loss=self.loss,
name='train_loss')
else:
raise ValueError("Loss type {} is not supported. Valid values are "
"'softmax_cross_entropy' or "
"'sampled_softmax_cross_entropy'".format(
self.loss[TYPE]))
self.eval_loss_function = SequenceLossMetric()
def __init__(self, name=None):
super(PerplexityMetric, self).__init__(name=name)
self.loss_function = SequenceLoss(from_logits=False)
def __init__(self, name=None):
super(SequenceLossMetric, self).__init__(name=name)
self.loss_function = SequenceLoss()
