コード例 #1
0
ファイル: sparsemax.py プロジェクト: DailyEquation/addons 
 def call(self, inputs):
     return sparsemax(inputs, axis=self.axis)
コード例 #2
0
def sparsemax_loss_from_logits(y_true: TensorLike,
                               logits_pred: TensorLike) -> tf.Tensor:
    y_pred = sparsemax(logits_pred)
    loss = sparsemax_loss(logits_pred, y_pred, y_true)
    return loss
コード例 #3
0
def sparsemax_loss_from_logits(y_true, logits_pred):
    y_pred = sparsemax(logits_pred)
    loss = sparsemax_loss(logits_pred, y_pred, y_true)
    return loss