Skip to content

Instantly share code, notes, and snippets.

@HennIdan

HennIdan/Deeplearning4j.java

Created July 7, 2016 15:30
Show Gist options
  • Save HennIdan/b932b32633d81c967f34d0fd769ef302 to your computer and use it in GitHub Desktop.
Save HennIdan/b932b32633d81c967f34d0fd769ef302 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Deeplearning4j.java
MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
.iterations(1)
.weightInit(WeightInit.XAVIER)
.activation("relu")
.optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
.learningRate(0.05)
// ... other hyperparameters
.backprop(true)
.build();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment