Examples with HiddenUnit org.deeplearning4j.nn.conf.layers.RBM.HiddenUnit used on opensource projects

Example 1 with HiddenUnit

use of org.deeplearning4j.nn.conf.layers.RBM.HiddenUnit in project deeplearning4j by deeplearning4j.

the class MultiNeuralNetConfLayerBuilderTest method testNeuralNetConfigAPI.

@Test
public void testNeuralNetConfigAPI() {
    LossFunction newLoss = LossFunction.SQUARED_LOSS;
    int newNumIn = numIn + 1;
    int newNumOut = numOut + 1;
    WeightInit newWeight = WeightInit.UNIFORM;
    double newDrop = 0.5;
    int[] newFS = new int[] { 3, 3 };
    int newFD = 7;
    int[] newStride = new int[] { 3, 3 };
    Convolution.Type newConvType = Convolution.Type.SAME;
    PoolingType newPoolType = PoolingType.AVG;
    double newCorrupt = 0.5;
    double newSparsity = 0.5;
    HiddenUnit newHidden = HiddenUnit.BINARY;
    VisibleUnit newVisible = VisibleUnit.BINARY;
    MultiLayerConfiguration multiConf1 = new NeuralNetConfiguration.Builder().list().layer(0, new DenseLayer.Builder().nIn(newNumIn).nOut(newNumOut).activation(act).build()).layer(1, new DenseLayer.Builder().nIn(newNumIn + 1).nOut(newNumOut + 1).activation(act).build()).build();
    NeuralNetConfiguration firstLayer = multiConf1.getConf(0);
    NeuralNetConfiguration secondLayer = multiConf1.getConf(1);
    assertFalse(firstLayer.equals(secondLayer));
}