Skip to content

Instantly share code, notes, and snippets.

@Tachyon5
Created February 24, 2016 18:02
Show Gist options
  • Save Tachyon5/23ba4e39e88a92acb1de to your computer and use it in GitHub Desktop.
Save Tachyon5/23ba4e39e88a92acb1de to your computer and use it in GitHub Desktop.
Explanation of Hidden and Visible in a layer.
MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
.seed(seed)
.gradientNormalization(GradientNormalization.ClipElementWiseAbsoluteValue)
.gradientNormalizationThreshold(1.0)
.iterations(iterations)
.momentum(0.5)
.momentumAfter(Collections.singletonMap(3, 0.9))
.optimizationAlgo(OptimizationAlgorithm.CONJUGATE_GRADIENT)
.list(4)
.layer(0, new RBM.Builder().nIn(numRows*numColumns).nOut(500)
.weightInit(WeightInit.XAVIER).lossFunction(LossFunction.RMSE_XENT)
.visibleUnit(RBM.VisibleUnit.BINARY)
.hiddenUnit(RBM.HiddenUnit.BINARY)
.activation("relu")
.build())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment