org.deeplearning4j.nn.modelimport.keras.layers.core

Class KerasSpatialDropout

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasLayer

org.deeplearning4j.nn.modelimport.keras.layers.core.KerasSpatialDropout

public class KerasSpatialDropout
extends KerasLayer

Keras wrapper for DL4J dropout layer with SpatialDropout, works 1D-3D.

Author:

Max Pumperla

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
KerasSpatialDropout(Integer kerasVersion) Pass-through constructor from KerasLayer
KerasSpatialDropout(Map<String,Object> layerConfig) Constructor from parsed Keras layer configuration dictionary.
KerasSpatialDropout(Map<String,Object> layerConfig, boolean enforceTrainingConfig) Constructor from parsed Keras layer configuration dictionary.

Method Summary

Modifier and Type	Method and Description
InputType	getOutputType(InputType... inputType) Get layer output type.
DropoutLayer	getSpatialDropoutLayer() Get DL4J DropoutLayer with spatial dropout.

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

KerasSpatialDropout

public KerasSpatialDropout(Integer kerasVersion)
                    throws UnsupportedKerasConfigurationException

Pass-through constructor from KerasLayer

Parameters:

kerasVersion - major keras version

Throws:

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasSpatialDropout

public KerasSpatialDropout(Map<String,Object> layerConfig)
                    throws InvalidKerasConfigurationException,
                           UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration.

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasSpatialDropout

public KerasSpatialDropout(Map<String,Object> layerConfig,
                           boolean enforceTrainingConfig)
                    throws InvalidKerasConfigurationException,
                           UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration

enforceTrainingConfig - whether to enforce training-related configuration options

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

Method Detail

getOutputType

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

Get layer output type.

Overrides:

getOutputType in class KerasLayer

Parameters:

inputType - Array of InputTypes

Returns:

output type as InputType

Throws:

InvalidKerasConfigurationException - Invalid Keras config

getSpatialDropoutLayer

public DropoutLayer getSpatialDropoutLayer()

Get DL4J DropoutLayer with spatial dropout.

Returns:

DropoutLayer