Skip to content

Instantly share code, notes, and snippets.

@juwalter

juwalter/D4JDeepAutoencoderExample

Created March 3, 2015 17:58
Show Gist options
  • Save juwalter/773c3c4b96778a87f243 to your computer and use it in GitHub Desktop.
Save juwalter/773c3c4b96778a87f243 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
D4JDeepAutoencoderExample
RandomGenerator gen = new MersenneTwister(123);
MultiLayerConfiguration encoderConf = new NeuralNetConfiguration.Builder().weightInit(WeightInit.VI)
.layerFactory(LayerFactories.getFactory(RBM.class))
.iterations(5)
.lossFunction(LossFunctions.LossFunction.RECONSTRUCTION_CROSSENTROPY).rng(gen)
.learningRate(1e-1).nIn(784).nOut(30).list(3)
.hiddenLayerSizes(new int[]{2000, 1000, 500})
.build();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment