Supported loss functions

DL4J supports all available Keras losses (except for logcosh), namely:

  • mean_squared_error
  • mean_absolute_error
  • mean_absolute_percentage_error
  • mean_squared_logarithmic_error
  • squared_hinge
  • hinge
  • categorical_hinge
  • logcosh
  • categorical_crossentropy
  • sparse_categorical_crossentropy
  • binary_crossentropy
  • kullback_leibler_divergence
  • poisson
  • cosine_proximity

The mapping of Keras loss functions can be found in KerasLossUtils.

Chat with us on Gitter