Skip to content

Instantly share code, notes, and snippets.

@ravithejaburugu

ravithejaburugu/D4JDeepAutoencoderExample

Forked from juwalter/D4JDeepAutoencoderExample
Created January 21, 2017 19:12
Show Gist options
  • Save ravithejaburugu/aa9da4356c5edf7b4df922aa690f1030 to your computer and use it in GitHub Desktop.
Save ravithejaburugu/aa9da4356c5edf7b4df922aa690f1030 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