org.deeplearning4j.nn.modelimport.keras.utils

Class KerasLossUtils

java.lang.Object

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

public class KerasLossUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
KerasLossUtils()

Method Summary

Modifier and Type	Method and Description
static void	clearCustomLoss() Clear all lambda layers
static ILossFunction	mapLossFunction(String kerasLoss, KerasLayerConfiguration conf) Map Keras to DL4J loss functions.
static void	registerCustomLoss(String lossName, ILossFunction lossFunction) Register a custom loss function

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

registerCustomLoss

public static void registerCustomLoss(String lossName,
                                      ILossFunction lossFunction)

Register a custom loss function

Parameters:

lossName - name of the lambda layer in the serialized Keras model

lossFunction - SameDiffLambdaLayer instance to map to Keras Lambda layer

clearCustomLoss

public static void clearCustomLoss()

Clear all lambda layers

mapLossFunction

public static ILossFunction 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