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

  • Select an option

    Learn more about clone URLs
  • Save ravithejaburugu/aa9da4356c5edf7b4df922aa690f1030 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ravithejaburugu/aa9da4356c5edf7b4df922aa690f1030 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @juwalter juwalter created this gist Mar 3, 2015.
    8 changes: 8 additions & 0 deletions D4JDeepAutoencoderExample
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,8 @@
    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();