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

Methods

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)