hex.genmodel.algos.deeplearning

Class NeuralNetwork

java.lang.Object

hex.genmodel.algos.deeplearning.NeuralNetwork

public class NeuralNetwork
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
java.lang.String	_activation
double[]	_inputs
int	_inSize
int	_maxK
double[]	_outputs
int	_outSize
DeeplearningMojoModel.StoreWeightsBias	_weightsAndBias

Constructor Summary

Constructors

Constructor and Description
NeuralNetwork(java.lang.String activation, double drop_out_ratio, DeeplearningMojoModel.StoreWeightsBias weightsAndBias, double[] inputs, int outSize)

Method Summary

Modifier and Type	Method and Description
ActivationUtils.ActivationFunctions	createActFuns(java.lang.String activation)
double[]	formNNInputs()
double[]	formNNInputsMaxOut()
double[]	fprop1Layer()
void	validateInputs(java.lang.String activation, double drop_out_ratio, int weightLen, int biasLen, int inSize, int outSize)

Methods inherited from class java.lang.Object

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

Field Detail

_activation

public java.lang.String _activation

_weightsAndBias

public DeeplearningMojoModel.StoreWeightsBias _weightsAndBias

_inputs

public double[] _inputs

_outputs

public double[] _outputs

_outSize

public int _outSize

_inSize

public int _inSize

_maxK

public int _maxK

Constructor Detail

NeuralNetwork

public NeuralNetwork(java.lang.String activation,
                     double drop_out_ratio,
                     DeeplearningMojoModel.StoreWeightsBias weightsAndBias,
                     double[] inputs,
                     int outSize)

Method Detail

fprop1Layer

public double[] fprop1Layer()

formNNInputs

public double[] formNNInputs()

formNNInputsMaxOut

public double[] formNNInputsMaxOut()

validateInputs

public void validateInputs(java.lang.String activation,
                           double drop_out_ratio,
                           int weightLen,
                           int biasLen,
                           int inSize,
                           int outSize)

createActFuns

public ActivationUtils.ActivationFunctions createActFuns(java.lang.String activation)