org.deeplearning4j.nn.modelimport.keras.layers

Class KerasLoss

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasLayer

org.deeplearning4j.nn.modelimport.keras.layers.KerasLoss

public class KerasLoss
extends KerasLayer

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

KerasLayer.DimOrder

Field Summary

Fields inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

className, conf, dimOrder, dropout, inboundLayerNames, inputShape, kerasMajorVersion, layer, layerName, originalLayerConfig, outboundLayerNames, vertex, weightL1Regularization, weightL2Regularization, weights

Constructor Summary

Constructors

Constructor and Description
KerasLoss(String layerName, String inboundLayerName, String kerasLoss) Constructor from layer name and input shape.
KerasLoss(String layerName, String inboundLayerName, String kerasLoss, boolean enforceTrainingConfig) Constructor from layer name and input shape.

Method Summary

Modifier and Type	Method and Description
FeedForwardLayer	getLossLayer(InputType type) Get DL4J LossLayer.
InputType	getOutputType(InputType... inputType) Get layer output type.

Methods inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

clearCustomLayers, clearLambdaLayers, copyWeightsToLayer, getClassName, getDimOrder, getInboundLayerNames, getInputPreprocessor, getInputShape, getKerasMajorVersion, getLayer, getLayerName, getNInFromConfig, getNumParams, getVertex, getWeights, isInputPreProcessor, isLayer, isValidInboundLayer, isVertex, registerCustomLayer, registerLambdaLayer, setInboundLayerNames, setLayer, setWeights, usesRegularization

Methods inherited from class java.lang.Object

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

Constructor Detail

KerasLoss

public KerasLoss(String layerName,
                 String inboundLayerName,
                 String kerasLoss)
          throws UnsupportedKerasConfigurationException,
                 InvalidKerasConfigurationException

Constructor from layer name and input shape.

Parameters:

layerName - layer name

inboundLayerName - name of inbound layer

kerasLoss - name of Keras loss function

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasLoss

public KerasLoss(String layerName,
                 String inboundLayerName,
                 String kerasLoss,
                 boolean enforceTrainingConfig)
          throws UnsupportedKerasConfigurationException,
                 InvalidKerasConfigurationException

Constructor from layer name and input shape.

Parameters:

layerName - layer name

inboundLayerName - name of inbound layer

kerasLoss - name of Keras loss function

enforceTrainingConfig - whether to enforce training-related configuration options

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

Method Detail

getLossLayer

public FeedForwardLayer getLossLayer(InputType type)
                              throws UnsupportedKerasConfigurationException

Get DL4J LossLayer.

Returns:

LossLayer

Throws:

UnsupportedKerasConfigurationException

getOutputType

public InputType getOutputType(InputType... inputType)
                        throws InvalidKerasConfigurationException,
                               UnsupportedKerasConfigurationException

Get layer output type.

Overrides:

getOutputType in class KerasLayer

Parameters:

inputType - Array of InputTypes

Returns:

output type as InputType

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException