org.deeplearning4j.nn.modelimport.keras.utils

Class KerasLossUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.utils.KerasLossUtils

public class KerasLossUtils
extends Object

Utility functionality for keras loss functions

Author:

Max Pumperla

Constructor Summary

Constructors

Constructor and Description
KerasLossUtils()

Method Summary

Modifier and Type	Method and Description
static LossFunctions.LossFunction	mapLossFunction(String kerasLoss, KerasLayerConfiguration conf) Map Keras to DL4J loss functions.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KerasLossUtils

public KerasLossUtils()

Method Detail

mapLossFunction

public static LossFunctions.LossFunction mapLossFunction(String kerasLoss,
                                                         KerasLayerConfiguration conf)
                                                  throws UnsupportedKerasConfigurationException

Map Keras to DL4J loss functions.

Parameters:

kerasLoss - String containing Keras loss function name

Returns:

String containing DL4J loss function

Throws:

UnsupportedKerasConfigurationException