org.nd4j.linalg.api.activation

Interface ActivationFunction

All Superinterfaces:

com.google.common.base.Function<INDArray,INDArray>, java.io.Serializable

All Known Implementing Classes:

BaseActivationFunction, Exp, HardTanh, Linear, MaxOut, RectifiedLinear, RoundedLinear, Sigmoid, SoftMax, Tanh

public interface ActivationFunction
extends com.google.common.base.Function<INDArray,INDArray>, java.io.Serializable

An activation function for a hidden layer for neural networks

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
INDArray	applyDerivative(INDArray input) Applies the derivative of this function
java.lang.Class<? extends ElementWiseOp>	transformClazz() The class used for transformation
java.lang.String	type() Name of the function

Methods inherited from interface com.google.common.base.Function

apply, equals

Method Detail

transformClazz

java.lang.Class<? extends ElementWiseOp> transformClazz()

The class used for transformation

Returns:

the class used for transformation

type

java.lang.String type()

Name of the function

Returns:

the name of the function

applyDerivative

INDArray applyDerivative(INDArray input)

Applies the derivative of this function

Parameters:

input - the input to apply it to

Returns:

the derivative of this function applied to the input