org.deeplearning4j.util

Class OutputLayerUtil

java.lang.Object

org.deeplearning4j.util.OutputLayerUtil

public class OutputLayerUtil
extends Object

Utility methods for output layer configuration/validation

Author:

Alex Black

Method Summary

Modifier and Type	Method and Description
static boolean	activationExceedsZeroOneRange(IActivation activation, boolean isLossLayer)
static boolean	lossFunctionExpectsProbability(ILossFunction lf)
static void	validateOutputLayer(String layerName, Layer layer) Validate the output layer (or loss layer) configuration, to detect invalid consfiugrations.
static void	validateOutputLayerConfiguration(String layerName, long nOut, boolean isLossLayer, IActivation activation, ILossFunction lossFunction) Validate the output layer (or loss layer) configuration, to detect invalid consfiugrations.

Methods inherited from class java.lang.Object

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

Method Detail

validateOutputLayer

public static void validateOutputLayer(String layerName,
                                       Layer layer)

Validate the output layer (or loss layer) configuration, to detect invalid consfiugrations. A DL4JInvalidConfigException will be thrown for invalid configurations (like softmax + nOut=1).
If the specified layer is not an output layer, this is a no-op

Parameters:

layerName - Name of the layer

layer - Layer

validateOutputLayerConfiguration

public static void validateOutputLayerConfiguration(String layerName,
                                                    long nOut,
                                                    boolean isLossLayer,
                                                    IActivation activation,
                                                    ILossFunction lossFunction)

Validate the output layer (or loss layer) configuration, to detect invalid consfiugrations. A DL4JInvalidConfigException will be thrown for invalid configurations (like softmax + nOut=1).

If the specified layer is not an output layer, this is a no-op

Parameters:

layerName - Name of the layer

nOut - Number of outputs for the layer

isLossLayer - Should be true for loss layers (no params), false for output layers

activation - Activation function

lossFunction - Loss function

lossFunctionExpectsProbability

public static boolean lossFunctionExpectsProbability(ILossFunction lf)

activationExceedsZeroOneRange

public static boolean activationExceedsZeroOneRange(IActivation activation,
                                                    boolean isLossLayer)