Forked from PatrickNestmeyer/gist:10ae63b04c3279d4d5abae642f4a94e9
Created
January 21, 2017 19:13
-
-
Save ravithejaburugu/ffd48c0b6520265148482062bcc98986 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| conf = new NeuralNetConfiguration.Builder() | |
| .seed(seed) | |
| .iterations(1) | |
| .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT) | |
| .learningRate(0.1) | |
| .updater(Updater.NESTEROVS) | |
| .momentum(0.9) | |
| .list() | |
| .layer(0, new DenseLayer.Builder().nIn(258).nOut(1000) | |
| .weightInit(WeightInit.XAVIER) | |
| .activation("relu") | |
| .build()) | |
| .layer(1, new OutputLayer.Builder(LossFunction.NEGATIVELOGLIKELIHOOD) | |
| .weightInit(WeightInit.XAVIER) | |
| .activation("softmax") | |
| .nIn(1000).nOut(4).build()) | |
| .pretrain(false).backprop(true).build(); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment