weka.dl4j.layers

Class GlobalPoolingLayer

java.lang.Object

weka.dl4j.layers.Layer<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

weka.dl4j.layers.GlobalPoolingLayer

All Implemented Interfaces:

java.io.Serializable, weka.core.OptionHandler, ApiWrapper<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

public class GlobalPoolingLayer
extends Layer<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

A version of DeepLearning4j's GlobalPooling that implements WEKA option handling.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GlobalPoolingLayer() Constructor for setting some defaults.

Method Summary

Modifier and Type	Method and Description
java.lang.String[]	getOptions() Gets the current settings of the Classifier.
int	getPnorm()
int[]	getPoolingDimensions()
PoolingType	getPoolingType()
java.lang.String	globalInfo() Global info.
void	initializeBackend() Initialize the DL4J backend.
boolean	isCollapseDimensions()
java.util.Enumeration<weka.core.Option>	listOptions() Returns an enumeration describing the available options.
void	setCollapseDimensions(boolean collapseDimensions)
void	setOptions(java.lang.String[] options) Parses a given list of options.
void	setPnorm(int pnorm)
void	setPoolingDimensions(int[] poolingDimensions)
void	setPoolingType(PoolingType poolingType)

Methods inherited from class weka.dl4j.layers.Layer

create, getBackend, getLayerName, setBackend, setLayerName

Methods inherited from class java.lang.Object

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

Methods inherited from interface weka.core.OptionHandler

makeCopy

Constructor Detail

GlobalPoolingLayer

public GlobalPoolingLayer()

Constructor for setting some defaults.

Method Detail

initializeBackend

public void initializeBackend()

Description copied from interface: ApiWrapper

Initialize the DL4J backend.

getPoolingType

@OptionMetadata(displayName="pooling type",
                description="The type of pooling to use (default = MAX; options: MAX, AVG, SUM, NONE).",
                commandLineParamName="poolingType",
                commandLineParamSynopsis="-poolingType <string>",
                displayOrder=10)
public PoolingType getPoolingType()

setPoolingType

public void setPoolingType(PoolingType poolingType)

getPoolingDimensions

@OptionMetadata(displayName="pooling dimensions",
                description="The pooling dimensions (default = [2,2]).",
                commandLineParamName="poolDimensions",
                commandLineParamSynopsis="-poolDimensions <int>",
                displayOrder=4)
public int[] getPoolingDimensions()

setPoolingDimensions

public void setPoolingDimensions(int[] poolingDimensions)

getPnorm

@OptionMetadata(displayName="pnorm",
                description="The value of the pnorm parameter (default = 2).",
                commandLineParamName="pnorm",
                commandLineParamSynopsis="-pnorm <int>",
                displayOrder=3)
public int getPnorm()

setPnorm

public void setPnorm(int pnorm)

isCollapseDimensions

@OptionMetadata(displayName="collapse dimensions",
                description="Wether to collapse dimensions (default = true).",
                commandLineParamName="collapseDimensions",
                commandLineParamSynopsis="-collapseDimensions <boolean>",
                displayOrder=11)
public boolean isCollapseDimensions()

setCollapseDimensions

public void setCollapseDimensions(boolean collapseDimensions)

globalInfo

public java.lang.String globalInfo()

Global info.

Returns:

string describing this class.

listOptions

public java.util.Enumeration<weka.core.Option> listOptions()

Returns an enumeration describing the available options.

Specified by:

listOptions in interface weka.core.OptionHandler

Overrides:

listOptions in class Layer<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Returns:

an enumeration of all the available options.

getOptions

public java.lang.String[] getOptions()

Gets the current settings of the Classifier.

Specified by:

getOptions in interface weka.core.OptionHandler

Overrides:

getOptions in class Layer<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Returns:

an array of strings suitable for passing to setOptions

setOptions

public void setOptions(java.lang.String[] options)
                throws java.lang.Exception

Parses a given list of options.

Specified by:

setOptions in interface weka.core.OptionHandler

Overrides:

setOptions in class Layer<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Parameters:

options - the list of options as an array of strings

Throws:

java.lang.Exception - if an option is not supported