Class CategoricalCrossEntropyLoss

java.lang.Object
com.aidenfavish.javaNeuralNetwork.Loss.Loss
com.aidenfavish.javaNeuralNetwork.Loss.CategoricalCrossEntropyLoss

public class CategoricalCrossEntropyLoss extends Loss
  • Constructor Details

    • CategoricalCrossEntropyLoss

      public CategoricalCrossEntropyLoss()
  • Method Details

    • forward

      public float[] forward(Matrix2D yPredict, int[] yTrue)
      Specified by:
      forward in class Loss