def loss_function(y_true, y_pred): if isinstance(transform, str) and transform.lower() == 'disc': return losses.discriminative_instance_loss(y_true, y_pred) if focal: return losses.weighted_focal_loss( y_true, y_pred, gamma=gamma, n_classes=n_classes) return losses.weighted_categorical_crossentropy( y_true, y_pred, n_classes=n_classes)
def loss_function(y_true, y_pred): if focal: return losses.weighted_focal_loss(y_true, y_pred, gamma=gamma, n_classes=n_classes, from_logits=False) return losses.weighted_categorical_crossentropy(y_true, y_pred, n_classes=n_classes, from_logits=False)