com.kotlinnlp.neuralparser.parsers.lhrparser.neuralmodels.labeler.utils.Softmax.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of neuralparser Show documentation

NeuralParser is a very simple to use dependency parser, based on the SimpleDNN library and the SyntaxDecoder transition systems framework.

There is a newer version: 0.6.5

Show newest version

/* Copyright 2018-present LHRParser Authors. All Rights Reserved.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * -----------------------------------------------------------------------------*/

package com.kotlinnlp.neuralparser.parsers.lhrparser.neuralmodels.labeler.utils

import com.kotlinnlp.simplednn.core.functionalities.losses.SoftmaxCrossEntropyCalculator
import com.kotlinnlp.simplednn.simplemath.ndarray.dense.DenseNDArray

/**
 * The loss criterion that calculates the errors with the cross-entropy softmax.
 */
class Softmax : LossCriterion {

  /**
   * @param prediction a prediction array
   * @param goldIndex the index of the gold value
   *
   * @return the errors of the given prediction
   */
  override fun getPredictionErrors(prediction: DenseNDArray, goldIndex: Int): DenseNDArray =
    SoftmaxCrossEntropyCalculator().calculateErrors(output = prediction, goldIndex = goldIndex)
}