public class OutputLayerUtil extends Object
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.
|
static void |
validateOutputLayerForClassifierEvaluation(Layer outputLayer,
Class<? extends IEvaluation> classifierEval)
Validates if the output layer configuration is valid for classifier evaluation.
|
public static void validateOutputLayer(String layerName, Layer layer)
layerName
- Name of the layerlayer
- Layerpublic static void validateOutputLayerConfiguration(String layerName, long nOut, boolean isLossLayer, IActivation activation, ILossFunction lossFunction)
If the specified layer is not an output layer, this is a no-op
layerName
- Name of the layernOut
- Number of outputs for the layerisLossLayer
- Should be true for loss layers (no params), false for output layersactivation
- Activation functionlossFunction
- Loss functionpublic static boolean lossFunctionExpectsProbability(ILossFunction lf)
public static boolean activationExceedsZeroOneRange(IActivation activation, boolean isLossLayer)
public static void validateOutputLayerForClassifierEvaluation(Layer outputLayer, Class<? extends IEvaluation> classifierEval)
outputLayer
- Output layerclassifierEval
- Class for the classifier evaluationCopyright © 2021. All rights reserved.