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

Revisions

  1. HennIdan created this gist Jul 7, 2016.
    9 changes: 9 additions & 0 deletions Deeplearning4j.java
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,9 @@
    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();